STT-tensorflow

__init__.py

…

adadelta_test.py

…

adagrad_da_test.py

…

adagrad_test.py

…

adam_test.py

…

add_n_test.py

…

argminmax_test.py

…

binary_ops_test.py

Enable complex types for TanhGrad legalization

2021-01-12 03:15:54 -08:00

bucketize_op_test.py

…

BUILD

PY2 removal cleanup

2021-01-15 16:48:57 -08:00

build_defs.bzl

…

case_test.py

Rationale:

2020-11-03 12:55:59 -08:00

categorical_op_test.py

…

cholesky_op_test.py

…

clustering_test.py

…

complex_div_test.py

…

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

…

conv2d_test.py

Support grouped convolution in Conv2DBackpropFilter legalization to HLO

2021-01-15 14:47:51 -08:00

conv3d_test.py

…

conv_node_name_test.py

…

data_format_ops_test.py

…

dense_layer_test.py

…

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

Rationale:

2020-11-03 12:55:59 -08:00

einsum_op_test.py

…

ensure_shape_op_test.py

…

extract_image_patches_op_test.py

…

fake_quant_ops_test.py

…

fft_test.py

…

fifo_queue_test.py

…

ftrl_test.py

…

function_test.py

…

fused_batchnorm_test.py

…

gather_nd_op_test.py

…

gather_test.py

…

image_ops_test.py

Merge pull request #46259 from ROCmSoftwarePlatform:google_updatream_misc_xla_updates_210107

2021-01-07 21:41:49 -08:00

jit_test.py

…

listdiff_op_test.py

…

lrn_ops_test.py

…

lstm_layer_inference.config.pbtxt

…

lstm_layer_inference.pbtxt

…

lstm_test.py

…

lstm.py

…

manip_ops_test.py

…

matrix_band_part_test.py

…

matrix_diag_ops_test.py

…

matrix_inverse_op_test.py

…

matrix_solve_op_test.py

…

matrix_triangular_solve_op_test.py

…

momentum_test.py

…

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

…

placeholder_test.py

…

plugin.bzl

…

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

…

proximal_gradient_descent_test.py

…

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

…

random_ops_test.py

…

randomized_tests.cc

…

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

…

reshape_op_test.py

…

reverse_ops_test.py

…

reverse_sequence_op_test.py

…

risc_ops_test.py

Add lowering from tf.RiscDot op to HLO_Dot.

2021-01-06 09:58:30 -08:00

rmsprop_test.py

…

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

…

segment_reduction_ops_test.py

…

self_adjoint_eig_op_test.py

…

slice_ops_test.py

…

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

…

special_math_test.py

Legalize Zeta and Polygamma ops with the tf2xla fallback

2021-01-11 09:24:03 -08:00

stack_ops_test.py

…

stateful_random_ops_test.py

…

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

…

tensor_list_ops_test.py

…

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

…

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

…

unary_ops_test.py

Fix TF Bitcast and Cast ops to HLO legalization

2021-01-13 01:55:25 -08:00

unstack_test.py

…

variable_ops_test.py

…

where_op_test.py

Move WhereOp kernel to the common directory tf2xla/kernels

2020-11-13 13:21:45 -08:00

while_test.py

…

xla_device_gpu_test.py

…

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

Add shape inference rule for XlaPad.

2020-12-30 14:15:33 -08:00

xla_test_test.py

…

xla_test.py

…