experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/tf2xla/kernels
History
Yanhua Sun b2f0928940 Add DeviceIndex xla op. 
DeviceIndex op: given a list of device names, this operation returns the index of the device this op runs.  In the case of XLA, we are not executing on any device, we return the length of the list.

PiperOrigin-RevId: 317740778
Change-Id: I0679aa0adc5508b83502eee0d2044584577ed5b4
2020-06-22 15:06:38 -07:00
..
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