1load("//tensorflow:tensorflow.bzl", "py_test") 2 3package( 4 default_visibility = [ 5 "//engedu/ml/tf_from_scratch:__pkg__", 6 "//tensorflow:internal", 7 ], 8 licenses = ["notice"], 9) 10 11py_library( 12 name = "losses", 13 srcs = [ 14 "__init__.py", 15 "losses.py", 16 "losses_impl.py", 17 "util.py", 18 ], 19 srcs_version = "PY3", 20 deps = [ 21 "//tensorflow/python:array_ops", 22 "//tensorflow/python:confusion_matrix", 23 "//tensorflow/python:control_flow_ops", 24 "//tensorflow/python:framework_for_generated_wrappers", 25 "//tensorflow/python:math_ops", 26 "//tensorflow/python:nn", 27 "//tensorflow/python:nn_ops", 28 "//tensorflow/python:platform", 29 "//tensorflow/python:util", 30 "//tensorflow/python:weights_broadcast_ops", 31 ], 32) 33 34py_test( 35 name = "util_test", 36 size = "small", 37 srcs = ["util_test.py"], 38 python_version = "PY3", 39 srcs_version = "PY3", 40 deps = [ 41 ":losses", 42 "//tensorflow/python:client_testlib", 43 "//tensorflow/python:framework_for_generated_wrappers", 44 ], 45) 46