Home | History | Annotate | Download | only in losses 
      1 package(
      2     default_visibility = [
      3         "//engedu/ml/tf_from_scratch:__pkg__",
      4         "//tensorflow:internal",
      5     ],
      6 )
      7 
      8 load("//tensorflow:tensorflow.bzl", "py_test")
      9 
     10 licenses(["notice"])  # Apache 2.0
     11 
     12 py_library(
     13     name = "losses",
     14     srcs = [
     15         "__init__.py",
     16         "losses.py",
     17         "losses_impl.py",
     18         "util.py",
     19     ],
     20     srcs_version = "PY2AND3",
     21     deps = [
     22         "//tensorflow/python:array_ops",
     23         "//tensorflow/python:confusion_matrix",
     24         "//tensorflow/python:control_flow_ops",
     25         "//tensorflow/python:framework_for_generated_wrappers",
     26         "//tensorflow/python:math_ops",
     27         "//tensorflow/python:nn",
     28         "//tensorflow/python:nn_ops",
     29         "//tensorflow/python:platform",
     30         "//tensorflow/python:util",
     31         "//tensorflow/python:weights_broadcast_ops",
     32     ],
     33 )
     34 
     35 py_test(
     36     name = "util_test",
     37     size = "small",
     38     srcs = ["util_test.py"],
     39     srcs_version = "PY2AND3",
     40     deps = [
     41         ":losses",
     42         "//tensorflow/python:client_testlib",
     43         "//tensorflow/python:framework_for_generated_wrappers",
     44     ],
     45 )
     46 
     47 filegroup(
     48     name = "all_files",
     49     srcs = glob(
     50         ["**/*"],
     51         exclude = [
     52             "**/METADATA",
     53             "**/OWNERS",
     54         ],
     55     ),
     56     visibility = ["//tensorflow:__subpackages__"],
     57 )
     58