experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

PY2 removal cleanup

Browse Source 
PiperOrigin-RevId: 352106691
Change-Id: I382d53c64f0d29da430b8cb6d2395a2cb281509e
This commit is contained in:
A. Unique TensorFlower 2021-01-15 16:34:55 -08:00 committed by TensorFlower Gardener
parent 31aae57d4f
commit 034633f23b
115 changed files with 823 additions and 694 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/BUILD
View File

@ -1088,7 +1088,7 @@ gen_api_init_files(
py_library( py_library(
name = "tensorflow_py", name = "tensorflow_py",
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = select({ deps = select({
"api_version_2": [], "api_version_2": [],
@ -1110,7 +1110,7 @@ py_library(
"//tensorflow/python/keras/api:keras_python_api_gen_compat_v1", "//tensorflow/python/keras/api:keras_python_api_gen_compat_v1",
"//tensorflow/python/keras/api:keras_python_api_gen_compat_v2", "//tensorflow/python/keras/api:keras_python_api_gen_compat_v2",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = ["//tensorflow/python:no_contrib"], deps = ["//tensorflow/python:no_contrib"],
) )

1
tensorflow/compiler/jit/ops/BUILD
View File

@ -22,5 +22,6 @@ tf_gen_op_wrapper_py(
py_library( py_library(
name = "xla_ops_grad", name = "xla_ops_grad",
srcs = ["xla_ops_grad.py"], srcs = ["xla_ops_grad.py"],
srcs_version = "PY3",
deps = ["//tensorflow/python:framework_ops"], deps = ["//tensorflow/python:framework_ops"],
) )

4
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "common", name = "common",
srcs = ["common.py"], srcs = ["common.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
], ],
@ -17,7 +17,7 @@ py_library(
py_library( py_library(
name = "common_v1", name = "common_v1",
srcs = ["common_v1.py"], srcs = ["common_v1.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
], ],

1
tensorflow/compiler/mlir/tfr/examples/mnist/BUILD
View File

@ -49,6 +49,7 @@ py_library(
name = "mnist_train", name = "mnist_train",
srcs = ["mnist_train.py"], srcs = ["mnist_train.py"],
data = [":mnist_ops_mlir"], data = [":mnist_ops_mlir"],
srcs_version = "PY3",
deps = [ deps = [
":mnist_ops", ":mnist_ops",
":mnist_ops_py", ":mnist_ops_py",

6
tensorflow/compiler/tests/BUILD
View File

@ -41,7 +41,7 @@ py_library(
name = "xla_test", name = "xla_test",
testonly = 1, testonly = 1,
srcs = ["xla_test.py"], srcs = ["xla_test.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [":friends"], visibility = [":friends"],
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -65,7 +65,7 @@ py_library(
"__init__.py", "__init__.py",
"test_utils.py", "test_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//third_party/py/numpy", "//third_party/py/numpy",
"@six_archive//:six", "@six_archive//:six",
@ -1729,7 +1729,7 @@ py_library(
name = "lstm", name = "lstm",
testonly = 1, testonly = 1,
srcs = ["lstm.py"], srcs = ["lstm.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework", "//tensorflow/python:framework",

1
tensorflow/compiler/xla/experimental/xla_sharding/BUILD
View File

@ -9,6 +9,7 @@ package(
py_library( py_library(
name = "xla_sharding", name = "xla_sharding",
srcs = ["xla_sharding.py"], srcs = ["xla_sharding.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/compiler/tf2xla/python:xla", "//tensorflow/compiler/tf2xla/python:xla",

2
tensorflow/compiler/xla/python/tpu_driver/client/BUILD
View File

@ -76,7 +76,7 @@ py_library(
srcs = [ srcs = [
"tpu_client.py", "tpu_client.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":tpu_client_extension", ":tpu_client_extension",

3
tensorflow/compiler/xla/python_api/BUILD
View File

@ -9,6 +9,7 @@ package(
py_library( py_library(
name = "types", name = "types",
srcs = ["types.py"], srcs = ["types.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/compiler/xla:xla_data_proto_py", "//tensorflow/compiler/xla:xla_data_proto_py",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -20,6 +21,7 @@ py_library(
py_library( py_library(
name = "xla_shape", name = "xla_shape",
srcs = ["xla_shape.py"], srcs = ["xla_shape.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":types", ":types",
@ -30,6 +32,7 @@ py_library(
py_library( py_library(
name = "xla_literal", name = "xla_literal",
srcs = ["xla_literal.py"], srcs = ["xla_literal.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":types", ":types",

10
tensorflow/examples/adding_an_op/BUILD
View File

@ -23,7 +23,7 @@ py_library(
name = "zero_out_op_1", name = "zero_out_op_1",
srcs = ["zero_out_op_1.py"], srcs = ["zero_out_op_1.py"],
data = [":zero_out_op_kernel_1.so"], data = [":zero_out_op_kernel_1.so"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow:tensorflow_py"], deps = ["//tensorflow:tensorflow_py"],
) )
@ -36,7 +36,7 @@ py_library(
name = "zero_out_op_2", name = "zero_out_op_2",
srcs = ["zero_out_op_2.py"], srcs = ["zero_out_op_2.py"],
data = [":zero_out_op_kernel_2.so"], data = [":zero_out_op_kernel_2.so"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow:tensorflow_py"], deps = ["//tensorflow:tensorflow_py"],
) )
@ -49,14 +49,14 @@ py_library(
name = "zero_out_op_3", name = "zero_out_op_3",
srcs = ["zero_out_op_3.py"], srcs = ["zero_out_op_3.py"],
data = [":zero_out_op_kernel_3.so"], data = [":zero_out_op_kernel_3.so"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow:tensorflow_py"], deps = ["//tensorflow:tensorflow_py"],
) )
py_library( py_library(
name = "zero_out_grad_2", name = "zero_out_grad_2",
srcs = ["zero_out_grad_2.py"], srcs = ["zero_out_grad_2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:framework_for_generated_wrappers",
@ -123,7 +123,7 @@ py_library(
name = "cuda_op", name = "cuda_op",
srcs = ["cuda_op.py"], srcs = ["cuda_op.py"],
data = [":cuda_op_kernel.so"], data = [":cuda_op_kernel.so"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow:tensorflow_py"], deps = ["//tensorflow:tensorflow_py"],
) )

24
tensorflow/examples/speech_commands/BUILD
View File

@ -52,7 +52,7 @@ py_library(
srcs = [ srcs = [
"models.py", "models.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -78,7 +78,7 @@ py_library(
srcs = [ srcs = [
"input_data.py", "input_data.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//tensorflow/lite/experimental/microfrontend:audio_microfrontend_py", "//tensorflow/lite/experimental/microfrontend:audio_microfrontend_py",
@ -115,7 +115,7 @@ py_library(
srcs = [ srcs = [
"train.py", "train.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":input_data", ":input_data",
":models", ":models",
@ -150,7 +150,7 @@ py_binary(
py_library( py_library(
name = "freeze_main_lib", name = "freeze_main_lib",
srcs = ["freeze.py"], srcs = ["freeze.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":freeze_lib"], deps = [":freeze_lib"],
) )
@ -159,7 +159,7 @@ py_library(
srcs = [ srcs = [
"freeze.py", "freeze.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
tags = [ tags = [
"no_pip", # b/131330719 "no_pip", # b/131330719
], ],
@ -195,7 +195,7 @@ py_binary(
py_library( py_library(
name = "wav_to_features_main_lib", name = "wav_to_features_main_lib",
srcs = ["wav_to_features.py"], srcs = ["wav_to_features.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":wav_to_features_lib"], deps = [":wav_to_features_lib"],
) )
@ -204,7 +204,7 @@ py_library(
srcs = [ srcs = [
"wav_to_features.py", "wav_to_features.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":input_data", ":input_data",
":models", ":models",
@ -237,7 +237,7 @@ py_binary(
py_library( py_library(
name = "generate_streaming_test_wav_main_lib", name = "generate_streaming_test_wav_main_lib",
srcs = ["generate_streaming_test_wav.py"], srcs = ["generate_streaming_test_wav.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":generate_streaming_test_wav_lib"], deps = [":generate_streaming_test_wav_lib"],
) )
@ -246,7 +246,7 @@ py_library(
srcs = [ srcs = [
"generate_streaming_test_wav.py", "generate_streaming_test_wav.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":input_data", ":input_data",
":models", ":models",
@ -295,7 +295,7 @@ py_binary(
py_library( py_library(
name = "label_wav_main_lib", name = "label_wav_main_lib",
srcs = ["label_wav.py"], srcs = ["label_wav.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":label_wav_lib"], deps = [":label_wav_lib"],
) )
@ -304,7 +304,7 @@ py_library(
srcs = [ srcs = [
"label_wav.py", "label_wav.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
], ],
@ -411,7 +411,7 @@ tf_cc_binary(
py_library( py_library(
name = "test_lib", name = "test_lib",
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":freeze", ":freeze",
":generate_streaming_test_wav", ":generate_streaming_test_wav",

8
tensorflow/lite/experimental/examples/lstm/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "input_data", name = "input_data",
srcs = ["input_data.py"], srcs = ["input_data.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:private"], visibility = ["//visibility:private"],
deps = [ deps = [
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -21,7 +21,7 @@ py_library(
py_library( py_library(
name = "rnn", name = "rnn",
srcs = ["rnn.py"], srcs = ["rnn.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/lite/python:op_hint", "//tensorflow/lite/python:op_hint",
@ -36,7 +36,7 @@ py_library(
py_library( py_library(
name = "rnn_cell", name = "rnn_cell",
srcs = ["rnn_cell.py"], srcs = ["rnn_cell.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/lite/python:op_hint", "//tensorflow/lite/python:op_hint",
@ -140,7 +140,7 @@ py_test(
py_library( py_library(
name = "tflite_lstm_ops", name = "tflite_lstm_ops",
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":rnn", ":rnn",

2
tensorflow/lite/experimental/tensorboard/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "ops_util", name = "ops_util",
srcs = ["ops_util.py"], srcs = ["ops_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/lite/python:wrap_toco", "//tensorflow/lite/python:wrap_toco",

2
tensorflow/lite/micro/examples/person_detection/utils/BUILD
View File

@ -16,7 +16,7 @@ py_binary(
py_library( py_library(
name = "raw_to_bitmap_lib", name = "raw_to_bitmap_lib",
srcs = ["raw_to_bitmap.py"], srcs = ["raw_to_bitmap.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//third_party/py/numpy", "//third_party/py/numpy",
], ],

28
tensorflow/lite/python/BUILD
View File

@ -19,7 +19,7 @@ py_library(
"interpreter.py", "interpreter.py",
], ],
compatible_with = get_compatible_with_portable(), compatible_with = get_compatible_with_portable(),
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/lite/python/interpreter_wrapper:_pywrap_tensorflow_interpreter_wrapper", "//tensorflow/lite/python/interpreter_wrapper:_pywrap_tensorflow_interpreter_wrapper",
@ -71,7 +71,7 @@ py_binary(
py_library( py_library(
name = "tflite_convert_main_lib", name = "tflite_convert_main_lib",
srcs = ["tflite_convert.py"], srcs = ["tflite_convert.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":tflite_convert_lib", ":tflite_convert_lib",
@ -83,7 +83,7 @@ py_library(
py_library( py_library(
name = "tflite_convert_lib", name = "tflite_convert_lib",
srcs = ["tflite_convert.py"], srcs = ["tflite_convert.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":lite", ":lite",
@ -98,7 +98,7 @@ py_library(
name = "test_util", name = "test_util",
testonly = 1, testonly = 1,
srcs = ["test_util.py"], srcs = ["test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":lite", ":lite",
":schema_util", ":schema_util",
@ -166,7 +166,7 @@ py_test(
py_library( py_library(
name = "lite", name = "lite",
srcs = ["lite.py"], srcs = ["lite.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":convert", ":convert",
@ -236,7 +236,7 @@ py_library(
name = "lite_v2_test_util", name = "lite_v2_test_util",
testonly = 1, testonly = 1,
srcs = ["lite_v2_test_util.py"], srcs = ["lite_v2_test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
tags = [ tags = [
"no_windows", "no_windows",
], ],
@ -265,7 +265,7 @@ py_test(
py_library( py_library(
name = "util", name = "util",
srcs = ["util.py"], srcs = ["util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow/lite:__subpackages__", "//tensorflow/lite:__subpackages__",
], ],
@ -321,7 +321,7 @@ py_library(
srcs = [ srcs = [
"tflite_keras_util.py", "tflite_keras_util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/eager:def_function", "//tensorflow/python/eager:def_function",
@ -333,7 +333,7 @@ py_library(
srcs = [ srcs = [
"wrap_toco.py", "wrap_toco.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:_pywrap_toco_api", "//tensorflow/python:_pywrap_toco_api",
"//tensorflow/python:pywrap_tensorflow", "//tensorflow/python:pywrap_tensorflow",
@ -344,7 +344,7 @@ py_library(
py_library( py_library(
name = "lite_constants", name = "lite_constants",
srcs = ["lite_constants.py"], srcs = ["lite_constants.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/lite/toco:toco_flags_proto_py", "//tensorflow/lite/toco:toco_flags_proto_py",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -354,7 +354,7 @@ py_library(
py_library( py_library(
name = "convert", name = "convert",
srcs = ["convert.py"], srcs = ["convert.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":lite_constants", ":lite_constants",
@ -372,7 +372,7 @@ py_library(
py_library( py_library(
name = "op_hint", name = "op_hint",
srcs = ["op_hint.py"], srcs = ["op_hint.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -402,7 +402,7 @@ py_test(
py_library( py_library(
name = "convert_saved_model", name = "convert_saved_model",
srcs = ["convert_saved_model.py"], srcs = ["convert_saved_model.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow/lite:__subpackages__", "//tensorflow/lite:__subpackages__",
], ],
@ -457,7 +457,7 @@ sh_test(
py_library( py_library(
name = "schema_util", name = "schema_util",
srcs = ["schema_util.py"], srcs = ["schema_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow/lite/schema:utils_friends"], visibility = ["//tensorflow/lite/schema:utils_friends"],
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",

2
tensorflow/lite/python/optimize/BUILD
View File

@ -47,7 +47,7 @@ py_library(
srcs = [ srcs = [
"calibrator.py", "calibrator.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":_pywrap_tensorflow_lite_calibration_wrapper", # buildcleaner: keep ":_pywrap_tensorflow_lite_calibration_wrapper", # buildcleaner: keep

1
tensorflow/lite/schema/BUILD
View File

@ -42,6 +42,7 @@ py_library(
"schema_v3.fbs", "schema_v3.fbs",
"@flatbuffers//:flatc", "@flatbuffers//:flatc",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//tensorflow/python:platform", "//tensorflow/python:platform",

9
tensorflow/lite/testing/BUILD
View File

@ -98,6 +98,7 @@ py_library(
data = [ data = [
"//tensorflow/lite/toco", "//tensorflow/lite/toco",
], ],
srcs_version = "PY3",
deps = [ deps = [
":zip_test_utils", ":zip_test_utils",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -108,7 +109,7 @@ py_library(
py_library( py_library(
name = "op_tests", name = "op_tests",
srcs = glob(["op_tests/*.py"]), srcs = glob(["op_tests/*.py"]),
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":zip_test_utils", ":zip_test_utils",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -119,7 +120,7 @@ py_library(
py_library( py_library(
name = "generate_examples_lib", name = "generate_examples_lib",
srcs = ["generate_examples_lib.py"], srcs = ["generate_examples_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":op_tests", ":op_tests",
":zip_test_utils", ":zip_test_utils",
@ -130,7 +131,7 @@ py_library(
py_library( py_library(
name = "zip_test_utils", name = "zip_test_utils",
srcs = ["zip_test_utils.py"], srcs = ["zip_test_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":_pywrap_string_util", ":_pywrap_string_util",
":generate_examples_report", ":generate_examples_report",
@ -156,7 +157,7 @@ py_binary(
py_library( py_library(
name = "generate_examples_report", name = "generate_examples_report",
srcs = ["generate_examples_report.py"], srcs = ["generate_examples_report.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
cc_library( cc_library(

2
tensorflow/lite/toco/logging/BUILD
View File

@ -66,7 +66,7 @@ py_library(
data = [ data = [
"html_template", "html_template",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
) )

1
tensorflow/lite/toco/python/BUILD
View File

@ -75,6 +75,7 @@ cc_library(
py_library( py_library(
name = "tensorflow_wrap_toco", name = "tensorflow_wrap_toco",
srcs = ["tensorflow_wrap_toco.py"], srcs = ["tensorflow_wrap_toco.py"],
srcs_version = "PY3",
visibility = [ visibility = [
"//learning/expander/pod/deep_pod/utils:__subpackages__", "//learning/expander/pod/deep_pod/utils:__subpackages__",
"//research/handwriting/converters/tflite:__subpackages__", "//research/handwriting/converters/tflite:__subpackages__",

4
tensorflow/lite/tools/BUILD
View File

@ -111,7 +111,7 @@ py_binary(
py_library( py_library(
name = "flatbuffer_utils", name = "flatbuffer_utils",
srcs = ["flatbuffer_utils.py"], srcs = ["flatbuffer_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/lite/python:schema_py", "//tensorflow/lite/python:schema_py",
"//tensorflow/python/platform", "//tensorflow/python/platform",
@ -138,7 +138,7 @@ py_test(
py_library( py_library(
name = "test_utils", name = "test_utils",
srcs = ["test_utils.py"], srcs = ["test_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/lite/python:schema_py", "//tensorflow/lite/python:schema_py",
"@flatbuffers//:runtime_py", "@flatbuffers//:runtime_py",

2
tensorflow/lite/tools/signature/BUILD
View File

@ -71,7 +71,7 @@ pybind_extension(
py_library( py_library(
name = "signature_def_utils", name = "signature_def_utils",
srcs = ["signature_def_utils.py"], srcs = ["signature_def_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":_pywrap_signature_def_util_wrapper", ":_pywrap_signature_def_util_wrapper",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",

345
tensorflow/python/BUILD
View File

File diff suppressed because it is too large Load Diff

4
tensorflow/python/compat/BUILD
View File

@ -7,7 +7,7 @@ package(
py_library( py_library(
name = "v2_compat", name = "v2_compat",
srcs = ["v2_compat.py"], srcs = ["v2_compat.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:control_flow_v2_toggles", "//tensorflow/python:control_flow_v2_toggles",
@ -20,7 +20,7 @@ py_library(
py_library( py_library(
name = "compat", name = "compat",
srcs = ["compat.py"], srcs = ["compat.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",

2
tensorflow/python/compiler/BUILD
View File

@ -13,7 +13,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "compiler", name = "compiler",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = if_windows( deps = if_windows(
["//tensorflow/python/compiler/tensorrt:trt_convert_windows"], ["//tensorflow/python/compiler/tensorrt:trt_convert_windows"],
otherwise = ["//tensorflow/python/compiler/tensorrt:init_py"], otherwise = ["//tensorflow/python/compiler/tensorrt:init_py"],

2
tensorflow/python/compiler/mlir/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "mlir", name = "mlir",
srcs = ["mlir.py"], srcs = ["mlir.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:pywrap_mlir", "//tensorflow/python:pywrap_mlir",
"//tensorflow/python/util:tf_export", "//tensorflow/python/util:tf_export",

7
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -27,7 +27,7 @@ exports_files(glob([
py_library( py_library(
name = "init_py", name = "init_py",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":tf_trt_integration_test_base", ":tf_trt_integration_test_base",
":trt_convert_py", ":trt_convert_py",
@ -40,7 +40,7 @@ py_library(
"trt_convert.py", "trt_convert.py",
"utils.py", "utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/compiler/tf2tensorrt:_pywrap_py_utils", "//tensorflow/compiler/tf2tensorrt:_pywrap_py_utils",
"//tensorflow/compiler/tf2tensorrt:trt_engine_instance_proto_py", "//tensorflow/compiler/tf2tensorrt:trt_engine_instance_proto_py",
@ -64,7 +64,7 @@ py_library(
py_library( py_library(
name = "trt_convert_windows", name = "trt_convert_windows",
srcs = ["trt_convert_windows.py"], srcs = ["trt_convert_windows.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
], ],
@ -73,6 +73,7 @@ py_library(
py_library( py_library(
name = "tf_trt_integration_test_base", name = "tf_trt_integration_test_base",
srcs = ["test/tf_trt_integration_test_base.py"], srcs = ["test/tf_trt_integration_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":trt_convert_py", ":trt_convert_py",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

4
tensorflow/python/compiler/xla/BUILD
View File

@ -11,7 +11,7 @@ py_library(
"__init__.py", "__init__.py",
"jit.py", "jit.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":xla", ":xla",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -50,7 +50,7 @@ cuda_py_test(
py_library( py_library(
name = "xla", name = "xla",
srcs = ["xla.py"], srcs = ["xla.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/compiler/jit:xla_ops_py", "//tensorflow/compiler/jit:xla_ops_py",
"//tensorflow/compiler/jit/ops:xla_ops_grad", "//tensorflow/compiler/jit/ops:xla_ops_grad",

2
tensorflow/python/data/BUILD
View File

@ -8,7 +8,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "data", name = "data",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",

2
tensorflow/python/data/benchmarks/BUILD
View File

@ -10,7 +10,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "benchmark_base", name = "benchmark_base",
srcs = ["benchmark_base.py"], srcs = ["benchmark_base.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python:session", "//tensorflow/python:session",

2
tensorflow/python/data/experimental/BUILD
View File

@ -8,7 +8,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "experimental", name = "experimental",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/data/experimental/ops:dataset_ops", "//tensorflow/python/data/experimental/ops:dataset_ops",

6
tensorflow/python/data/experimental/kernel_tests/BUILD
View File

@ -621,7 +621,7 @@ py_library(
srcs = [ srcs = [
"reader_dataset_ops_test_base.py", "reader_dataset_ops_test_base.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -754,7 +754,7 @@ tf_py_test(
py_library( py_library(
name = "sql_dataset_test_base", name = "sql_dataset_test_base",
srcs = ["sql_dataset_test_base.py"], srcs = ["sql_dataset_test_base.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow/python/data/experimental/kernel_tests:__pkg__", "//tensorflow/python/data/experimental/kernel_tests:__pkg__",
"//tensorflow/python/data/experimental/kernel_tests/serialization:__pkg__", "//tensorflow/python/data/experimental/kernel_tests/serialization:__pkg__",
@ -833,7 +833,7 @@ tf_py_test(
py_library( py_library(
name = "stats_dataset_test_base", name = "stats_dataset_test_base",
srcs = ["stats_dataset_test_base.py"], srcs = ["stats_dataset_test_base.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

2
tensorflow/python/data/experimental/kernel_tests/serialization/BUILD
View File

@ -12,7 +12,7 @@ py_library(
srcs = [ srcs = [
"dataset_serialization_test_base.py", "dataset_serialization_test_base.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",

70
tensorflow/python/data/experimental/ops/BUILD
View File

@ -8,7 +8,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "batching", name = "batching",
srcs = ["batching.py"], srcs = ["batching.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
@ -26,7 +26,7 @@ py_library(
py_library( py_library(
name = "cardinality", name = "cardinality",
srcs = ["cardinality.py"], srcs = ["cardinality.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -36,7 +36,7 @@ py_library(
py_library( py_library(
name = "compression_ops", name = "compression_ops",
srcs = ["compression_ops.py"], srcs = ["compression_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
], ],
@ -45,7 +45,7 @@ py_library(
py_library( py_library(
name = "counter", name = "counter",
srcs = ["counter.py"], srcs = ["counter.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":scan_ops", ":scan_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -60,7 +60,7 @@ py_library(
srcs = [ srcs = [
"data_service_ops.py", "data_service_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -74,7 +74,7 @@ py_library(
srcs = [ srcs = [
"distribute.py", "distribute.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -86,7 +86,7 @@ py_library(
py_library( py_library(
name = "distribute_options", name = "distribute_options",
srcs = ["distribute_options.py"], srcs = ["distribute_options.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/data/util:options", "//tensorflow/python/data/util:options",
@ -96,7 +96,7 @@ py_library(
py_library( py_library(
name = "enumerate_ops", name = "enumerate_ops",
srcs = ["enumerate_ops.py"], srcs = ["enumerate_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
@ -106,7 +106,7 @@ py_library(
py_library( py_library(
name = "error_ops", name = "error_ops",
srcs = ["error_ops.py"], srcs = ["error_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
@ -118,7 +118,7 @@ py_library(
py_library( py_library(
name = "get_single_element", name = "get_single_element",
srcs = ["get_single_element.py"], srcs = ["get_single_element.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
@ -130,7 +130,7 @@ py_library(
py_library( py_library(
name = "grouping", name = "grouping",
srcs = ["grouping.py"], srcs = ["grouping.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",
@ -148,7 +148,7 @@ py_library(
py_library( py_library(
name = "interleave_ops", name = "interleave_ops",
srcs = ["interleave_ops.py"], srcs = ["interleave_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":random_ops", ":random_ops",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -169,7 +169,7 @@ py_library(
srcs = [ srcs = [
"io.py", "io.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
@ -181,7 +181,7 @@ py_library(
srcs = [ srcs = [
"iterator_ops.py", "iterator_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:basic_session_run_hooks", "//tensorflow/python:basic_session_run_hooks",
"//tensorflow/python:checkpoint_management", "//tensorflow/python:checkpoint_management",
@ -197,7 +197,7 @@ py_library(
py_library( py_library(
name = "map_defun", name = "map_defun",
srcs = ["map_defun.py"], srcs = ["map_defun.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -208,7 +208,7 @@ py_library(
py_library( py_library(
name = "matching_files", name = "matching_files",
srcs = ["matching_files.py"], srcs = ["matching_files.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -221,7 +221,7 @@ py_library(
py_library( py_library(
name = "optimization", name = "optimization",
srcs = ["optimization.py"], srcs = ["optimization.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
@ -234,7 +234,7 @@ py_library(
py_library( py_library(
name = "optimization_options", name = "optimization_options",
srcs = ["optimization_options.py"], srcs = ["optimization_options.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/data/util:options", "//tensorflow/python/data/util:options",
@ -244,7 +244,7 @@ py_library(
py_library( py_library(
name = "parsing_ops", name = "parsing_ops",
srcs = ["parsing_ops.py"], srcs = ["parsing_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -260,6 +260,7 @@ py_library(
py_library( py_library(
name = "prefetching_ops", name = "prefetching_ops",
srcs = ["prefetching_ops.py"], srcs = ["prefetching_ops.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -277,7 +278,7 @@ py_library(
srcs = [ srcs = [
"random_ops.py", "random_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -295,7 +296,7 @@ py_library(
srcs = [ srcs = [
"readers.py", "readers.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":batching", ":batching",
":interleave_ops", ":interleave_ops",
@ -323,7 +324,7 @@ py_library(
py_library( py_library(
name = "resampling", name = "resampling",
srcs = ["resampling.py"], srcs = ["resampling.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":batching", ":batching",
":interleave_ops", ":interleave_ops",
@ -343,7 +344,7 @@ py_library(
py_library( py_library(
name = "scan_ops", name = "scan_ops",
srcs = ["scan_ops.py"], srcs = ["scan_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -361,7 +362,7 @@ py_library(
srcs = [ srcs = [
"shuffle_ops.py", "shuffle_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
], ],
@ -372,7 +373,7 @@ py_library(
srcs = [ srcs = [
"snapshot.py", "snapshot.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
@ -384,7 +385,7 @@ py_library(
py_library( py_library(
name = "stats_aggregator", name = "stats_aggregator",
srcs = ["stats_aggregator.py"], srcs = ["stats_aggregator.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -394,7 +395,7 @@ py_library(
py_library( py_library(
name = "stats_ops", name = "stats_ops",
srcs = ["stats_ops.py"], srcs = ["stats_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -409,7 +410,7 @@ py_library(
py_library( py_library(
name = "stats_options", name = "stats_options",
srcs = ["stats_options.py"], srcs = ["stats_options.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":stats_aggregator", ":stats_aggregator",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -420,7 +421,7 @@ py_library(
py_library( py_library(
name = "take_while_ops", name = "take_while_ops",
srcs = ["take_while_ops.py"], srcs = ["take_while_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
@ -433,7 +434,7 @@ py_library(
py_library( py_library(
name = "testing", name = "testing",
srcs = ["testing.py"], srcs = ["testing.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -444,7 +445,7 @@ py_library(
py_library( py_library(
name = "threading_options", name = "threading_options",
srcs = ["threading_options.py"], srcs = ["threading_options.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/data/util:options", "//tensorflow/python/data/util:options",
@ -454,7 +455,7 @@ py_library(
py_library( py_library(
name = "threadpool", name = "threadpool",
srcs = ["threadpool.py"], srcs = ["threadpool.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
"//tensorflow/python:resource_variable_ops", "//tensorflow/python:resource_variable_ops",
@ -469,7 +470,7 @@ py_library(
srcs = [ srcs = [
"unique.py", "unique.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:experimental_dataset_ops_gen", "//tensorflow/python:experimental_dataset_ops_gen",
@ -484,7 +485,7 @@ py_library(
srcs = [ srcs = [
"writers.py", "writers.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:dataset_ops",
@ -493,6 +494,7 @@ py_library(
py_library( py_library(
name = "dataset_ops", name = "dataset_ops",
srcs_version = "PY3",
deps = [ deps = [
":batching", ":batching",
":cardinality", ":cardinality",

3
tensorflow/python/data/experimental/service/BUILD
View File

@ -25,6 +25,7 @@ tf_python_pybind_extension(
py_library( py_library(
name = "server_lib", name = "server_lib",
srcs = ["server_lib.py"], srcs = ["server_lib.py"],
srcs_version = "PY3",
visibility = [ visibility = [
"//visibility:public", "//visibility:public",
], ],
@ -48,7 +49,7 @@ tf_py_test(
py_library( py_library(
name = "service", name = "service",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":server_lib", ":server_lib",
"//tensorflow/python/data/experimental/ops:data_service_ops", "//tensorflow/python/data/experimental/ops:data_service_ops",

1
tensorflow/python/data/kernel_tests/BUILD
View File

@ -741,6 +741,7 @@ tf_py_test(
py_library( py_library(
name = "test_base", name = "test_base",
srcs = ["test_base.py"], srcs = ["test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

10
tensorflow/python/data/ops/BUILD
View File

@ -8,7 +8,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "dataset_ops", name = "dataset_ops",
srcs = ["dataset_ops.py"], srcs = ["dataset_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":iterator_ops", ":iterator_ops",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -45,7 +45,7 @@ py_library(
py_library( py_library(
name = "readers", name = "readers",
srcs = ["readers.py"], srcs = ["readers.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":dataset_ops", ":dataset_ops",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -65,7 +65,7 @@ py_library(
py_library( py_library(
name = "iterator_ops", name = "iterator_ops",
srcs = ["iterator_ops.py"], srcs = ["iterator_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":optional_ops", ":optional_ops",
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
@ -86,7 +86,7 @@ py_library(
py_library( py_library(
name = "optional_ops", name = "optional_ops",
srcs = ["optional_ops.py"], srcs = ["optional_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dataset_ops_gen", "//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -101,7 +101,7 @@ py_library(
py_library( py_library(
name = "multi_device_iterator_ops", name = "multi_device_iterator_ops",
srcs = ["multi_device_iterator_ops.py"], srcs = ["multi_device_iterator_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":dataset_ops", ":dataset_ops",
":iterator_ops", ":iterator_ops",

14
tensorflow/python/data/util/BUILD
View File

@ -10,7 +10,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "nest", name = "nest",
srcs = ["nest.py"], srcs = ["nest.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"@six_archive//:six", "@six_archive//:six",
@ -37,7 +37,7 @@ py_test(
py_library( py_library(
name = "sparse", name = "sparse",
srcs = ["sparse.py"], srcs = ["sparse.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":nest", ":nest",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -69,7 +69,7 @@ py_test(
py_library( py_library(
name = "structure", name = "structure",
srcs = ["structure.py"], srcs = ["structure.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":nest", ":nest",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -120,7 +120,7 @@ py_test(
py_library( py_library(
name = "options", name = "options",
srcs = ["options.py"], srcs = ["options.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_test( py_test(
@ -138,7 +138,7 @@ py_test(
py_library( py_library(
name = "convert", name = "convert",
srcs = ["convert.py"], srcs = ["convert.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -164,7 +164,7 @@ py_test(
py_library( py_library(
name = "random_seed", name = "random_seed",
srcs = ["random_seed.py"], srcs = ["random_seed.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -189,7 +189,7 @@ py_test(
py_library( py_library(
name = "traverse", name = "traverse",
srcs = ["traverse.py"], srcs = ["traverse.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
], ],
) )

95
tensorflow/python/debug/BUILD
View File

@ -21,7 +21,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "debug_py", name = "debug_py",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":check_numerics_callback", ":check_numerics_callback",
@ -47,6 +47,7 @@ py_library(
py_library( py_library(
name = "debug_pip", name = "debug_pip",
data = [":grpc_tensorflow_server"], data = [":grpc_tensorflow_server"],
srcs_version = "PY3",
deps = [ deps = [
":cli_test_utils", ":cli_test_utils",
":debug_py", ":debug_py",
@ -64,13 +65,13 @@ py_library(
py_library( py_library(
name = "op_callbacks_common", name = "op_callbacks_common",
srcs = ["lib/op_callbacks_common.py"], srcs = ["lib/op_callbacks_common.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "check_numerics_callback", name = "check_numerics_callback",
srcs = ["lib/check_numerics_callback.py"], srcs = ["lib/check_numerics_callback.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":op_callbacks_common", ":op_callbacks_common",
":source_utils", ":source_utils",
@ -84,7 +85,7 @@ py_library(
py_library( py_library(
name = "dumping_callback", name = "dumping_callback",
srcs = ["lib/dumping_callback.py"], srcs = ["lib/dumping_callback.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_events_writer", ":debug_events_writer",
":op_callbacks_common", ":op_callbacks_common",
@ -99,7 +100,7 @@ py_library(
py_library( py_library(
name = "dumping_callback_test_lib", name = "dumping_callback_test_lib",
srcs = ["lib/dumping_callback_test_lib.py"], srcs = ["lib/dumping_callback_test_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":check_numerics_callback", ":check_numerics_callback",
":debug_events_reader", ":debug_events_reader",
@ -111,13 +112,13 @@ py_library(
py_library( py_library(
name = "common", name = "common",
srcs = ["lib/common.py"], srcs = ["lib/common.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "debug_events_reader", name = "debug_events_reader",
srcs = ["lib/debug_events_reader.py"], srcs = ["lib/debug_events_reader.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -129,7 +130,7 @@ py_library(
py_library( py_library(
name = "debug_events_monitors", name = "debug_events_monitors",
srcs = ["lib/debug_events_monitors.py"], srcs = ["lib/debug_events_monitors.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -141,7 +142,7 @@ py_library(
py_library( py_library(
name = "debug_events_writer", name = "debug_events_writer",
srcs = ["lib/debug_events_writer.py"], srcs = ["lib/debug_events_writer.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -151,7 +152,7 @@ py_library(
py_library( py_library(
name = "debug_graphs", name = "debug_graphs",
srcs = ["lib/debug_graphs.py"], srcs = ["lib/debug_graphs.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -165,7 +166,7 @@ py_library(
py_library( py_library(
name = "debug_data", name = "debug_data",
srcs = ["lib/debug_data.py"], srcs = ["lib/debug_data.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__", "//third_party/py/tf_slim:__subpackages__",
@ -185,7 +186,7 @@ py_library(
py_library( py_library(
name = "debug_gradients", name = "debug_gradients",
srcs = ["lib/debug_gradients.py"], srcs = ["lib/debug_gradients.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_data", ":debug_data",
":debug_graphs", ":debug_graphs",
@ -199,7 +200,7 @@ py_library(
py_library( py_library(
name = "debug_utils", name = "debug_utils",
srcs = ["lib/debug_utils.py"], srcs = ["lib/debug_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_binary( py_binary(
@ -215,7 +216,7 @@ py_library(
srcs = [ srcs = [
"lib/grpc_tensorflow_server.py", "lib/grpc_tensorflow_server.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:platform", "//tensorflow/python:platform",
@ -226,7 +227,7 @@ py_library(
py_library( py_library(
name = "source_utils", name = "source_utils",
srcs = ["lib/source_utils.py"], srcs = ["lib/source_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":profiling", ":profiling",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -236,7 +237,7 @@ py_library(
py_library( py_library(
name = "source_remote", name = "source_remote",
srcs = ["lib/source_remote.py"], srcs = ["lib/source_remote.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":common", ":common",
":debug_service_pb2_grpc", ":debug_service_pb2_grpc",
@ -248,7 +249,7 @@ py_library(
py_library( py_library(
name = "framework", name = "framework",
srcs = ["wrappers/framework.py"], srcs = ["wrappers/framework.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_utils", ":debug_utils",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -262,7 +263,7 @@ py_library(
py_library( py_library(
name = "debugger_cli_common", name = "debugger_cli_common",
srcs = ["cli/debugger_cli_common.py"], srcs = ["cli/debugger_cli_common.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:platform", "//tensorflow/python:platform",
"//tensorflow/python:pywrap_tf_session", "//tensorflow/python:pywrap_tf_session",
@ -274,20 +275,20 @@ py_library(
py_library( py_library(
name = "cli_config", name = "cli_config",
srcs = ["cli/cli_config.py"], srcs = ["cli/cli_config.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":debugger_cli_common"], deps = [":debugger_cli_common"],
) )
py_library( py_library(
name = "command_parser", name = "command_parser",
srcs = ["cli/command_parser.py"], srcs = ["cli/command_parser.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "tensor_format", name = "tensor_format",
srcs = ["cli/tensor_format.py"], srcs = ["cli/tensor_format.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_data", ":debug_data",
":debugger_cli_common", ":debugger_cli_common",
@ -298,7 +299,7 @@ py_library(
py_library( py_library(
name = "cli_shared", name = "cli_shared",
srcs = ["cli/cli_shared.py"], srcs = ["cli/cli_shared.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":command_parser", ":command_parser",
":common", ":common",
@ -315,7 +316,7 @@ py_library(
py_library( py_library(
name = "evaluator", name = "evaluator",
srcs = ["cli/evaluator.py"], srcs = ["cli/evaluator.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_data", ":debug_data",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -325,7 +326,7 @@ py_library(
py_library( py_library(
name = "analyzer_cli", name = "analyzer_cli",
srcs = ["cli/analyzer_cli.py"], srcs = ["cli/analyzer_cli.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":cli_config", ":cli_config",
":cli_shared", ":cli_shared",
@ -342,13 +343,13 @@ py_library(
py_library( py_library(
name = "profiling", name = "profiling",
srcs = ["lib/profiling.py"], srcs = ["lib/profiling.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "profile_analyzer_cli", name = "profile_analyzer_cli",
srcs = ["cli/profile_analyzer_cli.py"], srcs = ["cli/profile_analyzer_cli.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":cli_shared", ":cli_shared",
":command_parser", ":command_parser",
@ -363,7 +364,7 @@ py_library(
py_library( py_library(
name = "base_ui", name = "base_ui",
srcs = ["cli/base_ui.py"], srcs = ["cli/base_ui.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":cli_config", ":cli_config",
":command_parser", ":command_parser",
@ -374,14 +375,14 @@ py_library(
py_library( py_library(
name = "curses_widgets", name = "curses_widgets",
srcs = ["cli/curses_widgets.py"], srcs = ["cli/curses_widgets.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [":debugger_cli_common"], deps = [":debugger_cli_common"],
) )
py_library( py_library(
name = "curses_ui", name = "curses_ui",
srcs = ["cli/curses_ui.py"], srcs = ["cli/curses_ui.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_ui", ":base_ui",
":cli_shared", ":cli_shared",
@ -396,7 +397,7 @@ py_library(
py_library( py_library(
name = "readline_ui", name = "readline_ui",
srcs = ["cli/readline_ui.py"], srcs = ["cli/readline_ui.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_ui", ":base_ui",
":debugger_cli_common", ":debugger_cli_common",
@ -406,7 +407,7 @@ py_library(
py_library( py_library(
name = "ui_factory", name = "ui_factory",
srcs = ["cli/ui_factory.py"], srcs = ["cli/ui_factory.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":curses_ui", ":curses_ui",
":readline_ui", ":readline_ui",
@ -416,7 +417,7 @@ py_library(
py_library( py_library(
name = "dumping_wrapper", name = "dumping_wrapper",
srcs = ["wrappers/dumping_wrapper.py"], srcs = ["wrappers/dumping_wrapper.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__", "//third_party/py/tf_slim:__subpackages__",
@ -432,7 +433,7 @@ py_library(
py_library( py_library(
name = "grpc_wrapper", name = "grpc_wrapper",
srcs = ["wrappers/grpc_wrapper.py"], srcs = ["wrappers/grpc_wrapper.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":common", ":common",
":framework", ":framework",
@ -443,7 +444,7 @@ py_library(
py_library( py_library(
name = "local_cli_wrapper", name = "local_cli_wrapper",
srcs = ["wrappers/local_cli_wrapper.py"], srcs = ["wrappers/local_cli_wrapper.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":analyzer_cli", ":analyzer_cli",
":cli_shared", ":cli_shared",
@ -461,7 +462,7 @@ py_library(
py_library( py_library(
name = "hooks", name = "hooks",
srcs = ["wrappers/hooks.py"], srcs = ["wrappers/hooks.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__", "//third_party/py/tf_slim:__subpackages__",
@ -488,7 +489,7 @@ py_binary(
py_library( py_library(
name = "offline_analyzer_lib", name = "offline_analyzer_lib",
srcs = ["cli/offline_analyzer.py"], srcs = ["cli/offline_analyzer.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":analyzer_cli", ":analyzer_cli",
":debug_data", ":debug_data",
@ -499,6 +500,7 @@ py_library(
py_library( py_library(
name = "debug_examples_v1", name = "debug_examples_v1",
srcs_version = "PY3",
deps = [ deps = [
":debug_errors_lib", ":debug_errors_lib",
":debug_fibonacci_lib", ":debug_fibonacci_lib",
@ -511,6 +513,7 @@ py_library(
py_library( py_library(
name = "debug_examples_v2", name = "debug_examples_v2",
srcs_version = "PY3",
deps = [ deps = [
":debug_fibonacci_lib", ":debug_fibonacci_lib",
":debug_mnist_lib", ":debug_mnist_lib",
@ -539,7 +542,7 @@ py_library(
"examples/v1/debug_fibonacci.py", "examples/v1/debug_fibonacci.py",
"examples/v2/debug_fibonacci_v2.py", "examples/v2/debug_fibonacci_v2.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_py", ":debug_py",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -559,7 +562,7 @@ py_binary(
py_library( py_library(
name = "debug_errors_lib", name = "debug_errors_lib",
srcs = ["examples/v1/debug_errors.py"], srcs = ["examples/v1/debug_errors.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_py", ":debug_py",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -598,7 +601,7 @@ py_library(
"examples/v1/debug_mnist_v1.py", "examples/v1/debug_mnist_v1.py",
"examples/v2/debug_mnist_v2.py", "examples/v2/debug_mnist_v2.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_py", ":debug_py",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -616,7 +619,7 @@ py_binary(
py_library( py_library(
name = "debug_tflearn_iris_lib", name = "debug_tflearn_iris_lib",
srcs = ["examples/v1/debug_tflearn_iris.py"], srcs = ["examples/v1/debug_tflearn_iris.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_py", ":debug_py",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -635,7 +638,7 @@ py_binary(
py_library( py_library(
name = "debug_keras_lib", name = "debug_keras_lib",
srcs = ["examples/v1/debug_keras.py"], srcs = ["examples/v1/debug_keras.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_py", ":debug_py",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -970,7 +973,7 @@ py_test(
py_library( py_library(
name = "session_debug_testlib", name = "session_debug_testlib",
srcs = ["lib/session_debug_testlib.py"], srcs = ["lib/session_debug_testlib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_data", ":debug_data",
":debug_graphs", ":debug_graphs",
@ -1001,7 +1004,7 @@ py_library(
py_library( py_library(
name = "debug_service_pb2_grpc", name = "debug_service_pb2_grpc",
srcs = ["lib/debug_service_pb2_grpc.py"], srcs = ["lib/debug_service_pb2_grpc.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core/debug:debug_service_proto_py", "//tensorflow/core/debug:debug_service_proto_py",
], ],
@ -1010,7 +1013,7 @@ py_library(
py_library( py_library(
name = "grpc_debug_server", name = "grpc_debug_server",
srcs = ["lib/grpc_debug_server.py"], srcs = ["lib/grpc_debug_server.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":debug_graphs", ":debug_graphs",
@ -1023,7 +1026,7 @@ py_library(
py_library( py_library(
name = "grpc_debug_test_server", name = "grpc_debug_test_server",
srcs = ["lib/grpc_debug_test_server.py"], srcs = ["lib/grpc_debug_test_server.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":debug_data", ":debug_data",
":debug_utils", ":debug_utils",
@ -1209,7 +1212,7 @@ py_test(
py_library( py_library(
name = "cli_test_utils", name = "cli_test_utils",
srcs = ["cli/cli_test_utils.py"], srcs = ["cli/cli_test_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
cuda_py_test( cuda_py_test(

56
tensorflow/python/distribute/BUILD
View File

@ -15,6 +15,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "distribute_test_lib_pip", name = "distribute_test_lib_pip",
srcs_version = "PY3",
deps = [ deps = [
":combinations", ":combinations",
":multi_worker_test_base", ":multi_worker_test_base",
@ -29,7 +30,7 @@ py_library(
py_library( py_library(
name = "cross_device_ops", name = "cross_device_ops",
srcs = ["cross_device_ops.py"], srcs = ["cross_device_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":collective_util", ":collective_util",
":cross_device_utils", ":cross_device_utils",
@ -61,7 +62,7 @@ py_library(
py_library( py_library(
name = "cross_device_utils", name = "cross_device_utils",
srcs = ["cross_device_utils.py"], srcs = ["cross_device_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":values", ":values",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -81,7 +82,7 @@ py_library(
py_library( py_library(
name = "device_util", name = "device_util",
srcs = ["device_util.py"], srcs = ["device_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:device", "//tensorflow/python:device",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -111,7 +112,7 @@ cuda_py_test(
py_library( py_library(
name = "distribute", name = "distribute",
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":cross_device_ops", ":cross_device_ops",
":distribute_lib", ":distribute_lib",
@ -132,7 +133,7 @@ py_library(
"distribute_lib.py", "distribute_lib.py",
"distribution_strategy_context.py", "distribution_strategy_context.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":collective_util", ":collective_util",
":device_util", ":device_util",
@ -184,6 +185,7 @@ py_library(
srcs = [ srcs = [
"distribute_config.py", "distribute_config.py",
], ],
srcs_version = "PY3",
deps = [], deps = [],
) )
@ -192,7 +194,7 @@ py_library(
srcs = [ srcs = [
"distribute_coordinator.py", "distribute_coordinator.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distribute_coordinator_context", ":distribute_coordinator_context",
":multi_worker_util", ":multi_worker_util",
@ -230,13 +232,14 @@ py_library(
srcs = [ srcs = [
"distribute_coordinator_context.py", "distribute_coordinator_context.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [], deps = [],
) )
py_library( py_library(
name = "mirrored_run", name = "mirrored_run",
srcs = ["mirrored_run.py"], srcs = ["mirrored_run.py"],
srcs_version = "PY3",
deps = [ deps = [
":device_util", ":device_util",
":distribute_lib", ":distribute_lib",
@ -267,6 +270,7 @@ py_library(
py_library( py_library(
name = "distribute_utils", name = "distribute_utils",
srcs = ["distribute_utils.py"], srcs = ["distribute_utils.py"],
srcs_version = "PY3",
deps = [ deps = [
":device_util", ":device_util",
":distribute_lib", ":distribute_lib",
@ -299,6 +303,7 @@ py_library(
py_library( py_library(
name = "mirrored_strategy", name = "mirrored_strategy",
srcs = ["mirrored_strategy.py"], srcs = ["mirrored_strategy.py"],
srcs_version = "PY3",
deps = [ deps = [
":collective_util", ":collective_util",
":cross_device_ops", ":cross_device_ops",
@ -327,6 +332,7 @@ py_library(
py_library( py_library(
name = "parameter_server_strategy", name = "parameter_server_strategy",
srcs = ["parameter_server_strategy.py"], srcs = ["parameter_server_strategy.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":cross_device_ops", ":cross_device_ops",
@ -356,6 +362,7 @@ py_library(
py_library( py_library(
name = "central_storage_strategy", name = "central_storage_strategy",
srcs = ["central_storage_strategy.py"], srcs = ["central_storage_strategy.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":device_util", ":device_util",
@ -368,6 +375,7 @@ py_library(
py_library( py_library(
name = "one_device_strategy", name = "one_device_strategy",
srcs = ["one_device_strategy.py"], srcs = ["one_device_strategy.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":distribute_lib", ":distribute_lib",
@ -387,6 +395,7 @@ py_library(
py_library( py_library(
name = "collective_all_reduce_strategy", name = "collective_all_reduce_strategy",
srcs = ["collective_all_reduce_strategy.py"], srcs = ["collective_all_reduce_strategy.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":collective_util", ":collective_util",
@ -420,7 +429,7 @@ py_library(
srcs = [ srcs = [
"multi_worker_util.py", "multi_worker_util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
@ -460,6 +469,7 @@ cuda_py_test(
py_library( py_library(
name = "numpy_dataset", name = "numpy_dataset",
srcs = ["numpy_dataset.py"], srcs = ["numpy_dataset.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -490,6 +500,7 @@ py_test(
py_library( py_library(
name = "input_lib", name = "input_lib",
srcs = ["input_lib.py"], srcs = ["input_lib.py"],
srcs_version = "PY3",
deps = [ deps = [
":device_util", ":device_util",
":distribute_lib", ":distribute_lib",
@ -507,6 +518,7 @@ py_library(
py_library( py_library(
name = "input_ops", name = "input_ops",
srcs = ["input_ops.py"], srcs = ["input_ops.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python/data/util:nest", "//tensorflow/python/data/util:nest",
@ -553,6 +565,7 @@ py_test(
py_library( py_library(
name = "tpu_strategy", name = "tpu_strategy",
srcs = ["tpu_strategy.py"], srcs = ["tpu_strategy.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":cross_device_ops", ":cross_device_ops",
@ -642,7 +655,7 @@ py_library(
srcs = [ srcs = [
"estimator_training.py", "estimator_training.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distribute_coordinator", ":distribute_coordinator",
":distribute_coordinator_context", ":distribute_coordinator_context",
@ -653,6 +666,7 @@ py_library(
py_library( py_library(
name = "reduce_util", name = "reduce_util",
srcs = ["reduce_util.py"], srcs = ["reduce_util.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python:variable_scope", "//tensorflow/python:variable_scope",
@ -662,7 +676,7 @@ py_library(
py_library( py_library(
name = "collective_util", name = "collective_util",
srcs = ["collective_util.py"], srcs = ["collective_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python:variable_scope", "//tensorflow/python:variable_scope",
@ -681,6 +695,7 @@ tf_py_test(
py_library( py_library(
name = "shared_variable_creator", name = "shared_variable_creator",
srcs = ["shared_variable_creator.py"], srcs = ["shared_variable_creator.py"],
srcs_version = "PY3",
) )
py_test( py_test(
@ -699,6 +714,7 @@ py_test(
py_library( py_library(
name = "summary_op_util", name = "summary_op_util",
srcs = ["summary_op_util.py"], srcs = ["summary_op_util.py"],
srcs_version = "PY3",
deps = [ deps = [
":distribute_lib", ":distribute_lib",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -709,6 +725,7 @@ py_library(
py_library( py_library(
name = "packed_distributed_variable", name = "packed_distributed_variable",
srcs = ["packed_distributed_variable.py"], srcs = ["packed_distributed_variable.py"],
srcs_version = "PY3",
deps = [ deps = [
":device_util", ":device_util",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -721,6 +738,7 @@ py_library(
py_library( py_library(
name = "values", name = "values",
srcs = ["values.py"], srcs = ["values.py"],
srcs_version = "PY3",
deps = [ deps = [
":device_util", ":device_util",
":distribute_lib", ":distribute_lib",
@ -749,6 +767,7 @@ py_library(
py_library( py_library(
name = "ps_values", name = "ps_values",
srcs = ["ps_values.py"], srcs = ["ps_values.py"],
srcs_version = "PY3",
deps = [ deps = [
":distribute_lib", ":distribute_lib",
":values", ":values",
@ -764,6 +783,7 @@ py_library(
py_library( py_library(
name = "values_util", name = "values_util",
srcs = ["values_util.py"], srcs = ["values_util.py"],
srcs_version = "PY3",
deps = [ deps = [
":distribute_lib", ":distribute_lib",
":reduce_util", ":reduce_util",
@ -780,6 +800,7 @@ py_library(
py_library( py_library(
name = "tpu_values", name = "tpu_values",
srcs = ["tpu_values.py"], srcs = ["tpu_values.py"],
srcs_version = "PY3",
deps = [ deps = [
":packed_distributed_variable", ":packed_distributed_variable",
":values", ":values",
@ -796,7 +817,7 @@ py_library(
py_library( py_library(
name = "combinations", name = "combinations",
srcs = ["combinations.py"], srcs = ["combinations.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//tensorflow_models:__subpackages__", "//tensorflow_models:__subpackages__",
@ -839,7 +860,7 @@ py_test(
py_library( py_library(
name = "strategy_combinations", name = "strategy_combinations",
srcs = ["strategy_combinations.py"], srcs = ["strategy_combinations.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//tensorflow_models:__subpackages__", "//tensorflow_models:__subpackages__",
@ -895,7 +916,7 @@ distribute_py_test(
py_library( py_library(
name = "multi_worker_test_base", name = "multi_worker_test_base",
srcs = ["multi_worker_test_base.py"], srcs = ["multi_worker_test_base.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distribute_coordinator", ":distribute_coordinator",
":multi_process_runner", ":multi_process_runner",
@ -1101,7 +1122,7 @@ cuda_py_test(
py_library( py_library(
name = "sharded_variable", name = "sharded_variable",
srcs = ["sharded_variable.py"], srcs = ["sharded_variable.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:composite_tensor", "//tensorflow/python:composite_tensor",
@ -1155,7 +1176,7 @@ tf_py_test(
py_library( py_library(
name = "strategy_test_lib", name = "strategy_test_lib",
srcs = ["strategy_test_lib.py"], srcs = ["strategy_test_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distribute_lib", ":distribute_lib",
":reduce_util", ":reduce_util",
@ -1392,6 +1413,7 @@ distribute_py_test(
py_library( py_library(
name = "single_loss_example", name = "single_loss_example",
srcs = ["single_loss_example.py"], srcs = ["single_loss_example.py"],
srcs_version = "PY3",
deps = [ deps = [
":step_fn", ":step_fn",
":strategy_test_lib", ":strategy_test_lib",
@ -1406,6 +1428,7 @@ py_library(
py_library( py_library(
name = "step_fn", name = "step_fn",
srcs = ["step_fn.py"], srcs = ["step_fn.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:training", "//tensorflow/python:training",
@ -1653,6 +1676,7 @@ py_library(
py_library( py_library(
name = "multi_process_lib", name = "multi_process_lib",
srcs = ["multi_process_lib.py"], srcs = ["multi_process_lib.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python/eager:test", "//tensorflow/python/eager:test",
@ -1838,7 +1862,7 @@ distribute_py_test(
py_library( py_library(
name = "parameter_server_strategy_v2", name = "parameter_server_strategy_v2",
srcs = ["parameter_server_strategy_v2.py"], srcs = ["parameter_server_strategy_v2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distribute_lib", ":distribute_lib",
":distribute_utils", ":distribute_utils",

16
tensorflow/python/distribute/cluster_resolver/BUILD
View File

@ -14,7 +14,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
@ -31,7 +31,7 @@ py_library(
py_library( py_library(
name = "base_cluster_resolver_py", name = "base_cluster_resolver_py",
srcs = ["cluster_resolver.py"], srcs = ["cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
], ],
@ -40,7 +40,7 @@ py_library(
py_library( py_library(
name = "gce_cluster_resolver_py", name = "gce_cluster_resolver_py",
srcs = ["gce_cluster_resolver.py"], srcs = ["gce_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
@ -50,7 +50,7 @@ py_library(
py_library( py_library(
name = "tfconfig_cluster_resolver_py", name = "tfconfig_cluster_resolver_py",
srcs = ["tfconfig_cluster_resolver.py"], srcs = ["tfconfig_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
@ -60,7 +60,7 @@ py_library(
py_library( py_library(
name = "sagemaker_cluster_resolver_py", name = "sagemaker_cluster_resolver_py",
srcs = ["sagemaker_cluster_resolver.py"], srcs = ["sagemaker_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
@ -70,14 +70,14 @@ py_library(
py_library( py_library(
name = "tpu_cluster_resolver_py", name = "tpu_cluster_resolver_py",
srcs = ["tpu_cluster_resolver.py"], srcs = ["tpu_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow/python/distribute/cluster_resolver/tpu:tpu_cluster_resolver_py"], deps = ["//tensorflow/python/distribute/cluster_resolver/tpu:tpu_cluster_resolver_py"],
) )
py_library( py_library(
name = "slurm_cluster_resolver_py", name = "slurm_cluster_resolver_py",
srcs = ["slurm_cluster_resolver.py"], srcs = ["slurm_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
@ -87,7 +87,7 @@ py_library(
py_library( py_library(
name = "kubernetes_cluster_resolver_py", name = "kubernetes_cluster_resolver_py",
srcs = ["kubernetes_cluster_resolver.py"], srcs = ["kubernetes_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_cluster_resolver_py", ":base_cluster_resolver_py",
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",

2
tensorflow/python/distribute/cluster_resolver/tpu/BUILD
View File

@ -16,7 +16,7 @@ package(
py_library( py_library(
name = "tpu_cluster_resolver_py", name = "tpu_cluster_resolver_py",
srcs = ["tpu_cluster_resolver.py"], srcs = ["tpu_cluster_resolver.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
"//tensorflow/python/distribute/cluster_resolver:base_cluster_resolver_py", "//tensorflow/python/distribute/cluster_resolver:base_cluster_resolver_py",

8
tensorflow/python/distribute/coordinator/BUILD
View File

@ -10,7 +10,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "cluster_coordinator", name = "cluster_coordinator",
srcs = ["cluster_coordinator.py"], srcs = ["cluster_coordinator.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":metric_utils", ":metric_utils",
":utils", ":utils",
@ -126,7 +126,7 @@ tf_py_test(
py_library( py_library(
name = "metric_utils", name = "metric_utils",
srcs = ["metric_utils.py"], srcs = ["metric_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/eager:monitoring", "//tensorflow/python/eager:monitoring",
], ],
@ -150,7 +150,7 @@ tf_py_test(
py_library( py_library(
name = "utils", name = "utils",
srcs = ["utils.py"], srcs = ["utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:training_server_lib", "//tensorflow/python:training_server_lib",
], ],
@ -158,6 +158,6 @@ py_library(
py_library( py_library(
name = "remote_eager_lib", name = "remote_eager_lib",
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
) )

2
tensorflow/python/distribute/experimental/BUILD
View File

@ -10,7 +10,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/distribute:central_storage_strategy", "//tensorflow/python/distribute:central_storage_strategy",
"//tensorflow/python/distribute:collective_all_reduce_strategy", "//tensorflow/python/distribute:collective_all_reduce_strategy",

4
tensorflow/python/distribute/parallel_device/BUILD
View File

@ -14,7 +14,7 @@ exports_files(
py_library( py_library(
name = "parallel_device", name = "parallel_device",
srcs = ["parallel_device.py"], srcs = ["parallel_device.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":saving", ":saving",
"//tensorflow/python:_pywrap_parallel_device", "//tensorflow/python:_pywrap_parallel_device",
@ -26,7 +26,7 @@ py_library(
py_library( py_library(
name = "saving", name = "saving",
srcs = ["saving.py"], srcs = ["saving.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["//tensorflow/python:framework_ops"], deps = ["//tensorflow/python:framework_ops"],
) )

2
tensorflow/python/distribute/v1/BUILD
View File

@ -46,7 +46,7 @@ py_library(
srcs = [ srcs = [
"all_reduce.py", "all_reduce.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",

2
tensorflow/python/dlpack/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "dlpack", name = "dlpack",
srcs = ["dlpack.py"], srcs = ["dlpack.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:__subpackages__"], visibility = ["//tensorflow:__subpackages__"],
deps = [ deps = [
"//tensorflow/python:pywrap_tensorflow", "//tensorflow/python:pywrap_tensorflow",

47
tensorflow/python/eager/BUILD
View File

@ -131,6 +131,7 @@ filegroup(
# Transitive dependencies of this target will be included in the pip package. # Transitive dependencies of this target will be included in the pip package.
py_library( py_library(
name = "eager_pip", name = "eager_pip",
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":backprop", ":backprop",
@ -159,7 +160,7 @@ py_library(
py_library( py_library(
name = "core", name = "core",
srcs = ["core.py"], srcs = ["core.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -171,7 +172,7 @@ py_library(
py_library( py_library(
name = "cancellation", name = "cancellation",
srcs = ["cancellation.py"], srcs = ["cancellation.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:pywrap_tfe", "//tensorflow/python:pywrap_tfe",
@ -192,7 +193,7 @@ cuda_py_test(
py_library( py_library(
name = "executor", name = "executor",
srcs = ["executor.py"], srcs = ["executor.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:pywrap_tfe", "//tensorflow/python:pywrap_tfe",
@ -202,7 +203,7 @@ py_library(
py_library( py_library(
name = "context", name = "context",
srcs = ["context.py"], srcs = ["context.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":executor", ":executor",
@ -277,7 +278,7 @@ cuda_py_test(
py_library( py_library(
name = "monitoring", name = "monitoring",
srcs = ["monitoring.py"], srcs = ["monitoring.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:c_api_util", "//tensorflow/python:c_api_util",
@ -300,7 +301,7 @@ cuda_py_test(
py_library( py_library(
name = "profiler", name = "profiler",
srcs = ["profiler.py"], srcs = ["profiler.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -326,7 +327,7 @@ cuda_py_test(
py_library( py_library(
name = "profiler_client", name = "profiler_client",
srcs = ["profiler_client.py"], srcs = ["profiler_client.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:internal", "//tensorflow:internal",
"//tensorflow/core/profiler:internal", "//tensorflow/core/profiler:internal",
@ -355,7 +356,7 @@ py_test(
py_library( py_library(
name = "tape", name = "tape",
srcs = ["tape.py"], srcs = ["tape.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
) )
@ -526,7 +527,7 @@ cuda_py_test(
py_library( py_library(
name = "test", name = "test",
srcs = ["test.py"], srcs = ["test.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":core", ":core",
@ -538,7 +539,7 @@ py_library(
py_library( py_library(
name = "execute", name = "execute",
srcs = ["execute.py"], srcs = ["execute.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -556,7 +557,7 @@ py_library(
py_library( py_library(
name = "graph_only_ops", name = "graph_only_ops",
srcs = ["graph_only_ops.py"], srcs = ["graph_only_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -582,7 +583,7 @@ cuda_py_test(
py_library( py_library(
name = "framework_for_generated_wrappers", name = "framework_for_generated_wrappers",
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":execute", ":execute",
@ -595,7 +596,7 @@ py_library(
py_library( py_library(
name = "function", name = "function",
srcs = ["function.py"], srcs = ["function.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -632,7 +633,7 @@ pybind_extension(
py_library( py_library(
name = "backprop", name = "backprop",
srcs = ["backprop.py"], srcs = ["backprop.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":backprop_util", ":backprop_util",
@ -660,7 +661,7 @@ py_library(
py_library( py_library(
name = "backprop_util", name = "backprop_util",
srcs = ["backprop_util.py"], srcs = ["backprop_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -671,7 +672,7 @@ py_library(
py_library( py_library(
name = "forwardprop", name = "forwardprop",
srcs = ["forwardprop.py"], srcs = ["forwardprop.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":forwardprop_util", ":forwardprop_util",
@ -686,7 +687,7 @@ py_library(
py_library( py_library(
name = "forwardprop_util", name = "forwardprop_util",
srcs = ["forwardprop_util.py"], srcs = ["forwardprop_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:pywrap_tfe", "//tensorflow/python:pywrap_tfe",
@ -696,7 +697,7 @@ py_library(
py_library( py_library(
name = "benchmarks_test_base", name = "benchmarks_test_base",
srcs = ["benchmarks_test_base.py"], srcs = ["benchmarks_test_base.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [":test"], deps = [":test"],
) )
@ -811,7 +812,7 @@ tf_py_test(
py_library( py_library(
name = "imperative_grad", name = "imperative_grad",
srcs = ["imperative_grad.py"], srcs = ["imperative_grad.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:pywrap_tfe", "//tensorflow/python:pywrap_tfe",
"//tensorflow/python:unconnected_gradients", "//tensorflow/python:unconnected_gradients",
@ -822,7 +823,7 @@ py_library(
py_library( py_library(
name = "def_function", name = "def_function",
srcs = ["def_function.py"], srcs = ["def_function.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -845,7 +846,7 @@ py_library(
py_library( py_library(
name = "lift_to_graph", name = "lift_to_graph",
srcs = ["lift_to_graph.py"], srcs = ["lift_to_graph.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -957,7 +958,7 @@ tf_xla_py_test(
py_library( py_library(
name = "wrap_function", name = "wrap_function",
srcs = ["wrap_function.py"], srcs = ["wrap_function.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",
@ -1005,7 +1006,7 @@ cuda_py_test(
py_library( py_library(
name = "remote", name = "remote",
srcs = ["remote.py"], srcs = ["remote.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
":context", ":context",

6
tensorflow/python/eager/benchmarks/resnet50/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "resnet50", name = "resnet50",
srcs = ["resnet50.py"], srcs = ["resnet50.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
], ],
@ -17,7 +17,7 @@ py_library(
py_library( py_library(
name = "resnet50_test_util", name = "resnet50_test_util",
srcs = ["resnet50_test_util.py"], srcs = ["resnet50_test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":resnet50", ":resnet50",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -27,7 +27,7 @@ py_library(
py_library( py_library(
name = "resnet50_test_lib", name = "resnet50_test_lib",
srcs = ["resnet50_test.py"], srcs = ["resnet50_test.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":resnet50", ":resnet50",
":resnet50_test_util", ":resnet50_test_util",

1
tensorflow/python/eager/memory_tests/BUILD
View File

@ -11,6 +11,7 @@ package(
py_library( py_library(
name = "memory_test_util", name = "memory_test_util",
srcs = ["memory_test_util.py"], srcs = ["memory_test_util.py"],
srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = ["@six_archive//:six"], deps = ["@six_archive//:six"],
) )

59
tensorflow/python/estimator/BUILD
View File

@ -8,7 +8,7 @@ py_library(
srcs = [ srcs = [
"estimator_lib.py", "estimator_lib.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//tensorflow:__pkg__", "//tensorflow:__pkg__",
"//tensorflow:internal", "//tensorflow:internal",
@ -36,7 +36,7 @@ py_library(
py_library( py_library(
name = "exporter", name = "exporter",
srcs = ["exporter.py"], srcs = ["exporter.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":gc", ":gc",
@ -49,7 +49,7 @@ py_library(
py_library( py_library(
name = "gc", name = "gc",
srcs = ["gc.py"], srcs = ["gc.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -59,7 +59,7 @@ py_library(
py_library( py_library(
name = "model_fn", name = "model_fn",
srcs = ["model_fn.py"], srcs = ["model_fn.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":export_output", ":export_output",
@ -71,7 +71,7 @@ py_library(
py_library( py_library(
name = "training", name = "training",
srcs = ["training.py"], srcs = ["training.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -85,7 +85,7 @@ py_library(
py_library( py_library(
name = "run_config", name = "run_config",
srcs = ["run_config.py"], srcs = ["run_config.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -96,7 +96,7 @@ py_library(
py_library( py_library(
name = "baseline", name = "baseline",
srcs = ["canned/baseline.py"], srcs = ["canned/baseline.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -111,7 +111,7 @@ py_library(
py_library( py_library(
name = "boosted_trees", name = "boosted_trees",
srcs = ["canned/boosted_trees.py"], srcs = ["canned/boosted_trees.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":boosted_trees_utils", ":boosted_trees_utils",
":estimator", ":estimator",
@ -125,7 +125,7 @@ py_library(
py_library( py_library(
name = "boosted_trees_utils", name = "boosted_trees_utils",
srcs = ["canned/boosted_trees_utils.py"], srcs = ["canned/boosted_trees_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -138,7 +138,7 @@ py_library(
py_library( py_library(
name = "dnn", name = "dnn",
srcs = ["canned/dnn.py"], srcs = ["canned/dnn.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -154,7 +154,7 @@ py_library(
name = "dnn_testing_utils", name = "dnn_testing_utils",
testonly = 1, testonly = 1,
srcs = ["canned/dnn_testing_utils.py"], srcs = ["canned/dnn_testing_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -173,7 +173,7 @@ py_library(
py_library( py_library(
name = "dnn_linear_combined", name = "dnn_linear_combined",
srcs = ["canned/dnn_linear_combined.py"], srcs = ["canned/dnn_linear_combined.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":dnn", ":dnn",
":estimator", ":estimator",
@ -192,7 +192,7 @@ py_library(
srcs = [ srcs = [
"util.py", "util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -204,7 +204,7 @@ py_library(
srcs = [ srcs = [
"estimator.py", "estimator.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":export_export", ":export_export",
@ -222,7 +222,7 @@ py_library(
srcs = [ srcs = [
"canned/parsing_utils.py", "canned/parsing_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -233,7 +233,7 @@ py_library(
py_library( py_library(
name = "export_output", name = "export_output",
srcs = ["export/export_output.py"], srcs = ["export/export_output.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -246,7 +246,7 @@ py_library(
srcs = [ srcs = [
"export/export_lib.py", "export/export_lib.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":export_export", ":export_export",
@ -260,7 +260,7 @@ py_library(
srcs = [ srcs = [
"export/export.py", "export/export.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":util", ":util",
@ -272,7 +272,7 @@ py_library(
py_library( py_library(
name = "head", name = "head",
srcs = ["canned/head.py"], srcs = ["canned/head.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":export_output", ":export_output",
@ -287,7 +287,7 @@ py_library(
py_library( py_library(
name = "inputs", name = "inputs",
srcs = ["inputs/inputs.py"], srcs = ["inputs/inputs.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":numpy_io", ":numpy_io",
@ -299,7 +299,7 @@ py_library(
py_library( py_library(
name = "linear", name = "linear",
srcs = ["canned/linear.py"], srcs = ["canned/linear.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -314,7 +314,7 @@ py_library(
name = "linear_testing_utils", name = "linear_testing_utils",
testonly = 1, testonly = 1,
srcs = ["canned/linear_testing_utils.py"], srcs = ["canned/linear_testing_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -332,7 +332,7 @@ py_library(
py_library( py_library(
name = "metric_keys", name = "metric_keys",
srcs = ["canned/metric_keys.py"], srcs = ["canned/metric_keys.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":model_fn", ":model_fn",
@ -342,7 +342,7 @@ py_library(
py_library( py_library(
name = "numpy_io", name = "numpy_io",
srcs = ["inputs/numpy_io.py"], srcs = ["inputs/numpy_io.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":inputs_queues", ":inputs_queues",
@ -352,7 +352,7 @@ py_library(
py_library( py_library(
name = "optimizers", name = "optimizers",
srcs = ["canned/optimizers.py"], srcs = ["canned/optimizers.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -363,7 +363,7 @@ py_library(
py_library( py_library(
name = "pandas_io", name = "pandas_io",
srcs = ["inputs/pandas_io.py"], srcs = ["inputs/pandas_io.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
":inputs_queues", ":inputs_queues",
@ -373,7 +373,7 @@ py_library(
py_library( py_library(
name = "prediction_keys", name = "prediction_keys",
srcs = ["canned/prediction_keys.py"], srcs = ["canned/prediction_keys.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
], ],
@ -385,7 +385,7 @@ py_library(
"inputs/queues/feeding_functions.py", "inputs/queues/feeding_functions.py",
"inputs/queues/feeding_queue_runner.py", "inputs/queues/feeding_queue_runner.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
"//tensorflow:tensorflow_py_no_contrib", "//tensorflow:tensorflow_py_no_contrib",
@ -396,7 +396,7 @@ py_library(
py_library( py_library(
name = "keras", name = "keras",
srcs = ["keras.py"], srcs = ["keras.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":estimator", ":estimator",
":expect_tensorflow_estimator_installed", ":expect_tensorflow_estimator_installed",
@ -409,6 +409,7 @@ py_library(
py_library( py_library(
name = "expect_tensorflow_estimator_installed", name = "expect_tensorflow_estimator_installed",
srcs_version = "PY3",
# This is a dummy rule used as a dependency in open-source. # This is a dummy rule used as a dependency in open-source.
# We expect tensorflow_estimator to already be installed. # We expect tensorflow_estimator to already be installed.
visibility = ["//visibility:public"], visibility = ["//visibility:public"],

10
tensorflow/python/feature_column/BUILD
View File

@ -8,7 +8,7 @@ package(
py_library( py_library(
name = "feature_column_py", name = "feature_column_py",
srcs = ["feature_column_lib.py"], srcs = ["feature_column_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":feature_column", ":feature_column",
":feature_column_v2", ":feature_column_v2",
@ -20,7 +20,7 @@ py_library(
py_library( py_library(
name = "feature_column", name = "feature_column",
srcs = ["feature_column.py"], srcs = ["feature_column.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":utils", ":utils",
"@six_archive//:six", "@six_archive//:six",
@ -65,7 +65,7 @@ py_library(
"sequence_feature_column.py", "sequence_feature_column.py",
"serialization.py", "serialization.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":feature_column", ":feature_column",
":utils", ":utils",
@ -127,6 +127,7 @@ py_library(
name = "feature_column_test_main_lib", name = "feature_column_test_main_lib",
srcs = ["feature_column_test.py"], srcs = ["feature_column_test.py"],
data = [":vocabulary_testdata"], data = [":vocabulary_testdata"],
srcs_version = "PY3",
deps = [ deps = [
":feature_column", ":feature_column",
":feature_column_py", ":feature_column_py",
@ -169,6 +170,7 @@ py_library(
name = "feature_column_v2_test_main_lib", name = "feature_column_v2_test_main_lib",
srcs = ["feature_column_v2_test.py"], srcs = ["feature_column_v2_test.py"],
data = [":vocabulary_testdata"], data = [":vocabulary_testdata"],
srcs_version = "PY3",
deps = [ deps = [
":feature_column_py", ":feature_column_py",
":feature_column_v2", ":feature_column_v2",
@ -197,7 +199,7 @@ py_library(
py_library( py_library(
name = "utils", name = "utils",
srcs = ["utils.py"], srcs = ["utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",

7
tensorflow/python/framework/experimental/BUILD
View File

@ -80,12 +80,14 @@ tf_python_pybind_extension(
py_library( py_library(
name = "gradient_registry", name = "gradient_registry",
srcs = ["gradient_registry.py"], srcs = ["gradient_registry.py"],
srcs_version = "PY3",
deps = [":_tape"], deps = [":_tape"],
) )
py_library( py_library(
name = "math_ops", name = "math_ops",
srcs = ["math_ops.py"], srcs = ["math_ops.py"],
srcs_version = "PY3",
deps = [ deps = [
":_math_ops", ":_math_ops",
":context_stack", ":context_stack",
@ -95,6 +97,7 @@ py_library(
py_library( py_library(
name = "nn_ops", name = "nn_ops",
srcs = ["nn_ops.py"], srcs = ["nn_ops.py"],
srcs_version = "PY3",
deps = [ deps = [
":_nn_ops", ":_nn_ops",
":context_stack", ":context_stack",
@ -104,6 +107,7 @@ py_library(
py_library( py_library(
name = "tape", name = "tape",
srcs = ["tape.py"], srcs = ["tape.py"],
srcs_version = "PY3",
deps = [ deps = [
":_tape", ":_tape",
":context_stack", ":context_stack",
@ -115,16 +119,19 @@ py_library(
py_library( py_library(
name = "def_function", name = "def_function",
srcs = ["def_function.py"], srcs = ["def_function.py"],
srcs_version = "PY3",
) )
py_library( py_library(
name = "thread_local_stack", name = "thread_local_stack",
srcs = ["thread_local_stack.py"], srcs = ["thread_local_stack.py"],
srcs_version = "PY3",
) )
py_library( py_library(
name = "context_stack", name = "context_stack",
srcs = ["context_stack.py"], srcs = ["context_stack.py"],
srcs_version = "PY3",
deps = [":thread_local_stack"], deps = [":thread_local_stack"],
) )

32
tensorflow/python/keras/BUILD
View File

@ -25,7 +25,7 @@ py_library(
"estimator/__init__.py", "estimator/__init__.py",
"keras_parameterized.py", "keras_parameterized.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
":engine", ":engine",
@ -53,7 +53,7 @@ py_library(
py_library( py_library(
name = "backend", name = "backend",
srcs = ["backend.py"], srcs = ["backend.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend_config", ":backend_config",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -104,7 +104,7 @@ py_library(
py_library( py_library(
name = "backend_config", name = "backend_config",
srcs = ["backend_config.py"], srcs = ["backend_config.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
], ],
@ -117,7 +117,7 @@ py_library(
":metrics", ":metrics",
":models", ":models",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/keras/engine", "//tensorflow/python/keras/engine",
], ],
@ -128,7 +128,7 @@ py_library(
srcs = [ srcs = [
"activations.py", "activations.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/layers:advanced_activations", "//tensorflow/python/keras/layers:advanced_activations",
@ -140,7 +140,7 @@ py_library(
py_library( py_library(
name = "base_layer", name = "base_layer",
srcs = [], srcs = [],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/keras/engine:base_layer", "//tensorflow/python/keras/engine:base_layer",
], ],
@ -151,7 +151,7 @@ py_library(
srcs = [ srcs = [
"callbacks.py", "callbacks.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/distribute:distributed_file_utils", "//tensorflow/python/keras/distribute:distributed_file_utils",
@ -169,6 +169,7 @@ py_library(
srcs = [ srcs = [
"combinations.py", "combinations.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":testing_utils", ":testing_utils",
"//tensorflow/python:framework_combinations", "//tensorflow/python:framework_combinations",
@ -182,7 +183,7 @@ py_library(
srcs = [ srcs = [
"callbacks_v1.py", "callbacks_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/utils:engine_utils", "//tensorflow/python/keras/utils:engine_utils",
@ -195,7 +196,7 @@ py_library(
srcs = [ srcs = [
"constraints.py", "constraints.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/utils:engine_utils", "//tensorflow/python/keras/utils:engine_utils",
@ -209,7 +210,7 @@ py_library(
"initializers/initializers_v1.py", "initializers/initializers_v1.py",
"initializers/initializers_v2.py", "initializers/initializers_v2.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -232,7 +233,7 @@ py_library(
srcs = [ srcs = [
"losses.py", "losses.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/utils:engine_utils", "//tensorflow/python/keras/utils:engine_utils",
@ -244,7 +245,7 @@ py_library(
srcs = [ srcs = [
"metrics.py", "metrics.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":activations", ":activations",
":backend", ":backend",
@ -284,7 +285,7 @@ py_library(
srcs = [ srcs = [
"models.py", "models.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
":metrics", ":metrics",
@ -305,7 +306,7 @@ py_library(
"optimizer_v1.py", "optimizer_v1.py",
"optimizers.py", "optimizers.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/optimizer_v2", "//tensorflow/python/keras/optimizer_v2",
@ -318,7 +319,7 @@ py_library(
srcs = [ srcs = [
"regularizers.py", "regularizers.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
"//tensorflow/python/keras/utils:engine_utils", "//tensorflow/python/keras/utils:engine_utils",
@ -330,6 +331,7 @@ py_library(
srcs = [ srcs = [
"testing_utils.py", "testing_utils.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":backend", ":backend",
":models", ":models",

2
tensorflow/python/keras/applications/BUILD
View File

@ -36,7 +36,7 @@ py_library(
"vgg19.py", "vgg19.py",
"xception.py", "xception.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:lib", "//tensorflow/python:lib",

4
tensorflow/python/keras/benchmarks/BUILD
View File

@ -30,6 +30,7 @@ filegroup(
# Add all benchmarks related utils here for pip testing dependencis. # Add all benchmarks related utils here for pip testing dependencis.
py_library( py_library(
name = "keras_benchmark_lib_pip", name = "keras_benchmark_lib_pip",
srcs_version = "PY3",
deps = [ deps = [
":benchmark_util", ":benchmark_util",
":distribution_util", ":distribution_util",
@ -40,6 +41,7 @@ py_library(
# This lib is mainly for running benchmarks on mlcompass infra. # This lib is mainly for running benchmarks on mlcompass infra.
py_library( py_library(
name = "profiler_lib", name = "profiler_lib",
srcs_version = "PY3",
visibility = [ visibility = [
"//learning/brain/contrib/keras_benchmark:__subpackages__", "//learning/brain/contrib/keras_benchmark:__subpackages__",
"//tensorflow/python/keras:__subpackages__", "//tensorflow/python/keras:__subpackages__",
@ -93,6 +95,7 @@ cuda_py_test(
py_library( py_library(
name = "benchmark_util", name = "benchmark_util",
srcs = ["benchmark_util.py"], srcs = ["benchmark_util.py"],
srcs_version = "PY3",
deps = [ deps = [
":distribution_util", ":distribution_util",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
@ -225,6 +228,7 @@ cuda_py_test(
py_library( py_library(
name = "distribution_util", name = "distribution_util",
srcs = ["distribution_util.py"], srcs = ["distribution_util.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
], ],

2
tensorflow/python/keras/benchmarks/layer_benchmarks/BUILD
View File

@ -40,12 +40,14 @@ BECHMARK_TAGS = [
py_library( py_library(
name = "run_xprof", name = "run_xprof",
srcs = ["run_xprof.py"], srcs = ["run_xprof.py"],
srcs_version = "PY3",
visibility = ["//tensorflow/python/keras:__subpackages__"], visibility = ["//tensorflow/python/keras:__subpackages__"],
) )
py_library( py_library(
name = "layer_benchmarks_test_base", name = "layer_benchmarks_test_base",
srcs = ["layer_benchmarks_test_base.py"], srcs = ["layer_benchmarks_test_base.py"],
srcs_version = "PY3",
visibility = ["//tensorflow/python/keras:__subpackages__"], visibility = ["//tensorflow/python/keras:__subpackages__"],
deps = [ deps = [
":run_xprof", ":run_xprof",

1
tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD
View File

@ -30,6 +30,7 @@ filegroup(
py_library( py_library(
name = "saved_model_benchmark_util", name = "saved_model_benchmark_util",
srcs = ["saved_model_benchmark_util.py"], srcs = ["saved_model_benchmark_util.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//tensorflow/python/keras/benchmarks:profiler_lib", "//tensorflow/python/keras/benchmarks:profiler_lib",

2
tensorflow/python/keras/datasets/BUILD
View File

@ -27,7 +27,7 @@ py_library(
"mnist.py", "mnist.py",
"reuters.py", "reuters.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:platform", "//tensorflow/python:platform",

21
tensorflow/python/keras/distribute/BUILD
View File

@ -29,7 +29,7 @@ py_library(
"distributed_training_utils.py", "distributed_training_utils.py",
"distributed_training_utils_v1.py", "distributed_training_utils_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":sidecar_evaluator", ":sidecar_evaluator",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -62,6 +62,7 @@ py_library(
py_library( py_library(
name = "distribute_test_lib_pip", name = "distribute_test_lib_pip",
srcs_version = "PY3",
deps = [ deps = [
":distribute_strategy_test_lib", ":distribute_strategy_test_lib",
":keras_correctness_test_lib", ":keras_correctness_test_lib",
@ -76,7 +77,7 @@ py_library(
py_library( py_library(
name = "optimizer_combinations", name = "optimizer_combinations",
srcs = ["optimizer_combinations.py"], srcs = ["optimizer_combinations.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:training", "//tensorflow/python:training",
"//tensorflow/python/distribute:combinations", "//tensorflow/python/distribute:combinations",
@ -90,7 +91,7 @@ py_library(
srcs = [ srcs = [
"worker_training_state.py", "worker_training_state.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":distributed_file_utils", ":distributed_file_utils",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -107,12 +108,13 @@ py_library(
py_library( py_library(
name = "model_collection_base", name = "model_collection_base",
srcs = ["model_collection_base.py"], srcs = ["model_collection_base.py"],
srcs_version = "PY3",
) )
py_library( py_library(
name = "model_combinations", name = "model_combinations",
srcs = ["model_combinations.py"], srcs = ["model_combinations.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":simple_models", ":simple_models",
"//tensorflow/python/distribute:combinations", "//tensorflow/python/distribute:combinations",
@ -122,6 +124,7 @@ py_library(
py_library( py_library(
name = "simple_models", name = "simple_models",
srcs = ["simple_models.py"], srcs = ["simple_models.py"],
srcs_version = "PY3",
deps = [ deps = [
":model_collection_base", ":model_collection_base",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -133,6 +136,7 @@ py_library(
py_library( py_library(
name = "saved_model_test_base", name = "saved_model_test_base",
srcs = ["saved_model_test_base.py"], srcs = ["saved_model_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":model_combinations", ":model_combinations",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -316,6 +320,7 @@ py_library(
srcs = [ srcs = [
"distribute_strategy_test.py", "distribute_strategy_test.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":optimizer_combinations", ":optimizer_combinations",
":strategy_combinations", ":strategy_combinations",
@ -404,6 +409,7 @@ py_library(
"keras_rnn_model_correctness_test.py", "keras_rnn_model_correctness_test.py",
"keras_stateful_lstm_model_correctness_test.py", "keras_stateful_lstm_model_correctness_test.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":strategy_combinations", ":strategy_combinations",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -612,6 +618,7 @@ py_library(
srcs = [ srcs = [
"keras_utils_test.py", "keras_utils_test.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python:training", "//tensorflow/python:training",
@ -775,6 +782,7 @@ py_library(
srcs = [ srcs = [
"multi_worker_testing_utils.py", "multi_worker_testing_utils.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -788,7 +796,7 @@ py_library(
py_library( py_library(
name = "tpu_strategy_test_utils", name = "tpu_strategy_test_utils",
srcs = ["tpu_strategy_test_utils.py"], srcs = ["tpu_strategy_test_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:platform", "//tensorflow/python:platform",
"//tensorflow/python/distribute:tpu_strategy", "//tensorflow/python/distribute:tpu_strategy",
@ -870,6 +878,7 @@ py_library(
srcs = [ srcs = [
"distributed_file_utils.py", "distributed_file_utils.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:lib", "//tensorflow/python:lib",
"//tensorflow/python/distribute:distribute_lib", "//tensorflow/python/distribute:distribute_lib",
@ -934,6 +943,7 @@ tf_py_test(
py_library( py_library(
name = "strategy_combinations", name = "strategy_combinations",
srcs = ["strategy_combinations.py"], srcs = ["strategy_combinations.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/distribute:strategy_combinations", "//tensorflow/python/distribute:strategy_combinations",
], ],
@ -942,6 +952,7 @@ py_library(
py_library( py_library(
name = "test_example", name = "test_example",
srcs = ["test_example.py"], srcs = ["test_example.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",

16
tensorflow/python/keras/engine/BUILD
View File

@ -42,7 +42,7 @@ py_library(
"training_utils_v1.py", "training_utils_v1.py",
"training_v1.py", "training_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_layer", ":base_layer",
":base_preprocessing_layer", ":base_preprocessing_layer",
@ -95,7 +95,7 @@ py_library(
py_library( py_library(
name = "base_layer_utils", name = "base_layer_utils",
srcs = ["base_layer_utils.py"], srcs = ["base_layer_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:auto_control_deps", "//tensorflow/python:auto_control_deps",
@ -119,7 +119,7 @@ py_library(
"base_layer.py", "base_layer.py",
"base_layer_v1.py", "base_layer_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_layer_utils", ":base_layer_utils",
":input_spec", ":input_spec",
@ -179,7 +179,7 @@ py_library(
py_library( py_library(
name = "data_adapter", name = "data_adapter",
srcs = ["data_adapter.py"], srcs = ["data_adapter.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -192,7 +192,7 @@ py_library(
py_library( py_library(
name = "input_spec", name = "input_spec",
srcs = ["input_spec.py"], srcs = ["input_spec.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:lib", "//tensorflow/python:lib",
@ -206,7 +206,7 @@ py_library(
py_library( py_library(
name = "keras_tensor", name = "keras_tensor",
srcs = ["keras_tensor.py"], srcs = ["keras_tensor.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:lib", "//tensorflow/python:lib",
@ -222,7 +222,7 @@ py_library(
"base_preprocessing_layer.py", "base_preprocessing_layer.py",
"base_preprocessing_layer_v1.py", "base_preprocessing_layer_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_layer", ":base_layer",
"//tensorflow/python/data", "//tensorflow/python/data",
@ -235,7 +235,7 @@ py_library(
py_library( py_library(
name = "node", name = "node",
srcs = ["node.py"], srcs = ["node.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":base_layer_utils", ":base_layer_utils",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",

5
tensorflow/python/keras/feature_column/BUILD
View File

@ -17,6 +17,7 @@ filegroup(
py_library( py_library(
name = "feature_column", name = "feature_column",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY3",
deps = [ deps = [
":base_feature_layer", ":base_feature_layer",
":dense_features", ":dense_features",
@ -28,6 +29,7 @@ py_library(
py_library( py_library(
name = "base_feature_layer", name = "base_feature_layer",
srcs = ["base_feature_layer.py"], srcs = ["base_feature_layer.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:variable_scope", "//tensorflow/python:variable_scope",
@ -42,6 +44,7 @@ py_library(
srcs = [ srcs = [
"dense_features.py", "dense_features.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":base_feature_layer", ":base_feature_layer",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -57,6 +60,7 @@ py_library(
srcs = [ srcs = [
"dense_features_v2.py", "dense_features_v2.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":base_feature_layer", ":base_feature_layer",
":dense_features", ":dense_features",
@ -120,6 +124,7 @@ tf_py_test(
py_library( py_library(
name = "sequence_feature_column", name = "sequence_feature_column",
srcs = ["sequence_feature_column.py"], srcs = ["sequence_feature_column.py"],
srcs_version = "PY3",
deps = [ deps = [
":base_feature_layer", ":base_feature_layer",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

42
tensorflow/python/keras/layers/BUILD
View File

@ -30,7 +30,7 @@ py_library(
"__init__.py", "__init__.py",
"serialization.py", "serialization.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":advanced_activations", ":advanced_activations",
":convolutional", ":convolutional",
@ -62,7 +62,7 @@ py_library(
py_library( py_library(
name = "advanced_activations", name = "advanced_activations",
srcs = ["advanced_activations.py"], srcs = ["advanced_activations.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -79,7 +79,7 @@ py_library(
py_library( py_library(
name = "convolutional", name = "convolutional",
srcs = ["convolutional.py"], srcs = ["convolutional.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":pooling", ":pooling",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -103,7 +103,7 @@ py_library(
py_library( py_library(
name = "convolutional_recurrent", name = "convolutional_recurrent",
srcs = ["convolutional_recurrent.py"], srcs = ["convolutional_recurrent.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":recurrent", ":recurrent",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -125,7 +125,7 @@ py_library(
py_library( py_library(
name = "core", name = "core",
srcs = ["core.py"], srcs = ["core.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -162,7 +162,7 @@ py_library(
py_library( py_library(
name = "cudnn_recurrent", name = "cudnn_recurrent",
srcs = ["cudnn_recurrent.py"], srcs = ["cudnn_recurrent.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":recurrent", ":recurrent",
":recurrent_v2", ":recurrent_v2",
@ -182,7 +182,7 @@ py_library(
py_library( py_library(
name = "dense_attention", name = "dense_attention",
srcs = ["dense_attention.py"], srcs = ["dense_attention.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -201,7 +201,7 @@ py_library(
py_library( py_library(
name = "einsum_dense", name = "einsum_dense",
srcs = ["einsum_dense.py"], srcs = ["einsum_dense.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:special_math_ops", "//tensorflow/python:special_math_ops",
"//tensorflow/python:tensor_shape", "//tensorflow/python:tensor_shape",
@ -217,7 +217,7 @@ py_library(
py_library( py_library(
name = "multi_head_attention", name = "multi_head_attention",
srcs = ["multi_head_attention.py"], srcs = ["multi_head_attention.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:platform", "//tensorflow/python:platform",
"//tensorflow/python:special_math_ops", "//tensorflow/python:special_math_ops",
@ -234,7 +234,7 @@ py_library(
py_library( py_library(
name = "embeddings", name = "embeddings",
srcs = ["embeddings.py"], srcs = ["embeddings.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:embedding_ops", "//tensorflow/python:embedding_ops",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -254,7 +254,7 @@ py_library(
py_library( py_library(
name = "kernelized", name = "kernelized",
srcs = ["kernelized.py"], srcs = ["kernelized.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -273,7 +273,7 @@ py_library(
py_library( py_library(
name = "local", name = "local",
srcs = ["local.py"], srcs = ["local.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/keras:activations", "//tensorflow/python/keras:activations",
@ -292,7 +292,7 @@ py_library(
py_library( py_library(
name = "merge", name = "merge",
srcs = ["merge.py"], srcs = ["merge.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -307,7 +307,7 @@ py_library(
py_library( py_library(
name = "noise", name = "noise",
srcs = ["noise.py"], srcs = ["noise.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -322,7 +322,7 @@ py_library(
py_library( py_library(
name = "normalization", name = "normalization",
srcs = ["normalization.py"], srcs = ["normalization.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -351,7 +351,7 @@ py_library(
py_library( py_library(
name = "normalization_v2", name = "normalization_v2",
srcs = ["normalization_v2.py"], srcs = ["normalization_v2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":normalization", ":normalization",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -361,7 +361,7 @@ py_library(
py_library( py_library(
name = "pooling", name = "pooling",
srcs = ["pooling.py"], srcs = ["pooling.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -378,7 +378,7 @@ py_library(
py_library( py_library(
name = "recurrent", name = "recurrent",
srcs = ["recurrent.py"], srcs = ["recurrent.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:control_flow_ops", "//tensorflow/python:control_flow_ops",
@ -409,7 +409,7 @@ py_library(
py_library( py_library(
name = "recurrent_v2", name = "recurrent_v2",
srcs = ["recurrent_v2.py"], srcs = ["recurrent_v2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":recurrent", ":recurrent",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -436,7 +436,7 @@ py_library(
py_library( py_library(
name = "rnn_cell_wrapper_v2", name = "rnn_cell_wrapper_v2",
srcs = ["rnn_cell_wrapper_v2.py"], srcs = ["rnn_cell_wrapper_v2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":recurrent", ":recurrent",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -447,7 +447,7 @@ py_library(
py_library( py_library(
name = "wrappers", name = "wrappers",
srcs = ["wrappers.py"], srcs = ["wrappers.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":recurrent", ":recurrent",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

2
tensorflow/python/keras/layers/legacy_rnn/BUILD
View File

@ -18,6 +18,7 @@ filegroup(
py_library( py_library(
name = "rnn_cell_impl", name = "rnn_cell_impl",
srcs = ["rnn_cell_impl.py"], srcs = ["rnn_cell_impl.py"],
srcs_version = "PY3",
deps = [ deps = [
":rnn_cell_wrapper_impl", ":rnn_cell_wrapper_impl",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -50,6 +51,7 @@ py_library(
py_library( py_library(
name = "rnn_cell_wrapper_impl", name = "rnn_cell_wrapper_impl",
srcs = ["rnn_cell_wrapper_impl.py"], srcs = ["rnn_cell_wrapper_impl.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",

2
tensorflow/python/keras/layers/ops/BUILD
View File

@ -17,7 +17,7 @@ filegroup(
py_library( py_library(
name = "core", name = "core",
srcs = ["core.py"], srcs = ["core.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:sparse_ops", "//tensorflow/python:sparse_ops",

30
tensorflow/python/keras/layers/preprocessing/BUILD
View File

@ -27,7 +27,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":category_crossing", ":category_crossing",
":discretization", ":discretization",
@ -49,7 +49,7 @@ py_library(
"discretization.py", "discretization.py",
"discretization_v1.py", "discretization_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:boosted_trees_ops", "//tensorflow/python:boosted_trees_ops",
@ -72,7 +72,7 @@ py_library(
srcs = [ srcs = [
"category_crossing.py", "category_crossing.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -95,7 +95,7 @@ py_library(
srcs = [ srcs = [
"hashing.py", "hashing.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -119,7 +119,7 @@ py_library(
srcs = [ srcs = [
"image_preprocessing.py", "image_preprocessing.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",
@ -150,7 +150,7 @@ py_library(
"index_lookup.py", "index_lookup.py",
"index_lookup_v1.py", "index_lookup_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":category_encoding", ":category_encoding",
":table_utils", ":table_utils",
@ -171,7 +171,7 @@ py_library(
"normalization.py", "normalization.py",
"normalization_v1.py", "normalization_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -193,7 +193,7 @@ py_library(
"integer_lookup.py", "integer_lookup.py",
"integer_lookup_v1.py", "integer_lookup_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":index_lookup", ":index_lookup",
":table_utils", ":table_utils",
@ -208,7 +208,7 @@ py_library(
srcs = [ srcs = [
"table_utils.py", "table_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -231,7 +231,7 @@ py_library(
"text_vectorization.py", "text_vectorization.py",
"text_vectorization_v1.py", "text_vectorization_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":category_encoding", ":category_encoding",
":string_lookup", ":string_lookup",
@ -260,7 +260,7 @@ py_library(
"category_encoding.py", "category_encoding.py",
"category_encoding_v1.py", "category_encoding_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:bincount_ops", "//tensorflow/python:bincount_ops",
@ -288,7 +288,7 @@ py_library(
srcs = [ srcs = [
"reduction.py", "reduction.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -303,7 +303,7 @@ py_library(
"string_lookup.py", "string_lookup.py",
"string_lookup_v1.py", "string_lookup_v1.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":index_lookup", ":index_lookup",
":table_utils", ":table_utils",
@ -318,7 +318,7 @@ py_library(
srcs = [ srcs = [
"preprocessing_stage.py", "preprocessing_stage.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -332,7 +332,7 @@ py_library(
py_library( py_library(
name = "preprocessing_test_utils", name = "preprocessing_test_utils",
srcs = ["preprocessing_test_utils.py"], srcs = ["preprocessing_test_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python:util", "//tensorflow/python:util",

6
tensorflow/python/keras/legacy_tf_layers/BUILD
View File

@ -24,7 +24,7 @@ py_library(
"__init__.py", "__init__.py",
"base.py", "base.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -42,6 +42,7 @@ py_library(
py_library( py_library(
name = "convolutional", name = "convolutional",
srcs = ["convolutional.py"], srcs = ["convolutional.py"],
srcs_version = "PY3",
deps = [ deps = [
":layers_base", ":layers_base",
"//tensorflow/python:init_ops", "//tensorflow/python:init_ops",
@ -53,6 +54,7 @@ py_library(
py_library( py_library(
name = "core", name = "core",
srcs = ["core.py"], srcs = ["core.py"],
srcs_version = "PY3",
deps = [ deps = [
":layers_base", ":layers_base",
"//tensorflow/python:init_ops", "//tensorflow/python:init_ops",
@ -64,6 +66,7 @@ py_library(
py_library( py_library(
name = "normalization", name = "normalization",
srcs = ["normalization.py"], srcs = ["normalization.py"],
srcs_version = "PY3",
deps = [ deps = [
":layers_base", ":layers_base",
"//tensorflow/python:init_ops", "//tensorflow/python:init_ops",
@ -75,6 +78,7 @@ py_library(
py_library( py_library(
name = "pooling", name = "pooling",
srcs = ["pooling.py"], srcs = ["pooling.py"],
srcs_version = "PY3",
deps = [ deps = [
":layers_base", ":layers_base",
"//tensorflow/python:util", "//tensorflow/python:util",

16
tensorflow/python/keras/mixed_precision/BUILD
View File

@ -40,7 +40,7 @@ filegroup(
py_library( py_library(
name = "mixed_precision_experimental", name = "mixed_precision_experimental",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":get_layer_policy", ":get_layer_policy",
":loss_scale_optimizer", ":loss_scale_optimizer",
@ -53,7 +53,7 @@ py_library(
srcs = [ srcs = [
"policy.py", "policy.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":device_compatibility_check", ":device_compatibility_check",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -84,7 +84,7 @@ tf_py_test(
py_library( py_library(
name = "device_compatibility_check", name = "device_compatibility_check",
srcs = ["device_compatibility_check.py"], srcs = ["device_compatibility_check.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:config", "//tensorflow/python:config",
], ],
@ -105,7 +105,7 @@ cuda_py_test(
py_library( py_library(
name = "get_layer_policy", name = "get_layer_policy",
srcs = ["get_layer_policy.py"], srcs = ["get_layer_policy.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/keras/engine:base_layer", "//tensorflow/python/keras/engine:base_layer",
], ],
@ -128,7 +128,7 @@ py_library(
srcs = [ srcs = [
"autocast_variable.py", "autocast_variable.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -164,7 +164,7 @@ tf_py_test(
py_library( py_library(
name = "loss_scale", name = "loss_scale",
srcs = ["loss_scale.py"], srcs = ["loss_scale.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:loss_scale", "//tensorflow/python:loss_scale",
"//tensorflow/python/keras/utils:generic_utils", "//tensorflow/python/keras/utils:generic_utils",
@ -174,7 +174,7 @@ py_library(
py_library( py_library(
name = "loss_scale_optimizer", name = "loss_scale_optimizer",
srcs = ["loss_scale_optimizer.py"], srcs = ["loss_scale_optimizer.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":loss_scale", ":loss_scale",
"//tensorflow/python/distribute:collective_all_reduce_strategy", "//tensorflow/python/distribute:collective_all_reduce_strategy",
@ -248,7 +248,7 @@ cuda_py_test(
py_library( py_library(
name = "test_util", name = "test_util",
srcs = ["test_util.py"], srcs = ["test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework", "//tensorflow/python:framework",
"//tensorflow/python/keras", "//tensorflow/python/keras",

6
tensorflow/python/keras/optimizer_v2/BUILD
View File

@ -34,7 +34,7 @@ py_library(
"rmsprop.py", "rmsprop.py",
"utils.py", "utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":learning_rate_schedule", ":learning_rate_schedule",
"//tensorflow/python:control_flow_ops", "//tensorflow/python:control_flow_ops",
@ -64,7 +64,7 @@ py_library(
srcs = [ srcs = [
"learning_rate_schedule.py", "learning_rate_schedule.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:control_flow_ops", "//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -77,7 +77,7 @@ py_library(
py_library( py_library(
name = "legacy_learning_rate_decay", name = "legacy_learning_rate_decay",
srcs = ["legacy_learning_rate_decay.py"], srcs = ["legacy_learning_rate_decay.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":learning_rate_schedule", ":learning_rate_schedule",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",

2
tensorflow/python/keras/premade/BUILD
View File

@ -22,7 +22,7 @@ py_library(
"linear.py", "linear.py",
"wide_deep.py", "wide_deep.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework", "//tensorflow/python:framework",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",

5
tensorflow/python/keras/preprocessing/BUILD
View File

@ -21,6 +21,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":image", ":image",
":sequence", ":sequence",
@ -37,6 +38,7 @@ py_library(
"image.py", "image.py",
"image_dataset.py", "image_dataset.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/keras:backend", "//tensorflow/python/keras:backend",
@ -50,6 +52,7 @@ py_library(
srcs = [ srcs = [
"sequence.py", "sequence.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/keras/utils:data_utils", "//tensorflow/python/keras/utils:data_utils",
@ -61,6 +64,7 @@ py_library(
srcs = [ srcs = [
"timeseries.py", "timeseries.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:math_ops", "//tensorflow/python:math_ops",
@ -76,6 +80,7 @@ py_library(
"text.py", "text.py",
"text_dataset.py", "text_dataset.py",
], ],
srcs_version = "PY3",
deps = ["//tensorflow/python:util"], deps = ["//tensorflow/python:util"],
) )

4
tensorflow/python/keras/saving/BUILD
View File

@ -45,7 +45,7 @@ py_library(
"saved_model_experimental.py", "saved_model_experimental.py",
"saving_utils.py", "saving_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":load_context", ":load_context",
"//tensorflow/python:lib", "//tensorflow/python:lib",
@ -75,7 +75,7 @@ py_library(
srcs = [ srcs = [
"saved_model/load_context.py", "saved_model/load_context.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [], deps = [],
) )

6
tensorflow/python/keras/tests/BUILD
View File

@ -165,7 +165,7 @@ py_library(
srcs = [ srcs = [
"model_architectures.py", "model_architectures.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/keras", "//tensorflow/python/keras",
], ],
@ -187,7 +187,7 @@ tf_py_test(
py_library( py_library(
name = "model_subclassing_test_util", name = "model_subclassing_test_util",
srcs = ["model_subclassing_test_util.py"], srcs = ["model_subclassing_test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/keras", "//tensorflow/python/keras",
], ],
@ -482,6 +482,6 @@ tf_xla_py_test(
py_library( py_library(
name = "get_config_samples", name = "get_config_samples",
srcs = ["get_config_samples.py"], srcs = ["get_config_samples.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [], deps = [],
) )

34
tensorflow/python/keras/utils/BUILD
View File

@ -24,6 +24,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":all_utils", ":all_utils",
], ],
@ -34,6 +35,7 @@ py_library(
srcs = [ srcs = [
"all_utils.py", "all_utils.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":control_flow_util", ":control_flow_util",
":engine_utils", ":engine_utils",
@ -48,14 +50,14 @@ py_library(
py_library( py_library(
name = "control_flow_util", name = "control_flow_util",
srcs = ["control_flow_util.py"], srcs = ["control_flow_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [], deps = [],
) )
py_library( py_library(
name = "data_utils", name = "data_utils",
srcs = ["data_utils.py"], srcs = ["data_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":generic_utils", ":generic_utils",
":io_utils", ":io_utils",
@ -69,7 +71,7 @@ py_library(
"conv_utils.py", "conv_utils.py",
"losses_utils.py", "losses_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":data_utils", ":data_utils",
":io_utils", ":io_utils",
@ -80,7 +82,7 @@ py_library(
py_library( py_library(
name = "io_utils", name = "io_utils",
srcs = ["io_utils.py"], srcs = ["io_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"@six_archive//:six", "@six_archive//:six",
], ],
@ -89,7 +91,7 @@ py_library(
py_library( py_library(
name = "tf_utils", name = "tf_utils",
srcs = ["tf_utils.py"], srcs = ["tf_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":object_identity", ":object_identity",
"//tensorflow/python:composite_tensor", "//tensorflow/python:composite_tensor",
@ -110,7 +112,7 @@ py_library(
srcs = [ srcs = [
"generic_utils.py", "generic_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":tf_contextlib", ":tf_contextlib",
":tf_inspect", ":tf_inspect",
@ -124,7 +126,7 @@ py_library(
srcs = [ srcs = [
"mode_keys.py", "mode_keys.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/saved_model/model_utils:mode_keys", "//tensorflow/python/saved_model/model_utils:mode_keys",
], ],
@ -136,7 +138,7 @@ py_library(
"kernelized_utils.py", "kernelized_utils.py",
"layer_utils.py", "layer_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":engine_utils", ":engine_utils",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -150,7 +152,7 @@ py_library(
srcs = [ srcs = [
"metrics_utils.py", "metrics_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":generic_utils", ":generic_utils",
":tf_utils", ":tf_utils",
@ -176,7 +178,7 @@ py_library(
srcs = [ srcs = [
"version_utils.py", "version_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -188,7 +190,7 @@ py_library(
srcs = [ srcs = [
"multi_gpu_utils.py", "multi_gpu_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
@ -203,7 +205,7 @@ py_library(
srcs = [ srcs = [
"np_utils.py", "np_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -213,14 +215,14 @@ py_library(
py_library( py_library(
name = "object_identity", name = "object_identity",
srcs = ["object_identity.py"], srcs = ["object_identity.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [], deps = [],
) )
py_library( py_library(
name = "tf_contextlib", name = "tf_contextlib",
srcs = ["tf_contextlib.py"], srcs = ["tf_contextlib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
], ],
@ -229,7 +231,7 @@ py_library(
py_library( py_library(
name = "tf_inspect", name = "tf_inspect",
srcs = ["tf_inspect.py"], srcs = ["tf_inspect.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
], ],
@ -240,7 +242,7 @@ py_library(
srcs = [ srcs = [
"vis_utils.py", "vis_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
], ],

2
tensorflow/python/keras/wrappers/BUILD
View File

@ -20,7 +20,7 @@ py_library(
"__init__.py", "__init__.py",
"scikit_learn.py", "scikit_learn.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//tensorflow/python/keras:engine", "//tensorflow/python/keras:engine",

2
tensorflow/python/kernel_tests/BUILD
View File

@ -367,6 +367,7 @@ cuda_py_test(
py_library( py_library(
name = "cudnn_deterministic_base", name = "cudnn_deterministic_base",
srcs = ["cudnn_deterministic_base.py"], srcs = ["cudnn_deterministic_base.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",
@ -1736,6 +1737,7 @@ cuda_py_test(
py_library( py_library(
name = "bias_op_base", name = "bias_op_base",
srcs = ["bias_op_base.py"], srcs = ["bias_op_base.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

9
tensorflow/python/kernel_tests/proto/BUILD
View File

@ -59,6 +59,7 @@ py_library(
name = "proto_op_test_base", name = "proto_op_test_base",
testonly = 1, testonly = 1,
srcs = ["proto_op_test_base.py"], srcs = ["proto_op_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":test_example_proto_py", ":test_example_proto_py",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -69,6 +70,7 @@ py_library(
name = "decode_proto_op_test_base", name = "decode_proto_op_test_base",
testonly = 1, testonly = 1,
srcs = ["decode_proto_op_test_base.py"], srcs = ["decode_proto_op_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":proto_op_test_base", ":proto_op_test_base",
":test_example_proto_py", ":test_example_proto_py",
@ -81,6 +83,7 @@ py_library(
name = "encode_proto_op_test_base", name = "encode_proto_op_test_base",
testonly = 1, testonly = 1,
srcs = ["encode_proto_op_test_base.py"], srcs = ["encode_proto_op_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":proto_op_test_base", ":proto_op_test_base",
":test_example_proto_py", ":test_example_proto_py",
@ -89,7 +92,10 @@ py_library(
], ],
) )
py_library(name = "py_test_deps") py_library(
name = "py_test_deps",
srcs_version = "PY3",
)
tf_proto_library( tf_proto_library(
name = "test_example_proto", name = "test_example_proto",
@ -110,6 +116,7 @@ py_library(
name = "descriptor_source_test_base", name = "descriptor_source_test_base",
testonly = 1, testonly = 1,
srcs = ["descriptor_source_test_base.py"], srcs = ["descriptor_source_test_base.py"],
srcs_version = "PY3",
deps = [ deps = [
":proto_op_test_base", ":proto_op_test_base",
"//third_party/py/numpy", "//third_party/py/numpy",

1
tensorflow/python/kernel_tests/random/BUILD
View File

@ -15,6 +15,7 @@ package(
py_library( py_library(
name = "util", name = "util",
srcs = ["util.py"], srcs = ["util.py"],
srcs_version = "PY3",
deps = [ deps = [
"//third_party/py/numpy", "//third_party/py/numpy",
], ],

2
tensorflow/python/kernel_tests/signal/BUILD
View File

@ -11,7 +11,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "test_util", name = "test_util",
srcs = ["test_util.py"], srcs = ["test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/lite/python:interpreter", "//tensorflow/lite/python:interpreter",

2
tensorflow/python/lib/io/BUILD
View File

@ -47,7 +47,7 @@ py_library(
"python_io.py", "python_io.py",
"tf_record.py", "tf_record.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":_pywrap_file_io", ":_pywrap_file_io",
":_pywrap_record_io", ":_pywrap_record_io",

1
tensorflow/python/module/BUILD
View File

@ -10,6 +10,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "module", name = "module",
srcs = ["module.py"], srcs = ["module.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:tf2", "//tensorflow/python:tf2",

4
tensorflow/python/ops/distributions/BUILD
View File

@ -16,7 +16,7 @@ py_library(
"will not receive new features, and will be removed by " + "will not receive new features, and will be removed by " +
"early 2019. You should update all usage of " + "early 2019. You should update all usage of " +
"`tf.distributions` to `tfp.distributions`."), "`tf.distributions` to `tfp.distributions`."),
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":util", ":util",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -38,7 +38,7 @@ py_library(
py_library( py_library(
name = "util", name = "util",
srcs = ["util.py"], srcs = ["util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",

6
tensorflow/python/ops/linalg/BUILD
View File

@ -6,7 +6,7 @@ package(
py_library( py_library(
name = "linalg", name = "linalg",
srcs = glob(["*.py"]), srcs = glob(["*.py"]),
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":linalg_impl", ":linalg_impl",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",
@ -26,7 +26,7 @@ py_library(
py_library( py_library(
name = "linalg_impl", name = "linalg_impl",
srcs = ["linalg_impl.py"], srcs = ["linalg_impl.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":linear_operator_util", ":linear_operator_util",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -40,7 +40,7 @@ py_library(
py_library( py_library(
name = "linear_operator_util", name = "linear_operator_util",
srcs = ["linear_operator_util.py"], srcs = ["linear_operator_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",

2
tensorflow/python/ops/linalg/sparse/BUILD
View File

@ -27,7 +27,7 @@ py_library(
"sparse_csr_matrix_grad.py", "sparse_csr_matrix_grad.py",
"sparse_csr_matrix_ops.py", "sparse_csr_matrix_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":gen_sparse_csr_matrix_ops", ":gen_sparse_csr_matrix_ops",
"//third_party/py/numpy", "//third_party/py/numpy",

2
tensorflow/python/ops/losses/BUILD
View File

@ -16,7 +16,7 @@ py_library(
"losses_impl.py", "losses_impl.py",
"util.py", "util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:confusion_matrix", "//tensorflow/python:confusion_matrix",

2
tensorflow/python/ops/numpy_ops/BUILD
View File

@ -19,7 +19,7 @@ py_library(
"np_random.py", "np_random.py",
"np_utils.py", "np_utils.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

2
tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD
View File

@ -27,6 +27,7 @@ py_binary(
py_library( py_library(
name = "numpy_mlp", name = "numpy_mlp",
srcs = ["numpy_mlp.py"], srcs = ["numpy_mlp.py"],
srcs_version = "PY3",
deps = [ deps = [
"//third_party/py/numpy", "//third_party/py/numpy",
], ],
@ -35,6 +36,7 @@ py_library(
py_library( py_library(
name = "tf_numpy_mlp", name = "tf_numpy_mlp",
srcs = ["tf_numpy_mlp.py"], srcs = ["tf_numpy_mlp.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//tensorflow/python/ops/numpy_ops:numpy", "//tensorflow/python/ops/numpy_ops:numpy",

10
tensorflow/python/ops/parallel_for/BUILD
View File

@ -16,7 +16,7 @@ py_library(
"pfor.py", "pfor.py",
"test_util.py", "test_util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":control_flow_ops", ":control_flow_ops",
":gradients", ":gradients",
@ -50,7 +50,7 @@ py_library(
py_library( py_library(
name = "pfor_lib", name = "pfor_lib",
srcs = ["pfor.py"], srcs = ["pfor.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/compiler/tf2xla/python:xla", "//tensorflow/compiler/tf2xla/python:xla",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -76,7 +76,7 @@ py_library(
py_library( py_library(
name = "control_flow_ops", name = "control_flow_ops",
srcs = ["control_flow_ops.py"], srcs = ["control_flow_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":pfor_lib", ":pfor_lib",
@ -98,7 +98,7 @@ py_library(
py_library( py_library(
name = "test_util", name = "test_util",
srcs = ["test_util.py"], srcs = ["test_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":pfor_lib", ":pfor_lib",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -189,7 +189,7 @@ cuda_py_test(
py_library( py_library(
name = "gradients", name = "gradients",
srcs = ["gradients.py"], srcs = ["gradients.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":control_flow_ops", ":control_flow_ops",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

51
tensorflow/python/ops/ragged/BUILD
View File

@ -15,7 +15,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "ragged", name = "ragged",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
tags = ["nofixdeps"], tags = ["nofixdeps"],
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
@ -47,7 +47,7 @@ py_library(
py_library( py_library(
name = "ragged_array_ops", name = "ragged_array_ops",
srcs = ["ragged_array_ops.py"], srcs = ["ragged_array_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_functional_ops", ":ragged_functional_ops",
":ragged_math_ops", ":ragged_math_ops",
@ -67,7 +67,7 @@ py_library(
py_library( py_library(
name = "ragged_batch_gather_ops", name = "ragged_batch_gather_ops",
srcs = ["ragged_batch_gather_ops.py"], srcs = ["ragged_batch_gather_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_gather_ops", ":ragged_gather_ops",
":ragged_tensor", ":ragged_tensor",
@ -84,7 +84,7 @@ py_library(
srcs = [ srcs = [
"ragged_batch_gather_with_default_op.py", "ragged_batch_gather_with_default_op.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_dispatch", ":ragged_dispatch",
@ -103,7 +103,7 @@ py_library(
py_library( py_library(
name = "ragged_concat_ops", name = "ragged_concat_ops",
srcs = ["ragged_concat_ops.py"], srcs = ["ragged_concat_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_gather_ops", ":ragged_gather_ops",
@ -120,7 +120,7 @@ py_library(
py_library( py_library(
name = "ragged_conversion_ops", name = "ragged_conversion_ops",
srcs = ["ragged_conversion_ops.py"], srcs = ["ragged_conversion_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_tensor", ":ragged_tensor",
"//tensorflow/python:dtypes", "//tensorflow/python:dtypes",
@ -130,6 +130,7 @@ py_library(
py_library( py_library(
name = "ragged_factory_ops", name = "ragged_factory_ops",
srcs = ["ragged_factory_ops.py"], srcs = ["ragged_factory_ops.py"],
srcs_version = "PY3",
deps = [ deps = [
":ragged_tensor", ":ragged_tensor",
":ragged_tensor_value", ":ragged_tensor_value",
@ -146,7 +147,7 @@ py_library(
py_library( py_library(
name = "ragged_functional_ops", name = "ragged_functional_ops",
srcs = ["ragged_functional_ops.py"], srcs = ["ragged_functional_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_config", ":ragged_config",
":ragged_tensor", ":ragged_tensor",
@ -161,7 +162,7 @@ py_library(
py_library( py_library(
name = "ragged_gather_ops", name = "ragged_gather_ops",
srcs = ["ragged_gather_ops.py"], srcs = ["ragged_gather_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_tensor", ":ragged_tensor",
@ -177,7 +178,7 @@ py_library(
py_library( py_library(
name = "ragged_getitem", name = "ragged_getitem",
srcs = ["ragged_getitem.py"], srcs = ["ragged_getitem.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_gather_ops", ":ragged_gather_ops",
":ragged_math_ops", ":ragged_math_ops",
@ -193,7 +194,7 @@ py_library(
py_library( py_library(
name = "ragged_math_ops", name = "ragged_math_ops",
srcs = ["ragged_math_ops.py"], srcs = ["ragged_math_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_functional_ops", ":ragged_functional_ops",
":ragged_tensor", ":ragged_tensor",
@ -214,7 +215,7 @@ py_library(
py_library( py_library(
name = "ragged_operators", name = "ragged_operators",
srcs = ["ragged_operators.py"], srcs = ["ragged_operators.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_getitem", ":ragged_getitem",
":ragged_tensor", ":ragged_tensor",
@ -226,7 +227,7 @@ py_library(
py_library( py_library(
name = "ragged_ops", name = "ragged_ops",
srcs = ["ragged_ops.py"], srcs = ["ragged_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_batch_gather_ops", ":ragged_batch_gather_ops",
@ -255,7 +256,7 @@ py_library(
py_library( py_library(
name = "ragged_string_ops", name = "ragged_string_ops",
srcs = ["ragged_string_ops.py"], srcs = ["ragged_string_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_math_ops", ":ragged_math_ops",
@ -271,7 +272,7 @@ py_library(
py_library( py_library(
name = "ragged_squeeze_op", name = "ragged_squeeze_op",
srcs = ["ragged_squeeze_op.py"], srcs = ["ragged_squeeze_op.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_tensor", ":ragged_tensor",
":ragged_util", ":ragged_util",
@ -287,13 +288,13 @@ py_library(
py_library( py_library(
name = "ragged_config", name = "ragged_config",
srcs = ["ragged_config.py"], srcs = ["ragged_config.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "row_partition", name = "row_partition",
srcs = ["row_partition.py"], srcs = ["row_partition.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":segment_id_ops", ":segment_id_ops",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -315,7 +316,7 @@ py_library(
py_library( py_library(
name = "ragged_tensor", name = "ragged_tensor",
srcs = ["ragged_tensor.py"], srcs = ["ragged_tensor.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_config", ":ragged_config",
":ragged_tensor_value", ":ragged_tensor_value",
@ -345,7 +346,7 @@ py_library(
py_library( py_library(
name = "ragged_tensor_shape", name = "ragged_tensor_shape",
srcs = ["ragged_tensor_shape.py"], srcs = ["ragged_tensor_shape.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_config", ":ragged_config",
@ -365,7 +366,7 @@ py_library(
py_library( py_library(
name = "ragged_tensor_value", name = "ragged_tensor_value",
srcs = ["ragged_tensor_value.py"], srcs = ["ragged_tensor_value.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -375,7 +376,7 @@ py_library(
py_library( py_library(
name = "ragged_util", name = "ragged_util",
srcs = ["ragged_util.py"], srcs = ["ragged_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",
@ -389,7 +390,7 @@ py_library(
py_library( py_library(
name = "ragged_where_op", name = "ragged_where_op",
srcs = ["ragged_where_op.py"], srcs = ["ragged_where_op.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_concat_ops", ":ragged_concat_ops",
":ragged_functional_ops", ":ragged_functional_ops",
@ -404,7 +405,7 @@ py_library(
py_library( py_library(
name = "segment_id_ops", name = "segment_id_ops",
srcs = ["segment_id_ops.py"], srcs = ["segment_id_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_util", ":ragged_util",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
@ -421,7 +422,7 @@ py_library(
py_library( py_library(
name = "ragged_map_ops", name = "ragged_map_ops",
srcs = ["ragged_map_ops.py"], srcs = ["ragged_map_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_config", ":ragged_config",
":ragged_tensor", ":ragged_tensor",
@ -444,7 +445,7 @@ py_library(
py_library( py_library(
name = "ragged_dispatch", name = "ragged_dispatch",
srcs = ["ragged_dispatch.py"], srcs = ["ragged_dispatch.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":ragged_array_ops", ":ragged_array_ops",
":ragged_batch_gather_ops", ":ragged_batch_gather_ops",
@ -474,7 +475,7 @@ py_library(
py_library( py_library(
name = "ragged_tensor_test_ops", name = "ragged_tensor_test_ops",
srcs = ["ragged_tensor_test_ops.py"], srcs = ["ragged_tensor_test_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:bitwise_ops", "//tensorflow/python:bitwise_ops",

4
tensorflow/python/ops/risc/BUILD
View File

@ -6,7 +6,7 @@ package(
py_library( py_library(
name = "risc_grad", name = "risc_grad",
srcs = ["risc_grad.py"], srcs = ["risc_grad.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
], ],
@ -15,7 +15,7 @@ py_library(
py_library( py_library(
name = "risc_ops", name = "risc_ops",
srcs = ["risc_ops.py"], srcs = ["risc_ops.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:risc_ops_gen", "//tensorflow/python:risc_ops_gen",
], ],

2
tensorflow/python/ops/signal/BUILD
View File

@ -19,7 +19,7 @@ py_library(
"util_ops.py", "util_ops.py",
"window_ops.py", "window_ops.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:constant_op", "//tensorflow/python:constant_op",

4
tensorflow/python/ops/structured/BUILD
View File

@ -15,7 +15,7 @@ exports_files(["LICENSE"])
py_library( py_library(
name = "structured", name = "structured",
srcs = ["__init__.py"], srcs = ["__init__.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
tags = ["nofixdeps"], tags = ["nofixdeps"],
deps = [ deps = [
":structured_array_ops", ":structured_array_ops",
@ -29,6 +29,7 @@ py_library(
"structured_array_ops.py", "structured_array_ops.py",
"structured_tensor.py", "structured_tensor.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:check_ops", "//tensorflow/python:check_ops",
@ -54,6 +55,7 @@ py_library(
srcs = [ srcs = [
"structured_array_ops.py", "structured_array_ops.py",
], ],
srcs_version = "PY3",
deps = [ deps = [
":structured_tensor", ":structured_tensor",
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

13
tensorflow/python/platform/BUILD
View File

@ -30,6 +30,7 @@ tf_py_build_info_genrule(
py_library( py_library(
name = "build_info", name = "build_info",
srcs = ["build_info.py"], srcs = ["build_info.py"],
srcs_version = "PY3",
) )
py_library( py_library(
@ -45,7 +46,7 @@ py_library(
"device_context.py", # In platform_device_context. "device_context.py", # In platform_device_context.
], ],
) + ["build_info.py"], ) + ["build_info.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":build_info", ":build_info",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -62,7 +63,7 @@ py_library(
py_library( py_library(
name = "benchmark", name = "benchmark",
srcs = ["benchmark.py"], srcs = ["benchmark.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":platform", ":platform",
"//tensorflow/python:client", "//tensorflow/python:client",
@ -73,13 +74,13 @@ py_library(
py_library( py_library(
name = "analytics", name = "analytics",
srcs = ["analytics.py"], srcs = ["analytics.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_library( py_library(
name = "device_context", name = "device_context",
srcs = ["device_context.py"], srcs = ["device_context.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:control_flow_ops", "//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -89,7 +90,7 @@ py_library(
py_library( py_library(
name = "test", name = "test",
srcs = ["googletest.py"], srcs = ["googletest.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":benchmark", ":benchmark",
"@absl_py//absl/testing:absltest", "@absl_py//absl/testing:absltest",
@ -184,7 +185,7 @@ tf_python_pybind_extension(
py_library( py_library(
name = "client_testlib", name = "client_testlib",
srcs = ["test.py"], srcs = ["test.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":test", ":test",
"//tensorflow/python:client", "//tensorflow/python:client",

20
tensorflow/python/profiler/BUILD
View File

@ -9,7 +9,7 @@ package(
py_library( py_library(
name = "profiler", name = "profiler",
srcs = ["profiler.py"], srcs = ["profiler.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":model_analyzer", ":model_analyzer",
@ -23,7 +23,7 @@ py_library(
py_library( py_library(
name = "profiler_client", name = "profiler_client",
srcs = ["profiler_client.py"], srcs = ["profiler_client.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:c_api_util", "//tensorflow/python:c_api_util",
"//tensorflow/python:util", "//tensorflow/python:util",
@ -45,7 +45,7 @@ cuda_py_test(
py_library( py_library(
name = "profiler_v2", name = "profiler_v2",
srcs = ["profiler_v2.py"], srcs = ["profiler_v2.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//tensorflow:internal"], visibility = ["//tensorflow:internal"],
deps = [ deps = [
"//tensorflow/python:errors", "//tensorflow/python:errors",
@ -74,7 +74,7 @@ cuda_py_test(
py_library( py_library(
name = "option_builder", name = "option_builder",
srcs = ["option_builder.py"], srcs = ["option_builder.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":tfprof_logger", ":tfprof_logger",
], ],
@ -83,7 +83,7 @@ py_library(
py_library( py_library(
name = "model_analyzer", name = "model_analyzer",
srcs = ["model_analyzer.py"], srcs = ["model_analyzer.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":option_builder", ":option_builder",
":tfprof_logger", ":tfprof_logger",
@ -137,7 +137,7 @@ cuda_py_test(
py_library( py_library(
name = "tfprof_logger", name = "tfprof_logger",
srcs = ["tfprof_logger.py"], srcs = ["tfprof_logger.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core/profiler:protos_all_py", "//tensorflow/core/profiler:protos_all_py",
"//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:framework_for_generated_wrappers",
@ -167,7 +167,7 @@ tf_py_test(
py_library( py_library(
name = "profile_context", name = "profile_context",
srcs = ["profile_context.py"], srcs = ["profile_context.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":model_analyzer", ":model_analyzer",
":tfprof_logger", ":tfprof_logger",
@ -197,7 +197,7 @@ cuda_py_test(
py_library( py_library(
name = "pprof_profiler", name = "pprof_profiler",
srcs = ["pprof_profiler.py"], srcs = ["pprof_profiler.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = ["@com_google_pprof//:pprof_proto_py"], deps = ["@com_google_pprof//:pprof_proto_py"],
) )
@ -223,7 +223,7 @@ py_test(
py_library( py_library(
name = "trace", name = "trace",
srcs = ["trace.py"], srcs = ["trace.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//perftools/accelerators/xprof/xprofilez/integration_tests:__pkg__", "//perftools/accelerators/xprof/xprofilez/integration_tests:__pkg__",
"//tensorflow:internal", "//tensorflow:internal",
@ -237,7 +237,7 @@ py_library(
py_library( py_library(
name = "traceme", name = "traceme",
srcs = ["traceme.py"], srcs = ["traceme.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":trace", ":trace",
], ],

1
tensorflow/python/profiler/integration_test/BUILD
View File

@ -8,6 +8,7 @@ package(
py_library( py_library(
name = "mnist_testing_utils", name = "mnist_testing_utils",
srcs = ["mnist_testing_utils.py"], srcs = ["mnist_testing_utils.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
"//tensorflow/python:extra_py_tests_deps", "//tensorflow/python:extra_py_tests_deps",

4
tensorflow/python/profiler/internal/BUILD
View File

@ -12,7 +12,7 @@ package(
py_library( py_library(
name = "flops_registry", name = "flops_registry",
srcs = ["flops_registry.py"], srcs = ["flops_registry.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python:framework_ops", "//tensorflow/python:framework_ops",
"//tensorflow/python:graph_util", "//tensorflow/python:graph_util",
@ -22,7 +22,7 @@ py_library(
py_library( py_library(
name = "model_analyzer_testlib", name = "model_analyzer_testlib",
srcs = ["model_analyzer_testlib.py"], srcs = ["model_analyzer_testlib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",

2
tensorflow/python/saved_model/BUILD
View File

@ -386,7 +386,7 @@ py_library(
srcs = [ srcs = [
"load.py", "load.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":function_deserialization", ":function_deserialization",
":load_options", ":load_options",

2
tensorflow/python/tf_program/BUILD
View File

@ -3,6 +3,7 @@ package(licenses = ["notice"])
py_library( py_library(
name = "pywrap_tfd", name = "pywrap_tfd",
srcs = ["pywrap_tfd.py"], srcs = ["pywrap_tfd.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/compiler/mlir/python/mlir_wrapper", "//tensorflow/compiler/mlir/python/mlir_wrapper",
], ],
@ -11,6 +12,7 @@ py_library(
py_library( py_library(
name = "mlir_gen", name = "mlir_gen",
srcs = ["mlir_gen.py"], srcs = ["mlir_gen.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":pywrap_tfd", ":pywrap_tfd",

27
tensorflow/python/tools/BUILD
View File

@ -25,6 +25,7 @@ py_library(
# Include the TF upgrade script to users can run it directly after install TF # Include the TF upgrade script to users can run it directly after install TF
"//tensorflow/tools/compatibility:tf_upgrade_v2", "//tensorflow/tools/compatibility:tf_upgrade_v2",
], ],
srcs_version = "PY3",
deps = [ deps = [
":saved_model_aot_compile", ":saved_model_aot_compile",
":saved_model_utils", ":saved_model_utils",
@ -41,7 +42,7 @@ py_library(
py_library( py_library(
name = "saved_model_utils", name = "saved_model_utils",
srcs = ["saved_model_utils.py"], srcs = ["saved_model_utils.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_test( py_test(
@ -62,7 +63,7 @@ py_test(
py_library( py_library(
name = "freeze_graph_lib", name = "freeze_graph_lib",
srcs = ["freeze_graph.py"], srcs = ["freeze_graph.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":saved_model_utils", ":saved_model_utils",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -90,7 +91,7 @@ py_binary(
py_library( py_library(
name = "freeze_graph_main_lib", name = "freeze_graph_main_lib",
srcs = ["freeze_graph.py"], srcs = ["freeze_graph.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":freeze_graph_lib", ":freeze_graph_lib",
], ],
@ -107,7 +108,7 @@ py_binary(
py_library( py_library(
name = "import_pb_to_tensorboard_lib", name = "import_pb_to_tensorboard_lib",
srcs = ["import_pb_to_tensorboard.py"], srcs = ["import_pb_to_tensorboard.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python", "//tensorflow/python",
@ -151,7 +152,7 @@ py_binary(
py_library( py_library(
name = "inspect_checkpoint_lib", name = "inspect_checkpoint_lib",
srcs = ["inspect_checkpoint.py"], srcs = ["inspect_checkpoint.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python", # TODO(b/34059704): remove when fixed "//tensorflow/python", # TODO(b/34059704): remove when fixed
"//tensorflow/python:platform", "//tensorflow/python:platform",
@ -162,7 +163,7 @@ py_library(
py_library( py_library(
name = "strip_unused_lib", name = "strip_unused_lib",
srcs = ["strip_unused_lib.py"], srcs = ["strip_unused_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
"//tensorflow/python:framework", "//tensorflow/python:framework",
@ -173,7 +174,7 @@ py_library(
py_library( py_library(
name = "module_util", name = "module_util",
srcs = ["module_util.py"], srcs = ["module_util.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
) )
py_binary( py_binary(
@ -211,7 +212,7 @@ py_test(
py_library( py_library(
name = "optimize_for_inference_lib", name = "optimize_for_inference_lib",
srcs = ["optimize_for_inference_lib.py"], srcs = ["optimize_for_inference_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":strip_unused_lib", ":strip_unused_lib",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -234,7 +235,7 @@ py_binary(
py_library( py_library(
name = "optimize_for_inference_main_lib", name = "optimize_for_inference_main_lib",
srcs = ["optimize_for_inference.py"], srcs = ["optimize_for_inference.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":optimize_for_inference_lib", ":optimize_for_inference_lib",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -270,7 +271,7 @@ py_test(
py_library( py_library(
name = "selective_registration_header_lib", name = "selective_registration_header_lib",
srcs = ["selective_registration_header_lib.py"], srcs = ["selective_registration_header_lib.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python", # TODO(b/34059704): remove when fixed "//tensorflow/python", # TODO(b/34059704): remove when fixed
@ -291,7 +292,7 @@ py_binary(
py_library( py_library(
name = "print_selective_registration_header_lib", name = "print_selective_registration_header_lib",
srcs = ["print_selective_registration_header.py"], srcs = ["print_selective_registration_header.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
":selective_registration_header_lib", ":selective_registration_header_lib",
@ -322,7 +323,7 @@ py_binary(
py_library( py_library(
name = "saved_model_cli_lib", name = "saved_model_cli_lib",
srcs = ["saved_model_cli.py"], srcs = ["saved_model_cli.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
# Note: if you make any changes here, make corresponding changes to the # Note: if you make any changes here, make corresponding changes to the
# deps of the "tools_pip" target in this file. Otherwise release builds # deps of the "tools_pip" target in this file. Otherwise release builds
@ -338,7 +339,7 @@ py_library(
py_library( py_library(
name = "saved_model_aot_compile", name = "saved_model_aot_compile",
srcs = ["saved_model_aot_compile.py"], srcs = ["saved_model_aot_compile.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python", "//tensorflow/python",
"//tensorflow/python:tf_optimizer", "//tensorflow/python:tf_optimizer",

4
tensorflow/python/tools/api/generator/BUILD
View File

@ -19,7 +19,7 @@ exports_files(
py_library( py_library(
name = "create_python_api", name = "create_python_api",
srcs = ["//tensorflow/python/tools/api/generator:create_python_api.py"], srcs = ["//tensorflow/python/tools/api/generator:create_python_api.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",
@ -31,7 +31,7 @@ py_library(
py_library( py_library(
name = "doc_srcs", name = "doc_srcs",
srcs = ["doc_srcs.py"], srcs = ["doc_srcs.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/python:util", "//tensorflow/python:util",

10
tensorflow/python/tpu/BUILD
View File

@ -122,7 +122,7 @@ py_library(
"tpu_estimator.py", "tpu_estimator.py",
"util.py", "util.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":async_checkpoint", ":async_checkpoint",
":feature_column", ":feature_column",
@ -155,7 +155,7 @@ py_library(
py_library( py_library(
name = "functional", name = "functional",
srcs = ["functional.py"], srcs = ["functional.py"],
srcs_version = "PY2AND3", srcs_version = "PY3",
visibility = [ visibility = [
"//visibility:public", "//visibility:public",
], ],
@ -180,7 +180,7 @@ py_library(
srcs = [ srcs = [
"__init__.py", "__init__.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":feature_column", ":feature_column",
":feature_column_v2", ":feature_column_v2",
@ -196,7 +196,7 @@ py_library(
"__init__.py", "__init__.py",
"api.py", "api.py",
], ],
srcs_version = "PY2AND3", srcs_version = "PY3",
deps = [ deps = [
":feature_column", ":feature_column",
":feature_column_v2", ":feature_column_v2",
@ -476,6 +476,7 @@ pytype_library(
py_library( py_library(
name = "tpu_name_util", name = "tpu_name_util",
srcs = ["tpu_name_util.py"], srcs = ["tpu_name_util.py"],
srcs_version = "PY3",
deps = [ deps = [
"//tensorflow/python/util:tf_export", "//tensorflow/python/util:tf_export",
], ],
@ -697,6 +698,7 @@ tpu_py_test(
# NOTE this target should only be depended on by the tpu_test_wrapper macro. # NOTE this target should only be depended on by the tpu_test_wrapper macro.
py_library( py_library(
name = "tpu_test_deps", name = "tpu_test_deps",
srcs_version = "PY3",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = ["//tensorflow/python:client_testlib"], deps = ["//tensorflow/python:client_testlib"],
) )

Some files were not shown because too many files have changed in this diff Show More