STT-tensorflow/tensorflow/compiler/tests
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
..
__init__.py Add a __init__.py file for tensorflow/compiler/tests. 2020-02-20 01:01:13 -08:00
adadelta_test.py Add complex support to optimizers 2019-11-01 10:32:30 -07:00
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 Merge pull request from DavidNorman:extend-argmin-argmax-test 2019-06-21 12:24:35 -07:00
binary_ops_test.py Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
bucketize_op_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
BUILD Merge pull request from ROCmSoftwarePlatform:google_upstream_rocm_add_remove_no_rocm_tag 2020-12-21 00:32:48 -08: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 Rationale: 2020-11-03 12:55:59 -08: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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
complex_div_test.py [XLA:CPU/GPU] Fix division of complex numbers in elemental IR emitter. 2019-07-02 03:11:37 -07:00
concat_ops_test.py Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
cond_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
conv2d_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
conv3d_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
dynamic_stitch_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
eager_test.py Rationale: 2020-11-03 12:55:59 -08: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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
fft_test.py [TF:XLA] Remove superfluous calls to .value in fft_test 2019-06-21 12:44:26 -07:00
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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
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 Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
lstm_layer_inference.config.pbtxt Branch 168917534 () 2017-09-15 19:38:25 -07:00
lstm_layer_inference.pbtxt Initial open-source release of XLA: Accelerated Linear Algebra. 2017-01-09 12:26:35 -08:00
lstm_test.py Disable some TF2 incompatible tests from running in v2 mode. 2020-02-11 07:03:23 -08:00
lstm.py ICM PY3 Migration - //tensorflow/compiler [1] 2019-10-10 19:59:19 -07:00
manip_ops_test.py [TF:XLA] Implement support for tf.roll 2019-06-22 02:26:38 -07:00
matrix_band_part_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
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 [TF:XLA] Implement MatrixInverse in terms of QR decomposition and triangular solve 2019-06-26 00:37:54 -07:00
matrix_solve_op_test.py [TF:XLA] Re-implement MatrixSolve in terms of QR decomposition. 2019-10-28 13:49:07 -07:00
matrix_triangular_solve_op_test.py Merge pull request 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 Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
nullary_ops_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
placeholder_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
plugin.bzl Formatting BUILD and bzl files. 2019-02-18 22:24:09 -08:00
pooling_ops_3d_test.py Enable fallback legalization for MaxPoolGradGrad and MaxPool3DGradGrad ops 2020-11-13 13:54:37 -08:00
pooling_ops_test.py Enable fallback legalization for MaxPoolGradGrad and MaxPool3DGradGrad ops 2020-11-13 13:54:37 -08: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 Add a test case for a larger QR decomposition. 2020-12-14 14:08:00 -08: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 Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
reduce_window_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
reshape_op_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
reverse_ops_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
reverse_sequence_op_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
risc_ops_test.py Add lowering from tf.RiscAdd op to HLO_Add. 2020-12-09 16:36:19 -08:00
rmsprop_test.py Add complex support to optimizers 2019-11-01 10:32:30 -07:00
scan_ops_test.py Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08: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 [TF2XLA] Support scalar index slicing. 2019-09-09 17:30:40 -07:00
sort_ops_test.py Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument, 2020-12-21 22:35:25 -08:00
spacetobatch_op_test.py Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
sparse_to_dense_op_test.py Build scatter HLO op with MlirHloBuilder 2020-06-12 17:39:11 -07:00
special_math_test.py Rationale: 2020-11-03 12:55:59 -08:00
stack_ops_test.py Change all compiler tests to use self.session 2019-05-15 17:32:14 -07:00
stateful_random_ops_test.py Internal change 2020-09-17 21:37:45 -07:00
stateless_random_ops_test.py Fix handling of negative seeds in random number generator op kernels for XLA 2020-12-05 19:01:14 -08:00
svd_op_test.py Fix output shapes in Svd op xla op kernel 2020-11-23 21:19:48 -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 Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
test_utils.py ICM PY3 Migration - //tensorflow/compiler [1] 2019-10-10 19:59:19 -07:00
tridiagonal_solve_ops_test.py Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08: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 Internal change 2020-12-16 18:27:38 -08:00
unstack_test.py Add test for unstack op. 2019-12-11 11:44:45 -08:00
variable_ops_test.py Move away from deprecated asserts 2020-06-30 16:10:22 -07:00
where_op_test.py Move WhereOp kernel to the common directory tf2xla/kernels 2020-11-13 13:21:45 -08: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 Always execute TF ops on the local host CPU while constant folding 2020-12-18 22:31:40 -08:00
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 Internal change. 2018-06-14 05:44:45 -07:00
xla_test.py Merge branch 'master' into tf32_test_fix 2020-08-25 10:30:06 -07:00