455750f362
This sampler supports broadcasting of its input parameters as well as puts the # samples at the left of the output shape, rather than the right. PiperOrigin-RevId: 317129622 Change-Id: I4b62ad2e89a9637ae8b30b73af4b662ad0caa943
3817 lines
108 KiB
Python
3817 lines
108 KiB
Python
# Tests of TensorFlow kernels written using the Python API.
|
|
|
|
load("//tensorflow:tensorflow.bzl", "sycl_py_test", "tf_custom_op_library")
|
|
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
|
|
|
|
# buildifier: disable=same-origin-load
|
|
load("//tensorflow:tensorflow.bzl", "tf_py_test")
|
|
|
|
package(
|
|
default_visibility = ["//tensorflow:internal"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
# CPU-only tests should use tf_py_test, GPU tests use cuda_py_test
|
|
# Please avoid the py_tests and cuda_py_tests (plural) while we
|
|
# fix the shared/overbroad dependencies.
|
|
|
|
tf_py_test(
|
|
name = "as_string_op_test",
|
|
size = "small",
|
|
srcs = ["as_string_op_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "attention_ops_test",
|
|
size = "small",
|
|
srcs = ["attention_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "barrier_ops_test",
|
|
size = "medium", # NOTE(ebrevdo): This test is NOT small.
|
|
srcs = ["barrier_ops_test.py"],
|
|
shard_count = 20,
|
|
# TODO(b/129706424): Re-enable this test on Mac.
|
|
tags = ["no_mac"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "base64_ops_test",
|
|
size = "small",
|
|
srcs = ["base64_ops_test.py"],
|
|
tags = ["nomac"], # b/35468214
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "batch_gather_op_test",
|
|
srcs = ["batch_gather_op_test.py"],
|
|
tags = [
|
|
"no_gpu", # b/127001953
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "batch_scatter_ops_test",
|
|
srcs = ["batch_scatter_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:session",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "bcast_ops_test",
|
|
size = "small",
|
|
srcs = ["bcast_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "list_ops_test",
|
|
size = "small",
|
|
srcs = ["list_ops_test.py"],
|
|
grpc_enabled = True,
|
|
tags = [
|
|
"noasan", # TODO(b/155406705): flaky
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients_impl",
|
|
"//tensorflow/python:list_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:tensor_shape",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "benchmark_test",
|
|
size = "small",
|
|
srcs = ["benchmark_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:platform_benchmark",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reduce_benchmark_test",
|
|
srcs = ["reduce_benchmark_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:platform_benchmark",
|
|
"//tensorflow/python/eager:backprop",
|
|
"//tensorflow/python/eager:context",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "bincount_op_test",
|
|
size = "small",
|
|
srcs = ["bincount_op_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:bincount_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "candidate_sampler_ops_test",
|
|
size = "small",
|
|
srcs = ["candidate_sampler_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:candidate_sampling_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "checkpoint_ops_test",
|
|
size = "medium",
|
|
srcs = ["checkpoint_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:checkpoint_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:partitioned_variables",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cholesky_op_test",
|
|
size = "medium",
|
|
srcs = ["cholesky_op_test.py"],
|
|
shard_count = 5,
|
|
tags = [
|
|
"no_rocm", # TODO(rocm): feature not supported on ROCm platform
|
|
"nomsan", # TODO(b/131773093): Re-enable.
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python/ops/linalg",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "clip_ops_test",
|
|
size = "small",
|
|
srcs = ["clip_ops_test.py"],
|
|
tags = [
|
|
"no_gpu", # b/127001953
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:clip_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "conditional_accumulator_test",
|
|
size = "small",
|
|
srcs = ["conditional_accumulator_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "ctc_decoder_ops_test",
|
|
size = "small",
|
|
srcs = ["ctc_decoder_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:ctc_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "ctc_loss_op_test",
|
|
size = "medium",
|
|
srcs = ["ctc_loss_op_test.py"],
|
|
xla_enable_strict_auto_jit = False, # b/148153828
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:ctc_ops",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "cudnn_deterministic_base",
|
|
srcs = ["cudnn_deterministic_base.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cudnn_deterministic_ops_test",
|
|
size = "small",
|
|
srcs = ["cudnn_deterministic_ops_test.py"],
|
|
xla_enable_strict_auto_jit = True,
|
|
deps = [
|
|
":cudnn_deterministic_base",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cudnn_deterministic_test",
|
|
size = "small",
|
|
srcs = ["cudnn_deterministic_test.py"],
|
|
deps = [
|
|
":cudnn_deterministic_base",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cumulative_logsumexp_test",
|
|
size = "medium",
|
|
srcs = ["cumulative_logsumexp_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:map_fn",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_csv_op_test",
|
|
size = "small",
|
|
srcs = ["decode_csv_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:parsing_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_png_op_test",
|
|
size = "small",
|
|
srcs = ["decode_png_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_bmp_op_test",
|
|
size = "small",
|
|
srcs = ["decode_bmp_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_jpeg_op_test",
|
|
srcs = ["decode_jpeg_op_test.py"],
|
|
data = ["//tensorflow/core:image_testdata"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_image_op_test",
|
|
size = "small",
|
|
srcs = ["decode_image_op_test.py"],
|
|
data = ["//tensorflow/core:image_testdata"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:io_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_raw_op_test",
|
|
size = "small",
|
|
srcs = ["decode_raw_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:parsing_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "decode_compressed_op_test",
|
|
size = "small",
|
|
srcs = ["decode_compressed_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:parsing_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "determinant_op_test",
|
|
size = "medium",
|
|
srcs = ["determinant_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "draw_bounding_box_op_test",
|
|
size = "small",
|
|
srcs = ["draw_bounding_box_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "edit_distance_op_test",
|
|
size = "small",
|
|
srcs = ["edit_distance_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "fifo_queue_test",
|
|
size = "small",
|
|
srcs = ["fifo_queue_test.py"],
|
|
tags = ["no_rocm"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:util",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "fingerprint_op_test",
|
|
size = "small",
|
|
srcs = ["fingerprint_op_test.py"],
|
|
deps = [
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "fractional_avg_pool_op_test",
|
|
size = "small",
|
|
srcs = ["fractional_avg_pool_op_test.py"],
|
|
shard_count = 5,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:nn_ops_gen",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "fractional_max_pool_op_test",
|
|
size = "small",
|
|
srcs = ["fractional_max_pool_op_test.py"],
|
|
shard_count = 5,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:nn_ops_gen",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "identity_op_py_test",
|
|
size = "small",
|
|
srcs = ["identity_op_py_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "identity_n_op_py_test",
|
|
size = "small",
|
|
srcs = ["identity_n_op_py_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "in_topk_op_test",
|
|
size = "small",
|
|
srcs = ["in_topk_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "record_input_test",
|
|
size = "medium",
|
|
srcs = ["record_input_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:io_ops",
|
|
"//tensorflow/python:util",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "io_ops_test",
|
|
size = "small",
|
|
srcs = ["io_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:io_ops",
|
|
"//tensorflow/python:util",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "listdiff_op_test",
|
|
size = "small",
|
|
srcs = ["listdiff_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:util",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "logging_ops_logging_level_test",
|
|
size = "small",
|
|
srcs = ["logging_ops_logging_level_test.py"],
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:logging_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "logging_ops_test",
|
|
size = "small",
|
|
srcs = ["logging_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:logging_ops",
|
|
"//tensorflow/python:math_ops",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "lookup_ops_test",
|
|
size = "small",
|
|
srcs = ["lookup_ops_test.py"],
|
|
grpc_enabled = True,
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:lookup_ops",
|
|
"//tensorflow/python:sparse_tensor",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/data/ops:readers",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "losses_test",
|
|
size = "medium",
|
|
srcs = ["losses_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/ops/losses",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "matrix_exponential_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_exponential_op_test.py"],
|
|
shard_count = 16,
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "matrix_logarithm_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_logarithm_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_inverse_op_test",
|
|
size = "small",
|
|
srcs = ["matrix_inverse_op_test.py"],
|
|
tags = ["optonly"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_solve_ls_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_solve_ls_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_square_root_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_square_root_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_solve_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_solve_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "banded_triangular_solve_op_test",
|
|
size = "small",
|
|
srcs = ["banded_triangular_solve_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_triangular_solve_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_triangular_solve_op_test.py"],
|
|
shard_count = 3,
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "parsing_ops_test",
|
|
size = "medium",
|
|
srcs = ["parsing_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:parsing_ops",
|
|
"//tensorflow/python:platform",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "parse_single_example_op_test",
|
|
size = "small",
|
|
srcs = ["parse_single_example_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:parsing_ops",
|
|
"//tensorflow/python:platform",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "partitioned_variables_test",
|
|
size = "small",
|
|
srcs = ["partitioned_variables_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:partitioned_variables",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "priority_queue_test",
|
|
size = "medium",
|
|
srcs = ["priority_queue_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "resource_variable_ops_test",
|
|
size = "medium",
|
|
srcs = ["resource_variable_ops_test.py"],
|
|
# TODO(b/128347673): Re-enable.
|
|
tags = ["no_windows"],
|
|
tfrt_enabled = True,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:variables",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "regex_replace_op_test",
|
|
size = "small",
|
|
srcs = ["regex_replace_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:string_ops",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "regex_full_match_op_test",
|
|
size = "small",
|
|
srcs = ["regex_full_match_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:string_ops",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "save_restore_ops_test",
|
|
size = "small",
|
|
srcs = ["save_restore_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:io_ops",
|
|
"//tensorflow/python:io_ops_gen",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "scatter_nd_ops_test",
|
|
size = "medium",
|
|
srcs = ["scatter_nd_ops_test.py"],
|
|
tags = ["noasan"], # http://b/32635055
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "segment_reduction_ops_test",
|
|
size = "medium",
|
|
srcs = ["segment_reduction_ops_test.py"],
|
|
shard_count = 10,
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_add_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_add_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_concat_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_concat_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_conditional_accumulator_test",
|
|
size = "small",
|
|
srcs = ["sparse_conditional_accumulator_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_reorder_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_reorder_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_reshape_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_reshape_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_split_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_split_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_slice_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_slice_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_to_dense_op_py_test",
|
|
size = "small",
|
|
srcs = ["sparse_to_dense_op_py_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparsemask_op_test",
|
|
size = "small",
|
|
srcs = ["sparsemask_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_format_op_test",
|
|
size = "small",
|
|
srcs = ["string_format_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:string_ops",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_join_op_test",
|
|
size = "small",
|
|
srcs = ["string_join_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:string_ops",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_split_op_test",
|
|
size = "small",
|
|
srcs = ["string_split_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:string_ops",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python/ops/ragged:ragged_factory_ops",
|
|
"//tensorflow/python/ops/ragged:ragged_string_ops",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_bytes_split_op_test",
|
|
size = "small",
|
|
srcs = ["string_bytes_split_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:string_ops",
|
|
"//tensorflow/python/ops/ragged",
|
|
"//tensorflow/python/ops/ragged:ragged_factory_ops",
|
|
"//tensorflow/python/ops/ragged:ragged_string_ops",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_length_op_test",
|
|
size = "small",
|
|
srcs = ["string_length_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_strip_op_test",
|
|
size = "small",
|
|
srcs = ["string_strip_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_lower_op_test",
|
|
size = "small",
|
|
srcs = ["string_lower_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "string_upper_op_test",
|
|
size = "small",
|
|
srcs = ["string_upper_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "substr_op_test",
|
|
size = "small",
|
|
srcs = ["substr_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "summary_ops_test",
|
|
size = "small",
|
|
srcs = ["summary_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:lib",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:summary_ops_v2",
|
|
"//tensorflow/python:tensor_spec",
|
|
"//tensorflow/python:tensor_util",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:function",
|
|
"@six_archive//:six",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "summary_v1_ops_test",
|
|
size = "small",
|
|
srcs = ["summary_v1_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:logging_ops",
|
|
"//tensorflow/python:summary",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "summary_v1_tensor_op_test",
|
|
size = "small",
|
|
srcs = ["summary_v1_tensor_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:summary",
|
|
"//third_party/py/numpy",
|
|
"@six_archive//:six",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "template_test",
|
|
size = "small",
|
|
srcs = ["template_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:template",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "template_mirrored_strategy_test",
|
|
size = "small",
|
|
srcs = ["template_mirrored_strategy_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:template",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/distribute:distribute_lib",
|
|
"//tensorflow/python/distribute:mirrored_strategy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "tridiagonal_solve_op_test",
|
|
size = "medium",
|
|
srcs = ["tridiagonal_solve_op_test.py"],
|
|
shard_count = 10,
|
|
tags = [
|
|
"no_oss", # TODO(b/142818120): Re-enable.
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "unicode_script_op_test",
|
|
size = "small",
|
|
srcs = ["unicode_script_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:string_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "topk_op_test",
|
|
size = "medium",
|
|
srcs = ["topk_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "nth_element_op_test",
|
|
size = "small",
|
|
srcs = ["nth_element_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "unicode_encode_op_test",
|
|
size = "small",
|
|
srcs = ["unicode_encode_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python/ops/ragged:ragged_factory_ops",
|
|
"//tensorflow/python/ops/ragged:ragged_string_ops",
|
|
"//tensorflow/python/ops/ragged:ragged_tensor",
|
|
"//tensorflow/python/ops/ragged:ragged_tensor_value",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "unicode_transcode_op_test",
|
|
size = "small",
|
|
srcs = ["unicode_transcode_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "unicode_decode_op_test",
|
|
size = "small",
|
|
srcs = ["unicode_decode_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:sparse_tensor",
|
|
"//tensorflow/python:string_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/ops/ragged",
|
|
"//tensorflow/python/ops/ragged:ragged_factory_ops",
|
|
"//tensorflow/python/ops/ragged:ragged_string_ops",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "unique_op_test",
|
|
size = "small",
|
|
srcs = ["unique_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "variable_scope_test",
|
|
size = "small",
|
|
srcs = ["variable_scope_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:layers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:function",
|
|
"//tensorflow/python/eager:wrap_function",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "variables_test",
|
|
size = "small",
|
|
srcs = ["variables_test.py"],
|
|
tags = ["no_windows"], # b/133869052
|
|
tfrt_enabled = True,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:state_ops_gen",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:function",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "where_op_test",
|
|
size = "medium",
|
|
srcs = ["where_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cast_op_test",
|
|
size = "small",
|
|
srcs = ["cast_op_test.py"],
|
|
tags = [
|
|
"no_windows",
|
|
"noasan",
|
|
"noguitar",
|
|
"notap",
|
|
"optonly",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dense_update_ops_no_tsan_test",
|
|
size = "small",
|
|
srcs = ["dense_update_ops_no_tsan_test.py"],
|
|
tags = ["notsan"],
|
|
# TODO (b/140294007): the test fails with XLA.
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "diag_op_test",
|
|
size = "medium",
|
|
srcs = ["diag_op_test.py"],
|
|
shard_count = 6,
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:platform",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "reader_ops_test",
|
|
size = "small",
|
|
srcs = ["reader_ops_test.py"],
|
|
data = ["//tensorflow/core:lmdb_testdata"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:io_ops",
|
|
"//tensorflow/python:lib",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variables",
|
|
"@six_archive//:six",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "aggregate_ops_test",
|
|
size = "small",
|
|
srcs = ["aggregate_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "argmax_op_test",
|
|
size = "small",
|
|
srcs = ["argmax_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "array_ops_test",
|
|
size = "medium",
|
|
srcs = ["array_ops_test.py"],
|
|
shard_count = 10,
|
|
tags = [
|
|
"noasan", # times out
|
|
"optonly", # times out
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:map_fn",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:test_ops",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "broadcast_to_ops_test",
|
|
size = "medium",
|
|
srcs = ["broadcast_to_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "inplace_ops_test",
|
|
size = "small",
|
|
srcs = ["inplace_ops_test.py"],
|
|
shard_count = 10,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "batch_matmul_op_test",
|
|
size = "medium",
|
|
srcs = ["batch_matmul_op_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "batchtospace_op_test",
|
|
size = "small",
|
|
srcs = ["batchtospace_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "betainc_op_test",
|
|
size = "small",
|
|
srcs = ["betainc_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "bias_op_base",
|
|
srcs = ["bias_op_base.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "bias_op_deterministic_test",
|
|
size = "medium",
|
|
srcs = ["bias_op_deterministic_test.py"],
|
|
deps = [
|
|
":bias_op_base",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "bias_op_test",
|
|
size = "medium",
|
|
srcs = ["bias_op_test.py"],
|
|
deps = [
|
|
":bias_op_base",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "bitcast_op_test",
|
|
size = "small",
|
|
srcs = ["bitcast_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "check_ops_test",
|
|
size = "small",
|
|
srcs = ["check_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:check_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "constant_op_test",
|
|
size = "small",
|
|
srcs = ["constant_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:util",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "constant_op_eager_test",
|
|
size = "small",
|
|
srcs = ["constant_op_eager_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:core",
|
|
"//tensorflow/python/eager:test",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "control_flow_ops_py_test",
|
|
size = "medium",
|
|
srcs = ["control_flow_ops_py_test.py"],
|
|
shard_count = 16,
|
|
tags = [
|
|
"notsan", # TODO(b/132205147): Re-enable this.
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:cond_v2",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:data_flow_ops_gen",
|
|
"//tensorflow/python:distributed_framework_test_lib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:functional_ops",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:logging_ops",
|
|
"//tensorflow/python:logging_ops_gen",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:script_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:state_ops_gen",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python:while_v2",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "control_flow_util_test",
|
|
size = "small",
|
|
srcs = ["control_flow_util_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:control_flow_ops_gen",
|
|
"//tensorflow/python:control_flow_util",
|
|
"//tensorflow/python:test_ops",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "control_flow_util_v2_test",
|
|
size = "small",
|
|
srcs = ["control_flow_util_v2_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:cond_v2",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:control_flow_util_v2",
|
|
"//tensorflow/python:while_v2",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv1d_test",
|
|
size = "small",
|
|
srcs = ["conv1d_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv1d_transpose_test",
|
|
size = "small",
|
|
srcs = ["conv1d_transpose_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv2d_transpose_test",
|
|
size = "small",
|
|
srcs = ["conv2d_transpose_test.py"],
|
|
|
|
# TODO(b/144432983): S32 convolutions should not be auto-clustered, only
|
|
# crashes tests.
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv3d_backprop_filter_v2_grad_test",
|
|
size = "small",
|
|
srcs = ["conv3d_backprop_filter_v2_grad_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cross_grad_test",
|
|
size = "small",
|
|
srcs = ["cross_grad_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:math_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "denormal_test",
|
|
size = "small",
|
|
srcs = ["denormal_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:platform",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dense_update_ops_test",
|
|
size = "small",
|
|
srcs = ["dense_update_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "depthtospace_op_test",
|
|
size = "medium",
|
|
srcs = ["depthtospace_op_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "division_past_test",
|
|
size = "medium",
|
|
srcs = ["division_past_test.py"],
|
|
tags = ["manual"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dynamic_partition_op_test",
|
|
size = "medium",
|
|
srcs = ["dynamic_partition_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_grad",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dynamic_stitch_op_test",
|
|
size = "small",
|
|
srcs = ["dynamic_stitch_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_grad",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "extract_image_patches_op_test",
|
|
size = "small",
|
|
srcs = ["extract_image_patches_op_test.py"],
|
|
# TODO(b/144432983): S32 convolutions should not be auto-clustered.
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "extract_volume_patches_op_test",
|
|
size = "small",
|
|
srcs = ["extract_volume_patches_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "functional_ops_test",
|
|
size = "medium",
|
|
srcs = ["functional_ops_test.py"],
|
|
grpc_enabled = True,
|
|
shard_count = 2,
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:functional_ops",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python:while_v2",
|
|
"//tensorflow/python/data/ops:iterator_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "gather_nd_op_test",
|
|
size = "small",
|
|
srcs = ["gather_nd_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "gather_op_test",
|
|
size = "medium",
|
|
srcs = ["gather_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "gradient_correctness_test",
|
|
size = "small",
|
|
srcs = ["gradient_correctness_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "init_ops_test",
|
|
size = "medium",
|
|
srcs = ["init_ops_test.py"],
|
|
shard_count = 4,
|
|
tags = [
|
|
"noasan",
|
|
"notap",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:layers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:partitioned_variables",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "linalg_ops_test",
|
|
size = "medium",
|
|
srcs = ["linalg_ops_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python/ops/linalg",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "lrn_op_test",
|
|
size = "medium",
|
|
srcs = ["lrn_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:nn",
|
|
"//tensorflow/python:nn_grad",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "lu_op_test",
|
|
size = "small",
|
|
srcs = ["lu_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python/ops/linalg",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "einsum_op_test",
|
|
size = "medium",
|
|
srcs = ["einsum_op_test.py"],
|
|
shard_count = 4,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python/ops/linalg",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "manip_ops_test",
|
|
size = "small",
|
|
srcs = ["manip_ops_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:manip_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matmul_op_test",
|
|
size = "medium",
|
|
srcs = ["matmul_op_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "morphological_ops_test",
|
|
size = "small",
|
|
srcs = ["morphological_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "numerics_test",
|
|
size = "small",
|
|
srcs = ["numerics_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:numerics",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "one_hot_op_test",
|
|
size = "small",
|
|
srcs = ["one_hot_op_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "stack_op_test",
|
|
size = "small",
|
|
srcs = ["stack_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "map_fn_test",
|
|
size = "small",
|
|
srcs = ["map_fn_test.py"],
|
|
grpc_enabled = True,
|
|
shard_count = 2,
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:map_fn",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "pad_op_test",
|
|
size = "small",
|
|
srcs = ["pad_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "padding_fifo_queue_test",
|
|
size = "small",
|
|
srcs = ["padding_fifo_queue_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "py_func_test",
|
|
size = "small",
|
|
srcs = ["py_func_test.py"],
|
|
grpc_enabled = True,
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:script_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:function",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reduce_join_op_test",
|
|
size = "small",
|
|
srcs = ["reduce_join_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "unsorted_segment_join_op_test",
|
|
size = "small",
|
|
srcs = ["unsorted_segment_join_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reduction_ops_test",
|
|
size = "medium",
|
|
srcs = ["reduction_ops_test.py"],
|
|
shard_count = 6,
|
|
tags = [
|
|
"no_windows_gpu",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reduction_ops_test_big",
|
|
size = "medium",
|
|
srcs = ["reduction_ops_test_big.py"],
|
|
shard_count = 3,
|
|
tags = [
|
|
"manual",
|
|
"no_gpu",
|
|
"noguitar",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "relu_op_test",
|
|
size = "small",
|
|
srcs = ["relu_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:tf2",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:backprop",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reshape_op_test",
|
|
size = "small",
|
|
srcs = ["reshape_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "reverse_sequence_op_test",
|
|
size = "small",
|
|
srcs = ["reverse_sequence_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "compare_and_bitpack_op_test",
|
|
size = "small",
|
|
srcs = ["compare_and_bitpack_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "scalar_test",
|
|
size = "small",
|
|
srcs = ["scalar_test.py"],
|
|
# b/140221961: Invalid dims for operations
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:io_ops_gen",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "scan_ops_test",
|
|
size = "medium",
|
|
srcs = ["scan_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "session_ops_test",
|
|
size = "small",
|
|
srcs = ["session_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:session_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "shape_ops_test",
|
|
size = "medium",
|
|
srcs = ["shape_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "softmax_op_test",
|
|
size = "medium",
|
|
srcs = ["softmax_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "softplus_op_test",
|
|
size = "small",
|
|
srcs = ["softplus_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "softsign_op_test",
|
|
size = "small",
|
|
srcs = ["softsign_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "spacetobatch_op_test",
|
|
size = "small",
|
|
srcs = ["spacetobatch_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "spacetodepth_op_test",
|
|
size = "medium",
|
|
srcs = ["spacetodepth_op_test.py"],
|
|
tags = [
|
|
"no_windows",
|
|
"no_windows_gpu",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_serialization_ops_test",
|
|
size = "small",
|
|
srcs = ["sparse_serialization_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_tensors_map_ops_test",
|
|
size = "small",
|
|
srcs = ["sparse_tensors_map_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "sparse_tensor_dense_matmul_grad_test",
|
|
size = "small",
|
|
srcs = ["sparse_tensor_dense_matmul_grad_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "sparse_xent_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_xent_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:nn_ops_gen",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "split_op_test",
|
|
size = "medium",
|
|
srcs = ["split_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "stack_ops_test",
|
|
size = "small",
|
|
srcs = ["stack_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:data_flow_ops_gen",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "string_to_hash_bucket_op_test",
|
|
size = "small",
|
|
srcs = ["string_to_hash_bucket_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:string_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "string_to_number_op_test",
|
|
size = "small",
|
|
srcs = ["string_to_number_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:parsing_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "summary_v1_audio_op_test",
|
|
size = "small",
|
|
srcs = ["summary_v1_audio_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:summary",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "summary_v1_image_op_test",
|
|
size = "small",
|
|
srcs = ["summary_v1_image_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:image_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:summary",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "tensor_array_ops_test",
|
|
size = "medium",
|
|
srcs = ["tensor_array_ops_test.py"],
|
|
flaky = 1, # create_local_cluster sometimes times out.
|
|
shard_count = 10,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:cond_v2",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:data_flow_ops_gen",
|
|
"//tensorflow/python:distributed_framework_test_lib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python:tensor_spec",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python:while_v2",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/eager:backprop",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "trace_op_test",
|
|
size = "small",
|
|
srcs = ["trace_op_test.py"],
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "transpose_op_test",
|
|
size = "medium",
|
|
srcs = ["transpose_op_test.py"],
|
|
shard_count = 10,
|
|
tags = [
|
|
"no_gpu",
|
|
"no_oss",
|
|
"notap", # flaky timeout/segfault, b/136498892
|
|
"optonly", # times out
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "unstack_op_test",
|
|
size = "small",
|
|
srcs = ["unstack_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "variable_ops_test",
|
|
size = "small",
|
|
srcs = ["variable_ops_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:state_ops_gen",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "xent_op_test",
|
|
size = "small",
|
|
srcs = ["xent_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:nn_ops_gen",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "zero_division_test",
|
|
size = "medium",
|
|
srcs = ["zero_division_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "atrous_conv2d_test",
|
|
size = "medium",
|
|
srcs = ["atrous_conv2d_test.py"],
|
|
shard_count = 2,
|
|
tags = [
|
|
"no_gpu", # Flaky: b/80127739, b/127001953
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "atrous_convolution_test",
|
|
size = "medium",
|
|
srcs = ["atrous_convolution_test.py"],
|
|
tags = ["manual"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "pool_test",
|
|
size = "medium",
|
|
srcs = ["pool_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv2d_backprop_filter_grad_test",
|
|
size = "medium",
|
|
srcs = ["conv2d_backprop_filter_grad_test.py"],
|
|
shard_count = 2,
|
|
tags = [
|
|
"optonly", # flaky timeouts unless optimized
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv3d_transpose_test",
|
|
size = "medium",
|
|
srcs = ["conv3d_transpose_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv_ops_test",
|
|
size = "medium",
|
|
srcs = ["conv_ops_test.py"],
|
|
shard_count = 4,
|
|
tags = [
|
|
"nogpu", # TODO(b/153763253); flaky
|
|
"optonly", # times out
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:nn",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "depthwise_conv_op_test",
|
|
size = "medium", # http://b/30603882
|
|
timeout = "long",
|
|
srcs = ["depthwise_conv_op_test.py"],
|
|
shard_count = 3,
|
|
# TODO(b/118842098): Re-enable this test in Kokoro.
|
|
tags = ["no_oss"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "neon_depthwise_conv_op_test",
|
|
size = "medium",
|
|
srcs = ["neon_depthwise_conv_op_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "division_future_test",
|
|
size = "medium",
|
|
srcs = ["division_future_test.py"],
|
|
tags = ["manual"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "pooling_ops_3d_test",
|
|
size = "medium",
|
|
srcs = ["pooling_ops_3d_test.py"],
|
|
tags = ["no_rocm"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "pooling_ops_test",
|
|
size = "medium",
|
|
srcs = ["pooling_ops_test.py"],
|
|
shard_count = 4,
|
|
xla_enable_strict_auto_jit = False, # Flaky in XLA b/149568654
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:nn_ops_gen",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "rnn_test",
|
|
size = "medium",
|
|
timeout = "long",
|
|
srcs = ["rnn_test.py"],
|
|
shard_count = 10,
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:data_flow_grad",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:rnn",
|
|
"//tensorflow/python:rnn_cell",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "rnn_cell_test",
|
|
size = "medium",
|
|
srcs = ["rnn_cell_test.py"],
|
|
shard_count = 15,
|
|
tags = ["no_windows"], # b/139739217
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:control_flow_v2_toggles",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:rnn",
|
|
"//tensorflow/python:rnn_cell",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "scatter_ops_test",
|
|
size = "medium", # NOTE: This is not run by default.
|
|
srcs = ["scatter_ops_test.py"],
|
|
shard_count = 2,
|
|
tags = ["optonly"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "slice_op_test",
|
|
size = "medium",
|
|
srcs = ["slice_op_test.py"],
|
|
tags = ["no_windows"], # b/126916429
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "huge_slice_op_test",
|
|
size = "medium",
|
|
srcs = ["huge_slice_op_test.py"],
|
|
tags = [
|
|
"no_oss", # Requires 4GB+ RAM
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "sparse_matmul_op_test",
|
|
size = "medium",
|
|
srcs = ["sparse_matmul_op_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "sparse_ops_test",
|
|
size = "medium",
|
|
srcs = ["sparse_ops_test.py"],
|
|
shard_count = 5,
|
|
tags = [
|
|
"noasan",
|
|
"optonly", # b/77589990
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:platform_test",
|
|
"//tensorflow/python:sparse_grad",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "sparse_tensor_dense_matmul_op_test",
|
|
size = "medium",
|
|
srcs = ["sparse_tensor_dense_matmul_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
# TODO(gpapan): Revisit the gradient of extract_image_patches_op to resolve
|
|
# http://b/31080670.
|
|
cuda_py_test(
|
|
name = "extract_image_patches_grad_test",
|
|
size = "medium",
|
|
srcs = ["extract_image_patches_grad_test.py"],
|
|
shard_count = 3,
|
|
tags = ["notap"], # http://b/31080670
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "extract_volume_patches_grad_test",
|
|
size = "medium",
|
|
srcs = ["extract_volume_patches_grad_test.py"],
|
|
tags = [
|
|
"no_pip",
|
|
"nomac", # http://b/139946976
|
|
"notap", # http://b/31080670
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "stage_op_test",
|
|
size = "medium",
|
|
srcs = ["stage_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:util",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "map_stage_op_test",
|
|
size = "medium",
|
|
srcs = ["map_stage_op_test.py"],
|
|
tags = ["no_oss"], # b/124474135
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:util",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "concat_op_test",
|
|
size = "medium",
|
|
srcs = ["concat_op_test.py"],
|
|
tags = ["no_windows"], # b/126916429
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "large_concat_op_test",
|
|
size = "medium",
|
|
srcs = ["large_concat_op_test.py"],
|
|
tags = [
|
|
"nomsan",
|
|
"notsan",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "conv_ops_3d_test",
|
|
size = "medium",
|
|
srcs = ["conv_ops_3d_test.py"],
|
|
shard_count = 30,
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:nn_ops",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cwise_ops_test",
|
|
size = "medium",
|
|
srcs = ["cwise_ops_test.py"],
|
|
shard_count = 50,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:math_ops_gen",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cwise_ops_binary_test",
|
|
size = "medium",
|
|
srcs = ["cwise_ops_binary_test.py"],
|
|
shard_count = 50,
|
|
# b/140155647: Error just outside of tolerance
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:math_ops_gen",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cwise_ops_unary_test",
|
|
size = "medium",
|
|
srcs = ["cwise_ops_unary_test.py"],
|
|
shard_count = 50,
|
|
# b/140155706: nans in result
|
|
xla_enable_strict_auto_jit = False,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:math_ops_gen",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "embedding_ops_test",
|
|
size = "medium",
|
|
srcs = ["embedding_ops_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:embedding_ops",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:partitioned_variables",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:state_ops",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "linalg_grad_test",
|
|
size = "medium",
|
|
srcs = ["linalg_grad_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python/ops/linalg",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "matrix_band_part_op_test",
|
|
size = "medium",
|
|
srcs = ["matrix_band_part_op_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "eig_op_test",
|
|
size = "medium",
|
|
srcs = ["eig_op_test.py"],
|
|
data = ["//tensorflow/python/kernel_tests/testdata:self_adjoint_eig_op_test_files"],
|
|
shard_count = 20,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
# b/127344411: xla_enable_strict_auto_jit = True,
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "self_adjoint_eig_op_test",
|
|
size = "medium",
|
|
srcs = ["self_adjoint_eig_op_test.py"],
|
|
data = ["//tensorflow/python/kernel_tests/testdata:self_adjoint_eig_op_test_files"],
|
|
shard_count = 20,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
# TODO(b/127344411): This test passes because XLA does not actually cluster
|
|
# the self_adjoint_eig op.
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "qr_op_test",
|
|
size = "medium",
|
|
srcs = ["qr_op_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "svd_op_test",
|
|
size = "medium",
|
|
srcs = ["svd_op_test.py"],
|
|
shard_count = 30,
|
|
tags = [
|
|
"no_oss", # b/117185141.
|
|
"nomsan", # TODO(b/117236102): Re-enable in msan build.
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:gradients_impl",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
# TODO(b/127344411): This test passes because XLA does not actually cluster
|
|
# the svd op.
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "norm_op_test",
|
|
size = "medium",
|
|
srcs = ["norm_op_test.py"],
|
|
shard_count = 20,
|
|
# TODO(b/117236102): Re-enable in msan build.
|
|
tags = [
|
|
"no_windows_gpu",
|
|
"nomsan",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "normalize_op_test",
|
|
size = "medium",
|
|
srcs = ["normalize_op_test.py"],
|
|
shard_count = 20,
|
|
# TODO(b/117236102): Re-enable in msan build.
|
|
tags = [
|
|
"no_windows_gpu",
|
|
"nomsan",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:nn",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "tensordot_op_test",
|
|
size = "medium",
|
|
srcs = ["tensordot_op_test.py"],
|
|
shard_count = 20,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
sycl_py_test(
|
|
name = "basic_gpu_test",
|
|
size = "small",
|
|
srcs = ["basic_gpu_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops_gen",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:math_ops_gen",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sets_test",
|
|
size = "medium",
|
|
srcs = ["sets_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform_test",
|
|
"//tensorflow/python:sets",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "weights_broadcast_test",
|
|
size = "small",
|
|
srcs = ["weights_broadcast_test.py"],
|
|
shard_count = 3,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:weights_broadcast_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "metrics_test",
|
|
size = "medium",
|
|
srcs = ["metrics_test.py"],
|
|
shard_count = 20,
|
|
tags = ["no_windows_gpu"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:data_flow_grad",
|
|
"//tensorflow/python:data_flow_ops",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:metrics",
|
|
"//tensorflow/python:nn_grad",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:variables",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "confusion_matrix_test",
|
|
size = "small",
|
|
srcs = ["confusion_matrix_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:confusion_matrix",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "bucketize_op_test",
|
|
size = "medium",
|
|
srcs = ["bucketize_op_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:math_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "sparse_cross_op_test",
|
|
size = "small",
|
|
srcs = ["sparse_cross_op_test.py"],
|
|
tags = ["no_windows"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:sparse_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "garbage_collection_test",
|
|
size = "small",
|
|
srcs = ["garbage_collection_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python/eager:context",
|
|
],
|
|
)
|
|
|
|
# Custom op tests
|
|
tf_custom_op_library(
|
|
name = "ackermann_op.so",
|
|
srcs = ["ackermann_op.cc"],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "ackermann_test",
|
|
size = "small",
|
|
srcs = ["ackermann_test.py"],
|
|
data = [":ackermann_op.so"],
|
|
tags = [
|
|
"no_pip",
|
|
"notap",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
)
|
|
|
|
tf_custom_op_library(
|
|
name = "duplicate_op.so",
|
|
srcs = ["duplicate_op.cc"],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "duplicate_op_test",
|
|
size = "small",
|
|
srcs = ["duplicate_op_test.py"],
|
|
data = [":duplicate_op.so"],
|
|
tags = [
|
|
"no_pip",
|
|
"notap",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
)
|
|
|
|
tf_custom_op_library(
|
|
name = "invalid_op.so",
|
|
srcs = ["invalid_op.cc"],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "invalid_op_test",
|
|
size = "small",
|
|
srcs = ["invalid_op_test.py"],
|
|
data = [":invalid_op.so"],
|
|
tags = [
|
|
"no_pip",
|
|
"notap",
|
|
],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:errors",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "cond_v2_test",
|
|
size = "medium",
|
|
srcs = ["cond_v2_test.py"],
|
|
grpc_enabled = True,
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:cond_v2",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python:test_ops",
|
|
"//tensorflow/python:training",
|
|
"//tensorflow/python:while_v2",
|
|
"//tensorflow/python/compat",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "while_v2_test",
|
|
size = "medium",
|
|
srcs = ["while_v2_test.py"],
|
|
grpc_enabled = True,
|
|
deps = [
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:control_flow_util",
|
|
"//tensorflow/python:control_flow_v2_toggles",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:framework",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:functional_ops",
|
|
"//tensorflow/python:gradients_impl",
|
|
"//tensorflow/python:list_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:tensor_array_grad",
|
|
"//tensorflow/python:tf_optimizer",
|
|
"//tensorflow/python:while_v2",
|
|
"//tensorflow/python/eager:def_function",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "critical_section_test",
|
|
size = "medium",
|
|
srcs = ["critical_section_test.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:control_flow_ops",
|
|
"//tensorflow/python:control_flow_v2_toggles",
|
|
"//tensorflow/python:critical_section_ops",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:framework_test_lib",
|
|
"//tensorflow/python:gradients",
|
|
"//tensorflow/python:platform_test",
|
|
"//tensorflow/python:resource_variable_ops",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python/data/experimental/ops:prefetching_ops",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "tridiagonal_matmul_op_test",
|
|
size = "medium",
|
|
srcs = ["tridiagonal_matmul_op_test.py"],
|
|
shard_count = 10,
|
|
tags = ["no_rocm"],
|
|
deps = [
|
|
"//tensorflow/python:client_testlib",
|
|
"//tensorflow/python:framework_for_generated_wrappers",
|
|
"//tensorflow/python:linalg_ops",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|