STT-tensorflow

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