experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
cherrypick_339370490
STT-tensorflow/tensorflow/compiler/tests
History
Smit Hinsu a309b4ee31 Fix TensorFlow ScatterNd op lowering to HLO 
* Use TensorScatterAdd to correctly handle repeated indices
* Support Complex typed operands

Also, enabled complex typed operands for ZerosLikeOp.

PiperOrigin-RevId: 338347731
Change-Id: Ieade5166c3d8e234bda2f090bc636dc6c98931b1
2020-10-21 15:47:51 -07:00
..
__init__.py Add a __init__.py file for tensorflow/compiler/tests. 2020-02-20 01:01:13 -08:00
adadelta_test.py
adagrad_da_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
adagrad_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
adam_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
add_n_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
argminmax_test.py
binary_ops_test.py Whitelist InTopKV2, NextAfter and XlaKeyValueSort ops for the fallback path 2020-10-05 02:45:36 -07:00
bucketize_op_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
BUILD Fix TensorFlow ScatterNd op lowering to HLO 2020-10-21 15:47:51 -07:00
build_defs.bzl Change tf_xla_py_test to use tf's implementation of py_test. 2020-07-22 14:21:20 -07:00
case_test.py [TF2XLA] Fix the test docstring 2020-07-22 14:15:49 -07:00
categorical_op_test.py Work around the following bug in python 3.8 2020-03-17 10:56:24 -07:00
cholesky_op_test.py Fix lint errors 2020-08-18 20:34:13 -07:00
clustering_test.py
complex_div_test.py
concat_ops_test.py Remove unnecessary eval() calls 2020-06-30 17:18:32 -07:00
cond_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
conv2d_test.py
conv3d_test.py
conv_node_name_test.py Reduce Conv2D.__call__ overhead by 45%, Conv3D.__call__ overhead by 30%, and 2020-06-11 10:06:48 -07:00
data_format_ops_test.py update xla data format map ops 2020-09-23 11:51:26 -07:00
dense_layer_test.py Disable some TF2 incompatible tests from running in v2 mode. 2020-02-11 07:03:23 -08:00
depthwise_conv_op_test.py Speed up backprop grouped convolutions 2019-12-26 12:49:55 -08:00
dynamic_slice_ops_test.py
dynamic_stitch_test.py
eager_test.py Replace instances of "blacklist" with "denylist" where possible. See Google Developer guidelines at https://developers.google.com/style/word-list#blacklist for more information. 2020-07-20 16:05:25 -07:00
einsum_op_test.py Remove expired forward compatibility horizons 2020-03-22 13:26:51 +00:00
ensure_shape_op_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
extract_image_patches_op_test.py
fake_quant_ops_test.py
fft_test.py
fifo_queue_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
ftrl_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
function_test.py
fused_batchnorm_test.py Remove expired forward compatibility horizons 2020-04-28 12:01:24 +01:00
gather_nd_op_test.py Enable MLIR bridge for compiler tests requiring unsigned integers 2020-05-05 18:23:23 -07:00
gather_test.py
image_ops_test.py Legalize TensorFlow NonMaxSuppressionV4 and SelfAdjointEigV2Op ops to HLO 2020-07-31 23:17:08 -07:00
jit_test.py Disable some TF2 incompatible tests from running in v2 mode. 2020-02-11 07:03:23 -08:00
listdiff_op_test.py
lrn_ops_test.py
lstm_layer_inference.config.pbtxt
lstm_layer_inference.pbtxt
lstm_test.py Disable some TF2 incompatible tests from running in v2 mode. 2020-02-11 07:03:23 -08:00
lstm.py
manip_ops_test.py
matrix_band_part_test.py
matrix_diag_ops_test.py Remove forward_compatibility / forward_compatible checks for dates that have already passed. 2020-01-02 16:14:15 -08:00
matrix_inverse_op_test.py
matrix_solve_op_test.py
matrix_triangular_solve_op_test.py Merge pull request #42350 from reedwm:tf32_test_fix 2020-08-26 10:50:55 -07:00
momentum_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
nary_ops_test.py
nullary_ops_test.py
placeholder_test.py
plugin.bzl
pooling_ops_3d_test.py Support SAME padding, non-default data formats and AvgPool3D in AvgPool lowering 2020-07-15 14:06:42 -07:00
pooling_ops_test.py Support SAME padding, non-default data formats and AvgPool3D in AvgPool lowering 2020-07-15 14:06:42 -07:00
proximal_adagrad_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
proximal_gradient_descent_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
qr_op_test.py [XLA] Add a larger QR test and slightly relax tolerances for unitary test. 2020-09-23 12:55:35 -07:00
quantized_ops_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
random_ops_test.py [XLA] Broadcast ParameterizedTruncatedNormal parameters to the uniform's shape 2020-04-18 17:47:00 -07:00
randomized_tests.cc Move bfloat16 header to tensorflow/core/platform. 2020-08-14 13:44:57 -07:00
reduce_ops_test.py PR #41916: [TF2XLA] Add EuclideanNorm kernel 2020-08-12 17:02:47 -07:00
reduce_window_test.py
reshape_op_test.py
reverse_ops_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
reverse_sequence_op_test.py
rmsprop_test.py
scan_ops_test.py Legalize TensorFlow Cumprod op to HLO 2020-08-14 23:00:06 -07:00
scatter_nd_op_test.py Fix TensorFlow ScatterNd op lowering to HLO 2020-10-21 15:47:51 -07:00
searchsorted_op_test.py Roll back to fix broken test. 2020-03-06 06:07:51 -08:00
segment_reduction_ops_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
self_adjoint_eig_op_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
slice_ops_test.py
sort_ops_test.py [XLA:CPU] Add a runtime function for F32 TopK and use TopkRewriter to target it 2020-08-05 02:20:45 -07:00
spacetobatch_op_test.py
sparse_to_dense_op_test.py Build scatter HLO op with MlirHloBuilder 2020-06-12 17:39:11 -07:00
special_math_test.py Add XLA registration for Polygamma, to be used for Digamma gradients. 2020-09-18 17:40:20 -07:00
stack_ops_test.py
stateful_random_ops_test.py Internal change 2020-09-17 21:37:45 -07:00
stateless_random_ops_test.py Internal change 2020-09-17 21:37:45 -07:00
svd_op_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
tensor_array_ops_test.py Use self.evaluate for global_variables_initializer 2020-07-01 11:07:20 -07:00
tensor_list_ops_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
ternary_ops_test.py Enable fallback lowering for following TensorFlow ops 2020-08-17 14:32:38 -07:00
test_utils.py
tridiagonal_solve_ops_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
unary_ops_composition_test.cc Rollback of rollback of disabling XLA:CPU/GPU devices by default. 2020-08-11 13:06:59 -07:00
unary_ops_test.py Create a V2 Op to stop the gradient when the input is out of range. 2020-10-12 10:39:36 -07:00
unstack_test.py
variable_ops_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
while_test.py Support Identity/StopGradient ops for TensorLists in tf2xla bridge. 2020-03-03 16:27:54 -08:00
xla_device_gpu_test.py Rollback of rollback of disabling XLA:CPU/GPU devices by default. 2020-08-11 13:06:59 -07:00
xla_device_test.py
xla_ops_test.py Exposes variadic reduce to TF python via xla.py 2020-10-13 13:16:03 -07:00
xla_test_test.py
xla_test.py Merge branch 'master' into tf32_test_fix 2020-08-25 10:30:06 -07:00