..
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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
broadcast_to_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
bucketize_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
BUILD
Add DeviceIndex xla op.
2020-06-22 15:06:38 -07:00
case_op.cc
Do not try to compile trivially dead branches in the Case tf2xla lowering
2020-06-19 14:18:26 -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
[XLA] Implement stable argmin and argmax
2020-04-18 00:09:30 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
const_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
conv_op_helpers.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
DataFormatVecPermute accepts a vector of size 2.
2020-05-20 16:01:33 -07:00
depthtospace_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
dequantize_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
identity_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
if_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
[XLA] Implement stable argmin and argmax
2020-04-18 00:09:30 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29: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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
reduction_ops.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
reduction_ops.h
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
resampler_ops.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
reshape_op.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
segment_reduction_ops.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
stateless_random_ops.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
strided_slice_op.cc
[Resubmit] Remove dynamic dimension of strided slice grad if input to strided slice is static.
2020-06-18 13:21:58 -07:00
tensor_array_ops.cc
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -07:00
tensor_list_ops.cc
Re-land "Make the tf2xla "tensor list size not set" error message a bit more ergonomic"
2020-05-28 00:15:55 -07:00
tensor_list_utils.cc
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -07:00
tensor_list_utils.h
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
[XLA] Softplus should be monontic
2020-06-08 15:16:26 -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
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -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
Automated g4 rollback of changelist 304856650.
2020-04-06 11:29:58 -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
b2f0928940