1load("//tensorflow:tensorflow.bzl", "cuda_py_test") 2 3package( 4 default_visibility = ["//visibility:public"], 5 licenses = ["notice"], 6) 7 8py_library( 9 name = "mnist_testing_utils", 10 srcs = ["mnist_testing_utils.py"], 11 srcs_version = "PY3", 12 deps = [ 13 "//tensorflow:tensorflow_py", 14 "//tensorflow/python:extra_py_tests_deps", 15 ], 16) 17 18cuda_py_test( 19 name = "profiler_api_test", 20 srcs = ["profiler_api_test.py"], 21 python_version = "PY3", 22 tags = [ 23 "no_pip", 24 "no_windows", # TODO(b/192257727) 25 ], 26 deps = [ 27 ":mnist_testing_utils", 28 "//tensorflow/python:constant_op", 29 "//tensorflow/python:errors", 30 "//tensorflow/python:framework_test_lib", 31 "//tensorflow/python:platform", 32 "//tensorflow/python/data/ops:dataset_ops", 33 "//tensorflow/python/distribute:collective_all_reduce_strategy", 34 "//tensorflow/python/distribute:multi_process_runner", 35 "//tensorflow/python/distribute:multi_worker_test_base", 36 "//tensorflow/python/eager:test", 37 "//tensorflow/python/profiler:profiler_client", 38 "//tensorflow/python/profiler:profiler_v2", 39 ], 40) 41