Add multiclass multilabel precision and recall metric operations.

Change: 121863903

1 files changed, 74 insertions, 7 deletions

diff --git a/tensorflow/contrib/metrics/BUILD b/tensorflow/contrib/metrics/BUILD
index f090839855..0e345ae4cf 100644
--- a/tensorflow/contrib/metrics/BUILD
+++ b/tensorflow/contrib/metrics/BUILD
@@ -8,20 +8,62 @@ exports_files(["LICENSE"])
 
 package(default_visibility = ["//tensorflow:__subpackages__"])
 
-load("//tensorflow:tensorflow.bzl", "cuda_py_tests")
+load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")
+load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs")
+load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")
+load("//tensorflow:tensorflow.bzl", "tf_kernel_library")
+
+tf_custom_op_library(
+    # TODO(sibyl-Mooth6ku,ptucker): Understand why 'python/ops/_' is needed and fix it.
+    name = "python/ops/_set_ops.so",
+    srcs = [
+        "ops/set_ops.cc",
+    ],
+    deps = [
+        "//tensorflow/contrib/metrics/kernels:set_kernels",
+    ],
+)
+
+tf_gen_op_libs(
+    op_lib_names = ["set_ops"],
+)
+
+tf_gen_op_wrapper_py(
+    name = "set_ops",
+    hidden = [
+        "DenseToDenseSetOperation",
+        "DenseToSparseSetOperation",
+        "SparseToSparseSetOperation",
+        "SetSize",
+    ],
+    deps = ["//tensorflow/contrib/metrics:set_ops_op_lib"],
+)
+
+tf_kernel_library(
+    name = "set_ops_kernels",
+    deps = [
+        "//tensorflow/contrib/metrics/kernels:set_kernels",
+        "//tensorflow/core:framework",
+    ],
+    alwayslink = 1,
+)
 
 py_library(
     name = "metrics_py",
     srcs = ["__init__.py"] + glob(["python/ops/*.py"]) + glob(["python/metrics/*.py"]),
+    data = [":python/ops/_set_ops.so"],
     srcs_version = "PY2AND3",
+    deps = [":set_ops"],
 )
 
-cuda_py_tests(
-    name = "metrics_ops_test",
-    size = "medium",
-    srcs = glob(["python/kernel_tests/*_test.py"]),
-    additional_deps = [
+py_test(
+    name = "set_ops_test",
+    size = "small",
+    srcs = ["python/kernel_tests/set_ops_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
         ":metrics_py",
+        "//tensorflow:tensorflow_py",
         "//tensorflow/python:framework_test_lib",
         "//tensorflow/python:platform_test",
     ],
@@ -40,6 +82,32 @@ py_test(
 )
 
 py_test(
+    name = "confusion_matrix_ops_test",
+    size = "medium",
+    srcs = ["python/kernel_tests/confusion_matrix_ops_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":metrics_py",
+        "//tensorflow:tensorflow_py",
+        "//tensorflow/python:framework_test_lib",
+        "//tensorflow/python:platform_test",
+    ],
+)
+
+py_test(
+    name = "histogram_ops_test",
+    size = "medium",
+    srcs = ["python/kernel_tests/histogram_ops_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":metrics_py",
+        "//tensorflow:tensorflow_py",
+        "//tensorflow/python:framework_test_lib",
+        "//tensorflow/python:platform_test",
+    ],
+)
+
+py_test(
     name = "metric_ops_test",
     srcs = ["python/ops/metric_ops_test.py"],
     srcs_version = "PY2AND3",
@@ -60,5 +128,4 @@ filegroup(
             "**/OWNERS",
         ],
     ),
-    visibility = ["//tensorflow:__subpackages__"],
 )