1 # Description: 2 # contains parts of TensorFlow that are experimental or unstable and which are not supported. 3 4 licenses(["notice"]) # Apache 2.0 5 6 exports_files(["LICENSE"]) 7 8 package(default_visibility = ["//tensorflow:__subpackages__"]) 9 10 load("//tensorflow:tensorflow.bzl", "py_test") 11 12 py_library( 13 name = "losses_py", 14 srcs = [ 15 "__init__.py", 16 "python/losses/__init__.py", 17 "python/losses/loss_ops.py", 18 "python/metric_learning/metric_loss_ops.py", 19 ], 20 srcs_version = "PY2AND3", 21 deps = [ 22 ":metric_learning_py", 23 "//tensorflow/contrib/framework:framework_py", 24 "//tensorflow/python:array_ops", 25 "//tensorflow/python:control_flow_ops", 26 "//tensorflow/python:framework_for_generated_wrappers", 27 "//tensorflow/python:logging_ops", 28 "//tensorflow/python:math_ops", 29 "//tensorflow/python:nn", 30 "//tensorflow/python:nn_ops", 31 "//tensorflow/python:script_ops", 32 "//tensorflow/python:sparse_ops", 33 "//tensorflow/python:sparse_tensor", 34 "//tensorflow/python:summary", 35 "//tensorflow/python:util", 36 ], 37 ) 38 39 py_test( 40 name = "loss_ops_test", 41 srcs = glob(["python/losses/loss_ops_test.py"]), 42 srcs_version = "PY2AND3", 43 deps = [ 44 ":losses_py", 45 "//tensorflow/contrib/framework:framework_py", 46 "//tensorflow/python:array_ops", 47 "//tensorflow/python:client_testlib", 48 "//tensorflow/python:errors", 49 "//tensorflow/python:framework_for_generated_wrappers", 50 "//tensorflow/python:init_ops", 51 "//tensorflow/python:math_ops", 52 "//tensorflow/python:random_ops", 53 "//tensorflow/python:random_seed", 54 "//tensorflow/python:training", 55 "//tensorflow/python:variable_scope", 56 "//tensorflow/python:variables", 57 "//third_party/py/numpy", 58 ], 59 ) 60 61 py_library( 62 name = "metric_learning_py", 63 srcs = [ 64 "python/metric_learning/__init__.py", 65 "python/metric_learning/metric_loss_ops.py", 66 ], 67 srcs_version = "PY2AND3", 68 deps = [ 69 "//tensorflow/python:array_ops", 70 "//tensorflow/python:control_flow_ops", 71 "//tensorflow/python:framework_for_generated_wrappers", 72 "//tensorflow/python:logging_ops", 73 "//tensorflow/python:math_ops", 74 "//tensorflow/python:nn", 75 "//tensorflow/python:script_ops", 76 "//tensorflow/python:sparse_ops", 77 "//tensorflow/python:sparse_tensor", 78 "//tensorflow/python:summary", 79 "//tensorflow/python:util", 80 ], 81 ) 82 83 py_test( 84 name = "metric_loss_ops_test", 85 size = "medium", 86 srcs = [ 87 "python/metric_learning/metric_loss_ops_test.py", 88 ], 89 shard_count = 4, 90 srcs_version = "PY2AND3", 91 deps = [ 92 ":metric_learning_py", 93 "//tensorflow/python:client_testlib", 94 "//tensorflow/python:framework_for_generated_wrappers", 95 "//tensorflow/python:math_ops", 96 "//tensorflow/python:nn", 97 "//tensorflow/python:sparse_tensor", 98 "//third_party/py/numpy", 99 ], 100 ) 101
