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

[tf.data] Refactor profiling dependencies.

Browse Source 
PiperOrigin-RevId: 341996327
Change-Id: Ifec5a1ddac9b4ec10070b71ea145d0f4d532cda6
This commit is contained in:
Ihor Indyk 2020-11-12 00:55:20 -08:00 committed by TensorFlower Gardener
parent 0c4a56f795
commit 6cf5129a84
2 changed files with 24 additions and 12 deletions
tensorflow/python/data
benchmarks
BUILD
experimental/benchmarks
BUILD

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

@ -7,18 +7,6 @@ package(
exports_files(["LICENSE"])
tf_py_test(
name = "meta_benchmark",
srcs = ["meta_benchmark.py"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
py_library(
name = "benchmark_base",
srcs = ["benchmark_base.py"],
@ -31,6 +19,19 @@ py_library(
],
)
tf_py_test(
name = "meta_benchmark",
srcs = ["meta_benchmark.py"],
deps = [
":benchmark_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "batch_benchmark",
srcs = ["batch_benchmark.py"],

11
tensorflow/python/data/experimental/benchmarks/BUILD
View File

@ -29,6 +29,7 @@ tf_py_test(
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
@ -42,6 +43,7 @@ tf_py_test(
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
@ -72,6 +74,7 @@ tf_py_test(
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/ops:readers",
"//third_party/py/numpy",
@ -90,6 +93,7 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:random_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
@ -106,6 +110,7 @@ tf_py_test(
"//tensorflow/python:functional_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:tensor_spec",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:map_defun",
"//tensorflow/python/eager:function",
],
@ -123,6 +128,7 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:parsing_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
@ -139,6 +145,7 @@ tf_py_test(
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:util",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:matching_files",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
@ -153,6 +160,7 @@ tf_py_test(
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
@ -165,6 +173,7 @@ tf_py_test(
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:interleave_ops",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
@ -178,6 +187,7 @@ tf_py_test(
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:resampling",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
@ -212,6 +222,7 @@ tf_py_test(
"//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",