github.com/alwaysproblem/mlserving-tutorial@v0.0.0-20221124033215-121cfddbfbf4/TFserving/CustomOp/custom-op/tensorflow_zero_out/BUILD (about)

     1  licenses(["notice"])  # Apache 2.0
     2  
     3  package(default_visibility = ["//visibility:public"])
     4  
     5  cc_binary(
     6      name = 'python/ops/_zero_out_ops.so',
     7      srcs = [
     8          "cc/kernels/zero_out_kernels.cc",
     9          "cc/ops/zero_out_ops.cc",
    10      ],
    11      linkshared = 1,
    12      deps = [
    13          "@local_config_tf//:libtensorflow_framework",
    14          "@local_config_tf//:tf_header_lib",
    15      ],
    16      copts = ["-pthread", "-std=c++17"],
    17  )
    18  
    19  py_library(
    20      name = "zero_out_ops_py",
    21      srcs = ([
    22          "python/ops/zero_out_ops.py",
    23      ]),
    24      data = [
    25          ":python/ops/_zero_out_ops.so"
    26      ],
    27      srcs_version = "PY2AND3",
    28  )
    29  
    30  py_test(
    31      name = "zero_out_ops_py_test",
    32      srcs = [
    33          "python/ops/zero_out_ops_test.py"
    34      ],
    35      main = "python/ops/zero_out_ops_test.py",
    36      deps = [
    37          ":zero_out_ops_py",
    38      ],
    39      srcs_version = "PY2AND3",
    40  )
    41  
    42  py_library(
    43      name = "zero_out_py",
    44      srcs = ([
    45          "__init__.py",
    46          "python/__init__.py",
    47          "python/ops/__init__.py",
    48      ]),
    49      deps = [
    50          ":zero_out_ops_py"
    51      ],
    52      srcs_version = "PY2AND3",
    53  )