1load("//tensorflow:tensorflow.bzl", "py_tests") 2 3package( 4 default_visibility = ["//tensorflow:internal"], 5 licenses = ["notice"], 6) 7 8py_library( 9 name = "writer", 10 srcs = glob( 11 ["*.py"], 12 exclude = [ 13 "fake*", 14 "*_test.py", 15 ], 16 ), 17 srcs_version = "PY3", 18 visibility = ["//visibility:public"], 19 deps = [ 20 "//tensorflow/core:protos_all_py", 21 "//tensorflow/python:client", 22 "//tensorflow/python:lib", 23 "//tensorflow/python:logging_ops_gen", 24 "//tensorflow/python:platform", 25 "//tensorflow/python:pywrap_tensorflow", 26 "//tensorflow/python:summary_op_util", 27 "//tensorflow/python:summary_ops_gen", 28 "//tensorflow/python:summary_ops_v2", 29 "//tensorflow/python/distribute:summary_op_util", 30 "//tensorflow/python/eager:context", 31 "//tensorflow/python/framework", 32 "//tensorflow/python/framework:constant_op", 33 "//tensorflow/python/framework:for_generated_wrappers", 34 "//tensorflow/python/util", 35 "@six_archive//:six", 36 ], 37) 38 39py_library( 40 name = "fake_summary_writer", 41 testonly = 1, 42 srcs = ["fake_summary_writer.py"], 43 srcs_version = "PY3", 44 visibility = ["//visibility:public"], 45 deps = [ 46 ":writer", 47 "//tensorflow/core:protos_all_py", 48 "//tensorflow/python/framework:test_lib", 49 ], 50) 51 52py_tests( 53 name = "writer_tests", 54 size = "small", 55 srcs = [ 56 "writer_test.py", 57 ], 58 python_version = "PY3", 59 deps = [ 60 ":writer", 61 "//tensorflow/core:protos_all_py", 62 "//tensorflow/python:array_ops", 63 "//tensorflow/python:client_testlib", 64 "//tensorflow/python:platform", 65 "//tensorflow/python:platform_test", 66 "//tensorflow/python:summary_ops_v2", 67 "//tensorflow/python:variables", 68 "//tensorflow/python/framework", 69 "//tensorflow/python/framework:for_generated_wrappers", 70 "//tensorflow/python/framework:test_lib", 71 ], 72) 73