1 package(default_visibility = ["//tensorflow:internal"]) 2 3 licenses(["notice"]) # Apache 2.0 4 5 exports_files(["LICENSE"]) 6 7 load("//tensorflow:tensorflow.bzl", "cuda_py_test") 8 load("//tensorflow:tensorflow.bzl", "py_test") 9 10 py_test( 11 name = "autotune_benchmark", 12 srcs = ["autotune_benchmark.py"], 13 srcs_version = "PY2AND3", 14 deps = [ 15 "//tensorflow/python:client_testlib", 16 "//tensorflow/python:math_ops", 17 "//tensorflow/python:session", 18 "//tensorflow/python/data/experimental/ops:batching", 19 "//tensorflow/python/data/experimental/ops:optimization", 20 "//tensorflow/python/data/ops:dataset_ops", 21 "//third_party/py/numpy", 22 ], 23 ) 24 25 py_test( 26 name = "csv_dataset_benchmark", 27 srcs = ["csv_dataset_benchmark.py"], 28 srcs_version = "PY2AND3", 29 tags = ["no_pip"], 30 deps = [ 31 "//tensorflow/python:client_testlib", 32 "//tensorflow/python:parsing_ops", 33 "//tensorflow/python:platform", 34 "//tensorflow/python:platform_test", 35 "//tensorflow/python:session", 36 "//tensorflow/python/data/experimental/ops:readers", 37 "//tensorflow/python/data/ops:readers", 38 "//third_party/py/numpy", 39 ], 40 ) 41 42 py_test( 43 name = "map_and_batch_benchmark", 44 srcs = ["map_and_batch_benchmark.py"], 45 srcs_version = "PY2AND3", 46 deps = [ 47 "//tensorflow/core:protos_all_py", 48 "//tensorflow/python:array_ops", 49 "//tensorflow/python:client_testlib", 50 "//tensorflow/python:constant_op", 51 "//tensorflow/python:dtypes", 52 "//tensorflow/python:math_ops", 53 "//tensorflow/python:random_ops", 54 "//tensorflow/python:session", 55 "//tensorflow/python/data/experimental/ops:batching", 56 "//tensorflow/python/data/ops:dataset_ops", 57 "//third_party/py/numpy", 58 ], 59 ) 60 61 py_test( 62 name = "map_defun_benchmark", 63 srcs = ["map_defun_benchmark.py"], 64 srcs_version = "PY2AND3", 65 deps = [ 66 "//tensorflow/python:array_ops", 67 "//tensorflow/python:client_testlib", 68 "//tensorflow/python:dtypes", 69 "//tensorflow/python:functional_ops", 70 "//tensorflow/python:math_ops", 71 "//tensorflow/python:tensor_spec", 72 "//tensorflow/python/data/experimental/ops:map_defun", 73 "//tensorflow/python/eager:function", 74 ], 75 ) 76 77 py_test( 78 name = "map_vectorization_benchmark", 79 srcs = ["map_vectorization_benchmark.py"], 80 srcs_version = "PY2AND3", 81 deps = [ 82 "//tensorflow/core:protos_all_py", 83 "//tensorflow/python:array_ops", 84 "//tensorflow/python:client_testlib", 85 "//tensorflow/python:constant_op", 86 "//tensorflow/python:dtypes", 87 "//tensorflow/python:math_ops", 88 "//tensorflow/python:parsing_ops", 89 "//tensorflow/python:session", 90 "//tensorflow/python/data/ops:dataset_ops", 91 "//tensorflow/python/data/util:nest", 92 "//third_party/py/numpy", 93 ], 94 ) 95 96 py_test( 97 name = "matching_files_benchmark", 98 size = "small", 99 srcs = ["matching_files_benchmark.py"], 100 srcs_version = "PY2AND3", 101 deps = [ 102 "//tensorflow/python:array_ops", 103 "//tensorflow/python:client_testlib", 104 "//tensorflow/python:dtypes", 105 "//tensorflow/python:errors", 106 "//tensorflow/python:util", 107 "//tensorflow/python/data/experimental/ops:matching_files", 108 "//tensorflow/python/data/ops:dataset_ops", 109 "//third_party/py/numpy", 110 ], 111 ) 112 113 py_test( 114 name = "choose_fastest_benchmark", 115 srcs = ["choose_fastest_benchmark.py"], 116 srcs_version = "PY2AND3", 117 deps = [ 118 "//tensorflow/python:client_testlib", 119 "//tensorflow/python:framework_ops", 120 "//tensorflow/python:math_ops", 121 "//tensorflow/python:session", 122 "//tensorflow/python/data/ops:dataset_ops", 123 "//third_party/py/numpy", 124 ], 125 ) 126 127 py_test( 128 name = "choose_fastest_branch_benchmark", 129 srcs = ["choose_fastest_branch_benchmark.py"], 130 srcs_version = "PY2AND3", 131 deps = [ 132 "//tensorflow/python:client_testlib", 133 "//tensorflow/python:framework_ops", 134 "//tensorflow/python:math_ops", 135 "//tensorflow/python:session", 136 "//tensorflow/python/data/benchmarks:benchmark_base", 137 "//tensorflow/python/data/ops:dataset_ops", 138 "//third_party/py/numpy", 139 ], 140 ) 141 142 py_test( 143 name = "optimize_benchmark", 144 srcs = ["optimize_benchmark.py"], 145 srcs_version = "PY2AND3", 146 deps = [ 147 "//tensorflow/python:client_testlib", 148 "//tensorflow/python:framework_ops", 149 "//tensorflow/python:math_ops", 150 "//tensorflow/python:session", 151 "//tensorflow/python/data/ops:dataset_ops", 152 "//third_party/py/numpy", 153 ], 154 ) 155 156 py_test( 157 name = "parallel_interleave_benchmark", 158 srcs = ["parallel_interleave_benchmark.py"], 159 srcs_version = "PY2AND3", 160 deps = [ 161 "//tensorflow/python:client_testlib", 162 "//tensorflow/python:math_ops", 163 "//tensorflow/python:session", 164 "//tensorflow/python/data/experimental/ops:interleave_ops", 165 "//tensorflow/python/data/experimental/ops:optimization", 166 "//tensorflow/python/data/experimental/ops:sleep", 167 "//tensorflow/python/data/ops:dataset_ops", 168 "//third_party/py/numpy", 169 ], 170 ) 171 172 py_test( 173 name = "rejection_resample_benchmark", 174 srcs = ["rejection_resample_benchmark.py"], 175 srcs_version = "PY2AND3", 176 tags = ["no_pip"], 177 deps = [ 178 "//tensorflow/python:client_testlib", 179 "//tensorflow/python/data/experimental/ops:resampling", 180 "//tensorflow/python/data/ops:dataset_ops", 181 "//third_party/py/numpy", 182 "@six_archive//:six", 183 ], 184 ) 185 186 py_test( 187 name = "unbatch_benchmark", 188 srcs = ["unbatch_benchmark.py"], 189 srcs_version = "PY2AND3", 190 deps = [ 191 "//tensorflow/python:array_ops", 192 "//tensorflow/python:client_testlib", 193 "//tensorflow/python:dtypes", 194 "//tensorflow/python:framework_ops", 195 "//tensorflow/python:session", 196 "//tensorflow/python/data/experimental/ops:batching", 197 "//tensorflow/python/data/ops:dataset_ops", 198 "//third_party/py/numpy", 199 ], 200 ) 201
