• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2load("//tensorflow:tensorflow.bzl", "tf_cc_test")
3
4package(
5    licenses = ["notice"],  # Apache 2.0
6)
7
8tf_cc_test(
9    name = "internal_test",
10    srcs = ["internal_test.cc"],
11    deps = [
12        ":internal",
13        "//tensorflow/c/eager:c_api_experimental",
14        "//tensorflow/c/eager:c_api_unified_internal",
15        "//tensorflow/core:test",
16        "//tensorflow/core:test_main",
17        "//tensorflow/core/platform:errors",
18        "@com_google_absl//absl/types:span",
19    ],
20)
21
22cc_library(
23    name = "internal",
24    srcs = ["internal.cc"],
25    hdrs = ["internal.h"],
26    deps = [
27        ":wrapper_operation",
28        "//tensorflow/c:conversion_macros",
29        "//tensorflow/c/eager:abstract_context",
30        "//tensorflow/c/eager:abstract_operation",
31        "//tensorflow/c/eager:abstract_tensor_handle",
32        "//tensorflow/c/eager:c_api_experimental",
33        "//tensorflow/core/platform:refcount",
34        "//tensorflow/core/platform:types",
35    ],
36)
37
38cc_library(
39    name = "wrapper_operation",
40    srcs = ["wrapper_operation.cc"],
41    hdrs = ["wrapper_operation.h"],
42    deps = ["//tensorflow/c/eager:abstract_operation"],
43)
44