| .. |
|
aggregate_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
arg_op.cc
|
Switch CallFrameInterface::GetArg(int, Tensor*) to take (int, const Tensor**).
|
2020-04-10 18:51:55 -07:00 |
|
assert_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
batch_matmul_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
batch_norm_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
batchtospace_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
bcast_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
beta_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
bias_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
binary_ops.cc
|
Add XLA registration for Polygamma, to be used for Digamma gradients.
|
2020-09-18 17:40:20 -07:00 |
|
broadcast_to_op.cc
|
[TF2XLA] Support dynamic slice size in strided slice op.
|
2020-08-21 16:35:46 -07:00 |
|
bucketize_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
BUILD
|
Add target environment constraints to a subset of TensorFlow packages and targets.
|
2020-09-21 10:57:01 -07:00 |
|
case_op.cc
|
generate stateless_case op if all ops in all branches are stateless
|
2020-07-30 16:23:47 -07:00 |
|
case_op.h
|
Do not try to compile trivially dead branches in the Case tf2xla lowering
|
2020-06-19 14:18:26 -07:00 |
|
cast_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
categorical_op.cc
|
Adds double support for StatelessMultinomial (which is really a categorical).
|
2020-10-13 10:34:21 -07:00 |
|
check_numerics_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
cholesky_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
clip_by_value_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
concat_op.cc
|
Fix tf2xla error message in the ConcatOffset and MatrixDiagPart ops
|
2020-08-03 22:06:58 -07:00 |
|
const_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
conv_op_helpers.cc
|
Clean up error message differences between XLA and TF for convolutions
|
2020-07-31 19:16:34 -07:00 |
|
conv_op_helpers.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
conv_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
cross_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
cwise_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
cwise_ops.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
data_format_ops.cc
|
Merge pull request #43226 from kaixih:reduce_ops_layout
|
2020-09-25 12:56:20 -07:00 |
|
depthtospace_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
dequantize_op.cc
|
[TF2XLA] Make dequantize support dynamic range.
|
2020-08-31 22:28:10 -07:00 |
|
device_index_op.cc
|
Add DeviceIndex xla op.
|
2020-06-22 15:06:38 -07:00 |
|
diag_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
dynamic_slice_ops.cc
|
Remove deprecated variants of DynamicSlice and DynamicUpdateSlice builders
|
2020-05-12 21:33:39 -07:00 |
|
dynamic_stitch_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
einsum_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
elu_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
elu_op.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
empty_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
ensure_shape_op.cc
|
Sanitize with clang-format
|
2020-04-14 16:35:30 +00:00 |
|
extract_image_patches_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
fake_param_op.cc
|
[TF2XLA] Support unknown ranked tensor in fake param.
|
2020-04-14 11:15:54 -07:00 |
|
fake_quantize_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
fft_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
fill_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
function_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
gather_op_helpers.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
gather_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
gather_scatter_ops.cc
|
Add CompileTimeConstantInput to XlaGather.
|
2020-08-11 23:13:52 -07:00 |
|
identity_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
if_op.cc
|
[TF2XLA] [NFC] Function are wonderful, let's use them (Simplifying IfOp::Compile)
|
2020-07-24 20:38:41 -07:00 |
|
if_op.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
if_while_utils.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
if_while_utils.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
image_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
image_resize_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
image_resize_ops.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
in_topk_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
index_ops.cc
|
[TF2XLA] Make tf.argmin stable on XLA:TPU
|
2020-08-04 11:32:39 -07:00 |
|
index_ops.h
|
[XLA] Use variadic reduce for argmin/argmax implementation to avoid issues with
|
2019-04-17 17:22:49 -07:00 |
|
l2loss_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
listdiff_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
lower_upper_bound_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
lrn_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
matmul_op.cc
|
[TF:XLA] Small change in tf2xla matmul to use BatchDot instead of Transpose + Dot.
|
2020-05-26 13:22:05 -07:00 |
|
matrix_band_part_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
matrix_diag_ops.cc
|
Fix tf2xla error message in the ConcatOffset and MatrixDiagPart ops
|
2020-08-03 22:06:58 -07:00 |
|
matrix_inverse_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
matrix_solve_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
matrix_triangular_solve_op.cc
|
[TF:XLA] Update TF:XLA tests for matrix_triangular_solve to test V1 and V2.
|
2020-06-17 00:47:58 -07:00 |
|
mirror_pad_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
next_after_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
no_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
one_hot_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
pack_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
pad_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
pooling_ops.cc
|
Additional cleanup of BUILD files.
|
2020-04-06 17:51:58 -07:00 |
|
qr_op.cc
|
[XLA] Add support for complex numbers to Qr decomposition expander.
|
2020-09-22 19:36:07 -07:00 |
|
quantize_and_dequantize_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
random_ops_util.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
random_ops.cc
|
[XLA] Broadcast ParameterizedTruncatedNormal parameters to the uniform's shape
|
2020-04-18 17:47:00 -07:00 |
|
reduce_window_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
reduction_ops_common.cc
|
PR #41916: [TF2XLA] Add EuclideanNorm kernel
|
2020-08-12 17:02:47 -07:00 |
|
reduction_ops.cc
|
PR #41916: [TF2XLA] Add EuclideanNorm kernel
|
2020-08-12 17:02:47 -07:00 |
|
reduction_ops.h
|
PR #41916: [TF2XLA] Add EuclideanNorm kernel
|
2020-08-12 17:02:47 -07:00 |
|
relu_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
relu_op.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
replica_id_op.cc
|
[tf2xla] Convert the result of xla::ReplicaId to S32
|
2020-08-19 10:21:27 -07:00 |
|
resampler_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
reshape_op.cc
|
Introduce dynamic reshape op.
|
2020-08-14 15:48:19 -07:00 |
|
retval_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
reverse_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
reverse_sequence_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
roll_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
scan_ops.cc
|
[TF/XLA] Support F64 conversion for tf.cumsum
|
2020-05-21 19:48:24 -07:00 |
|
scatter_nd_op.cc
|
Add XLA implementation for tensor_scatter_nd_min and tensor_scatter_nd_max, and implement gradient for these functions.
|
2020-09-21 15:40:42 -07:00 |
|
segment_reduction_ops.cc
|
Fix segment_reduction to support dynamic dims correctly.
|
2020-08-21 13:11:42 -07:00 |
|
select_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
sendrecv_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
sequence_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
shape_op.cc
|
Add xla.set_bound op.
|
2020-09-01 19:11:00 -07:00 |
|
shape_util.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
shape_util.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
sharding_op.cc
|
Add gradient annotation for XlaSharding op.
|
2020-07-23 14:12:54 -07:00 |
|
slice_op.cc
|
Remove deprecated variants of DynamicSlice and DynamicUpdateSlice builders
|
2020-05-12 21:33:39 -07:00 |
|
softmax_op.cc
|
Convert Softmax to custom op
|
2020-04-10 09:56:10 -07:00 |
|
sort_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
spacetobatch_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
spacetodepth_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
sparse_to_dense_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
split_op.cc
|
Unify TF and XLA error messages for the SplitV op
|
2020-08-03 20:49:53 -07:00 |
|
spmd_manual_sharding_ops.cc
|
[XLA:SPMD] Add API for the experimental mechanism to mix auto and manual partitioning
|
2020-05-04 16:57:03 -07:00 |
|
stack_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
stateful_random_ops.cc
|
Access stateful_random_ops algorithm value as int64. Algorithm is specified to have int64 type here: https://cs.opensource.google/tensorflow/tensorflow/+/master:tensorflow/core/ops/stateful_random_ops.cc;l=38?q=stateful_random_ops.cc and below in this file.
|
2020-10-21 13:29:40 -07:00 |
|
stateless_random_ops_v2.cc
|
Internal change
|
2020-09-17 21:37:45 -07:00 |
|
stateless_random_ops.cc
|
Internal change
|
2020-09-17 21:37:45 -07:00 |
|
strided_slice_op.cc
|
[XLA] Support strided slice grad in a while loop.
|
2020-09-10 12:12:25 -07:00 |
|
tensor_array_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
tensor_list_ops.cc
|
Support dynamic leading dimension for tensorlist.
|
2020-08-11 15:58:45 -07:00 |
|
tensor_list_utils.cc
|
Support dynamic leading dimension for tensorlist.
|
2020-08-11 15:58:45 -07:00 |
|
tensor_list_utils.h
|
Support dynamic leading dimension for tensorlist.
|
2020-08-11 15:58:45 -07:00 |
|
tile_ops.cc
|
[TF2XLA]Tile Op: Support dynamic multipliers.
|
2020-04-10 12:14:39 -07:00 |
|
topk_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
training_ops.cc
|
Added CPU and GPU implementations of a modified version of FTRL, triggered using an attribute, that multiplies the "linear" accumulator by the learning rate. This enables a learning rate of zero to work correctly.
|
2020-04-07 12:11:41 -07:00 |
|
transpose_op.cc
|
Add support for int64 InvertPermutation Op
|
2020-07-14 10:37:55 -07:00 |
|
tridiagonal_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
unary_ops_composition.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
unary_ops.cc
|
[tf2xla] Add support for PopulationCount
|
2020-08-03 16:33:16 -07:00 |
|
unpack_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
variable_ops.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
while_op.cc
|
Support dynamic leading dimension for tensorlist.
|
2020-08-11 15:58:45 -07:00 |
|
while_op.h
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_broadcast_helper_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_conv_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_dequantize_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_dot_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_pad_op.cc
|
Enabled support for negative edge padding for XlaPad.
|
2020-06-16 07:05:30 -07:00 |
|
xla_reduce_op.cc
|
Exposes variadic reduce to TF python via xla.py
|
2020-10-13 13:16:03 -07:00 |
|
xla_select_and_scatter_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_self_adjoint_eig_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
|
xla_svd_op.cc
|
Automated g4 rollback of changelist 304856650.
|
2020-04-06 11:29:58 -07:00 |
