n/a
PiperOrigin-RevId: 248282244
This commit is contained in:
parent
d8070bb0ec
commit
0d2442c867
tensorflow
compiler/tests
examples
adding_an_op
get_started/regression
how_tos/reading_data
speech_commands
tf2_showcase
tutorials/word2vec
lite
python
tools/build_info
@ -65,6 +65,7 @@ py_test(
|
||||
name = "xla_test_test",
|
||||
size = "small",
|
||||
srcs = ["xla_test_test.py"],
|
||||
python_version = "PY2",
|
||||
deps = [
|
||||
":xla_test",
|
||||
],
|
||||
|
@ -68,6 +68,7 @@ py_test(
|
||||
name = "zero_out_1_test",
|
||||
size = "small",
|
||||
srcs = ["zero_out_1_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_pip",
|
||||
@ -83,6 +84,7 @@ py_test(
|
||||
name = "zero_out_2_test",
|
||||
size = "small",
|
||||
srcs = ["zero_out_2_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_pip",
|
||||
@ -99,6 +101,7 @@ py_test(
|
||||
name = "zero_out_3_test",
|
||||
size = "small",
|
||||
srcs = ["zero_out_3_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_pip",
|
||||
@ -129,6 +132,7 @@ py_test(
|
||||
size = "small",
|
||||
srcs = ["cuda_op_test.py"],
|
||||
exec_compatible_with = tf_exec_compatible_with({"tags": tf_cuda_tests_tags()}),
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = tf_cuda_tests_tags() + [
|
||||
"notap",
|
||||
@ -144,6 +148,7 @@ py_test(
|
||||
name = "fact_test",
|
||||
size = "small",
|
||||
srcs = ["fact_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = ["//tensorflow:tensorflow_py"],
|
||||
)
|
||||
|
@ -12,6 +12,7 @@ py_test(
|
||||
"linear_regression_categorical.py",
|
||||
"test.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"manual",
|
||||
|
@ -10,6 +10,7 @@ exports_files(["LICENSE"])
|
||||
py_binary(
|
||||
name = "convert_to_records",
|
||||
srcs = ["convert_to_records.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow:tensorflow_py",
|
||||
@ -22,6 +23,7 @@ py_binary(
|
||||
srcs = [
|
||||
"fully_connected_reader.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow:tensorflow_py",
|
||||
|
@ -70,6 +70,7 @@ tf_py_test(
|
||||
py_binary(
|
||||
name = "train",
|
||||
srcs = ["train.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":train_main_lib"],
|
||||
)
|
||||
@ -94,7 +95,7 @@ tf_py_test(
|
||||
size = "small",
|
||||
srcs = ["train_test.py"],
|
||||
additional_deps = [
|
||||
":train",
|
||||
":train_main_lib",
|
||||
"//tensorflow/python:client_testlib",
|
||||
],
|
||||
tags = [
|
||||
@ -105,6 +106,7 @@ tf_py_test(
|
||||
py_binary(
|
||||
name = "freeze",
|
||||
srcs = ["freeze.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":freeze_main_lib"],
|
||||
)
|
||||
@ -150,6 +152,7 @@ tf_py_test(
|
||||
py_binary(
|
||||
name = "wav_to_features",
|
||||
srcs = ["wav_to_features.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":wav_to_features_main_lib"],
|
||||
)
|
||||
@ -192,6 +195,7 @@ tf_py_test(
|
||||
py_binary(
|
||||
name = "generate_streaming_test_wav",
|
||||
srcs = ["generate_streaming_test_wav.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":generate_streaming_test_wav_main_lib"],
|
||||
)
|
||||
@ -249,6 +253,7 @@ tf_cc_binary(
|
||||
py_binary(
|
||||
name = "label_wav",
|
||||
srcs = ["label_wav.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":label_wav_main_lib"],
|
||||
)
|
||||
|
@ -19,6 +19,7 @@ test_suite(
|
||||
py_test(
|
||||
name = "mnist",
|
||||
srcs = ["mnist.py"],
|
||||
python_version = "PY2",
|
||||
tags = [
|
||||
"manual",
|
||||
"no_oss",
|
||||
|
@ -12,6 +12,7 @@ py_binary(
|
||||
srcs = [
|
||||
"word2vec_basic.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no-internal-py3",
|
||||
|
@ -6,6 +6,7 @@ py_binary(
|
||||
name = "label_image",
|
||||
srcs = ["label_image.py"],
|
||||
main = "label_image.py",
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/lite/python:lite",
|
||||
|
@ -18,6 +18,7 @@ py_library(
|
||||
py_test(
|
||||
name = "ops_util_test",
|
||||
srcs = ["ops_util_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ops_util",
|
||||
|
@ -10,6 +10,7 @@ load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite")
|
||||
py_binary(
|
||||
name = "upgrade_schema",
|
||||
srcs = ["upgrade_schema.py"],
|
||||
python_version = "PY2",
|
||||
deps = [":upgrade_schema_main_lib"],
|
||||
)
|
||||
|
||||
@ -35,6 +36,7 @@ py_test(
|
||||
name = "upgrade_schema_test",
|
||||
size = "small",
|
||||
srcs = ["upgrade_schema_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"manual",
|
||||
|
@ -18,6 +18,7 @@ py_binary(
|
||||
"//tensorflow/python:platform",
|
||||
"@flatbuffers//:flatc",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
)
|
||||
|
||||
|
@ -14,6 +14,7 @@ py_binary(
|
||||
"dataset.py",
|
||||
"mnist_tflite.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
deps = [
|
||||
"//tensorflow:tensorflow_py",
|
||||
],
|
||||
|
@ -1130,6 +1130,7 @@ py_test(
|
||||
name = "framework_composite_tensor_test",
|
||||
srcs = ["framework/composite_tensor_test.py"],
|
||||
main = "framework/composite_tensor_test.py",
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":composite_tensor",
|
||||
@ -2027,6 +2028,7 @@ py_test(
|
||||
name = "batch_ops_test",
|
||||
size = "small",
|
||||
srcs = ["ops/batch_ops_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"manual",
|
||||
@ -2910,6 +2912,7 @@ py_test(
|
||||
name = "loss_scale_optimizer_test",
|
||||
size = "small",
|
||||
srcs = ["training/experimental/loss_scale_optimizer_test.py"],
|
||||
python_version = "PY2",
|
||||
deps = [
|
||||
":loss_scale_optimizer",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -2924,6 +2927,7 @@ py_test(
|
||||
name = "loss_scale_test",
|
||||
size = "medium",
|
||||
srcs = ["training/experimental/loss_scale_test.py"],
|
||||
python_version = "PY2",
|
||||
deps = [
|
||||
":loss_scale",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -6345,6 +6349,7 @@ py_binary(
|
||||
srcs = [
|
||||
"grappler/cost_analyzer_tool.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":cost_analyzer",
|
||||
@ -6509,6 +6514,7 @@ py_binary(
|
||||
srcs = [
|
||||
"grappler/graph_analyzer.py",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":framework_for_generated_wrappers",
|
||||
|
@ -10,6 +10,7 @@ load("//tensorflow:tensorflow.bzl", "py_test")
|
||||
py_test(
|
||||
name = "autotune_benchmark",
|
||||
srcs = ["autotune_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -24,6 +25,7 @@ py_test(
|
||||
py_test(
|
||||
name = "csv_dataset_benchmark",
|
||||
srcs = ["csv_dataset_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = ["no_pip"],
|
||||
deps = [
|
||||
@ -41,6 +43,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_and_batch_benchmark",
|
||||
srcs = ["map_and_batch_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/core:protos_all_py",
|
||||
@ -60,6 +63,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_defun_benchmark",
|
||||
srcs = ["map_defun_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
@ -76,6 +80,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_vectorization_benchmark",
|
||||
srcs = ["map_vectorization_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/core:protos_all_py",
|
||||
@ -96,6 +101,7 @@ py_test(
|
||||
name = "matching_files_benchmark",
|
||||
size = "small",
|
||||
srcs = ["matching_files_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
@ -112,6 +118,7 @@ py_test(
|
||||
py_test(
|
||||
name = "choose_fastest_benchmark",
|
||||
srcs = ["choose_fastest_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -126,6 +133,7 @@ py_test(
|
||||
py_test(
|
||||
name = "choose_fastest_branch_benchmark",
|
||||
srcs = ["choose_fastest_branch_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -142,6 +150,7 @@ py_test(
|
||||
py_test(
|
||||
name = "optimize_benchmark",
|
||||
srcs = ["optimize_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -156,6 +165,7 @@ py_test(
|
||||
py_test(
|
||||
name = "parallel_interleave_benchmark",
|
||||
srcs = ["parallel_interleave_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -172,6 +182,7 @@ py_test(
|
||||
py_test(
|
||||
name = "rejection_resample_benchmark",
|
||||
srcs = ["rejection_resample_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = ["no_pip"],
|
||||
deps = [
|
||||
@ -205,6 +216,7 @@ py_test(
|
||||
py_test(
|
||||
name = "unbatch_benchmark",
|
||||
srcs = ["unbatch_benchmark.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
|
@ -10,6 +10,7 @@ py_test(
|
||||
name = "assert_next_dataset_test",
|
||||
size = "medium",
|
||||
srcs = ["assert_next_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -29,6 +30,7 @@ py_test(
|
||||
name = "filter_fusion_test",
|
||||
size = "medium",
|
||||
srcs = ["filter_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -53,6 +55,7 @@ py_test(
|
||||
name = "filter_with_random_uniform_fusion_test",
|
||||
size = "medium",
|
||||
srcs = ["filter_with_random_uniform_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"manual",
|
||||
@ -79,6 +82,7 @@ py_test(
|
||||
name = "hoist_random_uniform_test",
|
||||
size = "small",
|
||||
srcs = ["hoist_random_uniform_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -106,6 +110,7 @@ py_test(
|
||||
name = "latency_all_edges_test",
|
||||
size = "small",
|
||||
srcs = ["latency_all_edges_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -126,6 +131,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_and_batch_fusion_test",
|
||||
srcs = ["map_and_batch_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -145,6 +151,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_and_filter_fusion_test",
|
||||
srcs = ["map_and_filter_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -168,6 +175,7 @@ py_test(
|
||||
py_test(
|
||||
name = "map_fusion_test",
|
||||
srcs = ["map_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -189,6 +197,7 @@ py_test(
|
||||
name = "map_parallelization_test",
|
||||
size = "small",
|
||||
srcs = ["map_parallelization_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -216,6 +225,7 @@ py_test(
|
||||
name = "map_vectorization_test",
|
||||
size = "small",
|
||||
srcs = ["map_vectorization_test.py"],
|
||||
python_version = "PY2",
|
||||
shard_count = 8,
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
@ -254,6 +264,7 @@ py_test(
|
||||
name = "choose_fastest_dataset_test",
|
||||
size = "small",
|
||||
srcs = ["choose_fastest_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -274,6 +285,7 @@ py_test(
|
||||
name = "choose_fastest_branch_dataset_test",
|
||||
size = "small",
|
||||
srcs = ["choose_fastest_branch_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -297,6 +309,7 @@ py_test(
|
||||
name = "model_dataset_test",
|
||||
size = "medium",
|
||||
srcs = ["model_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -318,6 +331,7 @@ py_test(
|
||||
name = "noop_elimination_test",
|
||||
size = "small",
|
||||
srcs = ["noop_elimination_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -340,6 +354,7 @@ py_test(
|
||||
name = "optimize_dataset_test",
|
||||
size = "medium",
|
||||
srcs = ["optimize_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
@ -369,6 +384,7 @@ py_test(
|
||||
py_test(
|
||||
name = "shuffle_and_repeat_fusion_test",
|
||||
srcs = ["shuffle_and_repeat_fusion_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_oss",
|
||||
|
@ -20,6 +20,7 @@ py_test(
|
||||
name = "nest_test",
|
||||
size = "small",
|
||||
srcs = ["nest_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":nest",
|
||||
@ -52,6 +53,7 @@ py_test(
|
||||
name = "sparse_test",
|
||||
size = "small",
|
||||
srcs = ["sparse_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":nest",
|
||||
@ -86,6 +88,7 @@ py_test(
|
||||
name = "structure_test",
|
||||
size = "small",
|
||||
srcs = ["structure_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":nest",
|
||||
@ -114,6 +117,7 @@ py_test(
|
||||
name = "options_test",
|
||||
size = "small",
|
||||
srcs = ["options_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":options",
|
||||
@ -137,6 +141,7 @@ py_test(
|
||||
name = "convert_test",
|
||||
size = "small",
|
||||
srcs = ["convert_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":convert",
|
||||
@ -161,6 +166,7 @@ py_test(
|
||||
name = "random_seed_test",
|
||||
size = "small",
|
||||
srcs = ["random_seed_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":random_seed",
|
||||
@ -182,6 +188,7 @@ py_test(
|
||||
name = "traverse_test",
|
||||
size = "small",
|
||||
srcs = ["traverse_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":traverse",
|
||||
|
@ -154,6 +154,7 @@ py_test(
|
||||
name = "distribute_lib_test",
|
||||
size = "small",
|
||||
srcs = ["distribute_lib_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":distribute_lib",
|
||||
@ -198,6 +199,7 @@ py_library(
|
||||
py_test(
|
||||
name = "distribute_coordinator_test",
|
||||
srcs = ["distribute_coordinator_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_pip", # b/131691139
|
||||
@ -364,6 +366,7 @@ py_test(
|
||||
name = "numpy_dataset_test",
|
||||
size = "small",
|
||||
srcs = ["numpy_dataset_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":numpy_dataset",
|
||||
@ -418,6 +421,7 @@ cuda_py_test(
|
||||
py_test(
|
||||
name = "multi_worker_util_test",
|
||||
srcs = ["multi_worker_util_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":multi_worker_util",
|
||||
@ -493,6 +497,7 @@ py_library(
|
||||
py_test(
|
||||
name = "shared_variable_creator_test",
|
||||
srcs = ["shared_variable_creator_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":shared_variable_creator",
|
||||
@ -575,6 +580,7 @@ py_library(
|
||||
py_test(
|
||||
name = "combinations_test",
|
||||
srcs = ["combinations_test.py"],
|
||||
python_version = "PY2",
|
||||
deps = [
|
||||
":combinations",
|
||||
"//tensorflow/python/eager:test",
|
||||
|
@ -469,6 +469,7 @@ py_test(
|
||||
name = "ragged_tensor_test",
|
||||
size = "medium",
|
||||
srcs = ["ragged_tensor_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_windows",
|
||||
@ -497,6 +498,7 @@ py_test(
|
||||
name = "ragged_eager_test",
|
||||
size = "medium",
|
||||
srcs = ["ragged_eager_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -510,6 +512,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_range_op_test",
|
||||
srcs = ["ragged_range_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_math_ops",
|
||||
@ -523,6 +526,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_tensor_bounding_shape_op_test",
|
||||
srcs = ["ragged_tensor_bounding_shape_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -536,6 +540,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_row_lengths_op_test",
|
||||
srcs = ["ragged_row_lengths_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -551,6 +556,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_gather_op_test",
|
||||
srcs = ["ragged_gather_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -571,6 +577,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_batch_gather_op_test",
|
||||
srcs = ["ragged_batch_gather_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -592,6 +599,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_gather_nd_op_test",
|
||||
srcs = ["ragged_gather_nd_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -611,6 +619,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_row_splits_to_segment_ids_op_test",
|
||||
srcs = ["ragged_row_splits_to_segment_ids_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_test_util",
|
||||
@ -624,6 +633,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_segment_ids_to_row_splits_op_test",
|
||||
srcs = ["ragged_segment_ids_to_row_splits_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_test_util",
|
||||
@ -637,6 +647,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_from_tensor_op_test",
|
||||
srcs = ["ragged_from_tensor_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_string_ops",
|
||||
@ -656,6 +667,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_to_sparse_op_test",
|
||||
srcs = ["ragged_to_sparse_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_windows",
|
||||
@ -680,6 +692,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_from_sparse_op_test",
|
||||
srcs = ["ragged_from_sparse_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_tensor",
|
||||
@ -697,6 +710,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_to_tensor_op_test",
|
||||
srcs = ["ragged_to_tensor_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -712,6 +726,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_segment_op_test",
|
||||
srcs = ["ragged_segment_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -730,6 +745,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_reduce_op_test",
|
||||
srcs = ["ragged_reduce_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -749,6 +765,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_map_flat_values_op_test",
|
||||
srcs = ["ragged_map_flat_values_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -768,6 +785,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_const_op_test",
|
||||
srcs = ["ragged_const_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged",
|
||||
@ -784,6 +802,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_constant_value_op_test",
|
||||
srcs = ["ragged_constant_value_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_windows",
|
||||
@ -803,6 +822,7 @@ py_test(
|
||||
py_test(
|
||||
name = "convert_to_tensor_or_ragged_tensor_op_test",
|
||||
srcs = ["convert_to_tensor_or_ragged_tensor_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -820,6 +840,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_boolean_mask_op_test",
|
||||
srcs = ["ragged_boolean_mask_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -838,6 +859,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_concat_op_test",
|
||||
srcs = ["ragged_concat_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_concat_ops",
|
||||
@ -857,6 +879,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_stack_op_test",
|
||||
srcs = ["ragged_stack_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_concat_ops",
|
||||
@ -872,6 +895,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_rank_op_test",
|
||||
srcs = ["ragged_rank_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -885,6 +909,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_tile_op_test",
|
||||
srcs = ["ragged_tile_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -902,6 +927,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_util_test",
|
||||
srcs = ["ragged_util_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_test_util",
|
||||
@ -918,6 +944,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_expand_dims_op_test",
|
||||
srcs = ["ragged_expand_dims_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -932,6 +959,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_where_op_test",
|
||||
srcs = ["ragged_where_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_factory_ops",
|
||||
@ -946,6 +974,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_dispatch_test",
|
||||
srcs = ["ragged_dispatch_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged", # fixdeps: keep
|
||||
@ -972,6 +1001,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_operators_test",
|
||||
srcs = ["ragged_operators_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged", # fixdeps: keep
|
||||
@ -986,6 +1016,7 @@ py_test(
|
||||
name = "ragged_map_fn_op_test",
|
||||
size = "small",
|
||||
srcs = ["ragged_map_fn_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged", # fixdeps: keep
|
||||
@ -1010,6 +1041,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_tensor_shape_test",
|
||||
srcs = ["ragged_tensor_shape_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged", # fixdeps: keep
|
||||
@ -1028,6 +1060,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_size_op_test",
|
||||
srcs = ["ragged_size_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_array_ops",
|
||||
@ -1043,6 +1076,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_placeholder_op_test",
|
||||
srcs = ["ragged_placeholder_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_math_ops",
|
||||
@ -1056,6 +1090,7 @@ py_test(
|
||||
py_test(
|
||||
name = "ragged_squeeze_op_test",
|
||||
srcs = ["ragged_squeeze_op_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":ragged_conversion_ops",
|
||||
|
@ -40,6 +40,7 @@ py_library(
|
||||
py_test(
|
||||
name = "print_model_analysis_test",
|
||||
srcs = ["print_model_analysis_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
"//tensorflow/core/profiler:protos_all_py",
|
||||
|
@ -46,6 +46,7 @@ py_test(
|
||||
name = "saved_model_utils_test",
|
||||
size = "small",
|
||||
srcs = ["saved_model_utils_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = ["no_windows"], # TODO: needs investigation on Windows
|
||||
visibility = ["//visibility:private"],
|
||||
@ -78,6 +79,7 @@ py_library(
|
||||
py_binary(
|
||||
name = "freeze_graph",
|
||||
srcs = ["freeze_graph.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":freeze_graph_main_lib"],
|
||||
)
|
||||
@ -94,6 +96,7 @@ py_library(
|
||||
py_binary(
|
||||
name = "import_pb_to_tensorboard",
|
||||
srcs = ["import_pb_to_tensorboard.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":import_pb_to_tensorboard_lib"],
|
||||
)
|
||||
@ -117,6 +120,7 @@ py_test(
|
||||
name = "freeze_graph_test",
|
||||
size = "small",
|
||||
srcs = ["freeze_graph_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":freeze_graph_lib",
|
||||
@ -135,6 +139,7 @@ py_test(
|
||||
py_binary(
|
||||
name = "inspect_checkpoint",
|
||||
srcs = ["inspect_checkpoint.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":inspect_checkpoint_lib"],
|
||||
)
|
||||
@ -170,6 +175,7 @@ py_library(
|
||||
py_binary(
|
||||
name = "strip_unused",
|
||||
srcs = ["strip_unused.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":strip_unused_lib",
|
||||
@ -183,6 +189,7 @@ py_test(
|
||||
name = "strip_unused_test",
|
||||
size = "small",
|
||||
srcs = ["strip_unused_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = ["notap"],
|
||||
deps = [
|
||||
@ -215,6 +222,7 @@ py_library(
|
||||
py_binary(
|
||||
name = "optimize_for_inference",
|
||||
srcs = ["optimize_for_inference.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":optimize_for_inference_main_lib"],
|
||||
)
|
||||
@ -238,6 +246,7 @@ py_test(
|
||||
name = "optimize_for_inference_test",
|
||||
size = "small",
|
||||
srcs = ["optimize_for_inference_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":optimize_for_inference_lib",
|
||||
@ -268,6 +277,7 @@ py_library(
|
||||
py_binary(
|
||||
name = "print_selective_registration_header",
|
||||
srcs = ["print_selective_registration_header.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [":print_selective_registration_header_lib"],
|
||||
@ -287,6 +297,7 @@ py_library(
|
||||
py_test(
|
||||
name = "print_selective_registration_header_test",
|
||||
srcs = ["print_selective_registration_header_test.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":selective_registration_header_lib",
|
||||
@ -298,6 +309,7 @@ py_test(
|
||||
py_binary(
|
||||
name = "saved_model_cli",
|
||||
srcs = ["saved_model_cli.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [":saved_model_cli_lib"],
|
||||
)
|
||||
@ -319,6 +331,7 @@ py_test(
|
||||
data = [
|
||||
"//tensorflow/cc/saved_model:saved_model_half_plus_two",
|
||||
],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"manual",
|
||||
|
@ -8,5 +8,6 @@ licenses(["notice"]) # Apache 2.0
|
||||
py_binary(
|
||||
name = "gen_build_info",
|
||||
srcs = ["gen_build_info.py"],
|
||||
python_version = "PY2",
|
||||
srcs_version = "PY2AND3",
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user