chore: import upstream snapshot with attribution
cffconvert / validate (push) Has been skipped
License Check / license-check (push) Failing after 2s

This commit is contained in:
wehub-resource-sync
2026-07-13 12:14:16 +08:00
commit 8a852e4b4e
36502 changed files with 9277225 additions and 0 deletions
@@ -0,0 +1,150 @@
load("@xla//third_party/rules_python/python:py_binary.bzl", "py_binary")
load(
"//tensorflow/tools/test:performance.bzl",
"tf_py_benchmark_test",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = ["//tensorflow:internal"],
licenses = ["notice"],
)
py_binary(
name = "autotune_benchmark_binary",
srcs = ["autotune_benchmark.py"],
main = "autotune_benchmark.py",
strict_deps = True,
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:options",
"//tensorflow/python/ops:math_ops",
"//third_party/py/numpy",
],
)
tf_py_benchmark_test(
name = "autotune_benchmark",
srcs = ["autotune_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:options",
"//tensorflow/python/ops:math_ops",
"//third_party/py/numpy",
],
)
tf_py_benchmark_test(
name = "parameter_value_benchmark",
srcs = ["parameter_value_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops:math_ops",
"//tensorflow/python/ops:script_ops",
"//third_party/py/numpy",
],
)
tf_py_benchmark_test(
name = "csv_dataset_benchmark",
srcs = ["csv_dataset_benchmark.py"],
tags = ["no_pip"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/ops:parsing_ops",
"//tensorflow/python/platform:gfile",
"//tensorflow/python/platform:test",
],
)
tf_py_benchmark_test(
name = "map_and_batch_benchmark",
srcs = ["map_and_batch_benchmark.py"],
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:options",
"//tensorflow/python/framework:constant_op",
"//tensorflow/python/ops:math_ops",
"//tensorflow/python/ops:random_ops",
"//third_party/py/numpy",
],
)
tf_py_benchmark_test(
name = "map_defun_benchmark",
srcs = ["map_defun_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:map_defun",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/framework:dtypes",
"//tensorflow/python/framework:tensor_spec",
"//tensorflow/python/ops:array_ops",
"//tensorflow/python/ops:map_fn",
"//tensorflow/python/ops:math_ops",
],
)
tf_py_benchmark_test(
name = "matching_files_benchmark",
size = "small",
srcs = ["matching_files_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:matching_files",
],
)
tf_py_benchmark_test(
name = "optimize_benchmark",
srcs = ["optimize_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:options",
"//tensorflow/python/ops:math_ops",
],
)
tf_py_benchmark_test(
name = "rejection_resample_benchmark",
srcs = ["rejection_resample_benchmark.py"],
tags = ["no_pip"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:resampling",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:options",
"//third_party/py/numpy",
],
)
tf_py_benchmark_test(
name = "snapshot_dataset_benchmark",
srcs = ["snapshot_dataset_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:snapshot",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops:array_ops_gen",
"//tensorflow/python/platform:client_testlib",
],
)
tf_py_benchmark_test(
name = "unbatch_benchmark",
srcs = ["unbatch_benchmark.py"],
deps = [
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
],
)