experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/tf2xla/kernels
History
A. Unique TensorFlower 682d130cf6 Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument, 
and XlaKeyValueSort for 2 arguments.)

The new op, XlaVariadicSort, fully exposes the xla::Sort functionality: can
take an arbitrary list of operands, can take dimension and
is_stable arguments, and a comparator function.

PiperOrigin-RevId: 348581487
Change-Id: I82b4ff0c37d433541167584475b004626bc8c82c
2020-12-21 22:35:25 -08: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
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
Support tf.unique on TPU
2020-12-07 12:41:46 -08: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
Support dynamic sample size in categorical op.
2020-12-09 14:43:16 -08: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
Fix a bug. The Span returned point to an initializer list that is removed. This isn't allowed.
2020-12-07 06:17:11 -08:00
conv_op_helpers.h
Fix a bug. The Span returned point to an initializer list that is removed. This isn't allowed.
2020-12-07 06:17:11 -08:00
conv_ops.cc
Support dynamic spatial dimensions in convolution.
2020-10-27 10:26:50 -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 #45057 from kaixih:layout_opt_other_ops_pr_v2
2020-11-24 09:20:22 -08: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
Optimize calls to std::string::find() and friends for a single char.
2020-12-17 18:11:13 -08: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
…
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
[tf2xla] Implement MirrorPadGrad
2020-11-02 19:10:11 -08: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.cc
Fix handling of negative seeds in random number generator op kernels for XLA
2020-12-05 19:01:14 -08:00
random_ops_util.h
Fix handling of negative seeds in random number generator op kernels for XLA
2020-12-05 19:01:14 -08: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
Support dynamic sample size in categorical op.
2020-12-09 14:43:16 -08:00
shape_op.cc
Add an op to create bounded dynamic dimensions.
2020-10-26 10:10: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
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
Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument,
2020-12-21 22:35:25 -08: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] Define manual sharding, instead of using fake replication
2020-11-25 11:10:08 -08: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
Fix handling of negative seeds in random number generator op kernels for XLA
2020-12-05 19:01:14 -08:00
stateless_random_ops.cc
Fix handling of negative seeds in random number generator op kernels for XLA
2020-12-05 19:01:14 -08:00
strided_slice_op.cc
Improve error message of strided slice op.
2020-11-03 17:31:56 -08: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
Enable FP16 support for TopKOp for XLA_CPU and XLA_GPU.
2020-11-06 15:36:08 -08:00
training_ops.cc
Decompose ResourceApplyProximalAdagrad op
2020-12-05 19:35:29 -08: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
unique_op.cc
Support tf.unique on TPU
2020-12-07 12:41:46 -08: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
where_op.cc
Move WhereOp kernel to the common directory tf2xla/kernels
2020-11-13 13:21:45 -08:00
while_op.cc
fix typos in compiler dir
2020-10-29 01:08:30 +03: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
Fix output shapes in Svd op xla op kernel
2020-11-23 21:19:48 -08:00