116 lines
2.9 KiB
Python
116 lines
2.9 KiB
Python
# Experimental extensions to the C API for eager execution of kernels.
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_cuda_cc_test",
|
|
"tf_cc_test",
|
|
"tf_copts",
|
|
"tf_cuda_library",
|
|
)
|
|
|
|
tf_cuda_library(
|
|
name = "c_api",
|
|
srcs = [
|
|
"c_api.cc",
|
|
"c_api_internal.h",
|
|
],
|
|
hdrs = ["c_api.h"],
|
|
copts = tf_copts(),
|
|
visibility = ["//visibility:public"],
|
|
deps = select({
|
|
"//tensorflow:android": [
|
|
"//tensorflow/core:android_tensorflow_lib_lite",
|
|
],
|
|
"//conditions:default": [
|
|
":runtime",
|
|
"//tensorflow/c:c_api",
|
|
"//tensorflow/c:c_api_internal",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
}),
|
|
)
|
|
|
|
tf_cuda_library(
|
|
name = "c_api_internal",
|
|
hdrs = ["c_api_internal.h"],
|
|
visibility = ["//tensorflow:internal"],
|
|
deps = [
|
|
":c_api",
|
|
":runtime",
|
|
"//tensorflow/c:c_api",
|
|
"//tensorflow/c:c_api_internal",
|
|
"//tensorflow/core:core_cpu_lib",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:framework_lite",
|
|
"//tensorflow/core:lib_internal",
|
|
],
|
|
)
|
|
|
|
tf_cuda_cc_test(
|
|
name = "c_api_test",
|
|
srcs = ["c_api_test.cc"],
|
|
deps = [
|
|
":c_api",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
],
|
|
)
|
|
|
|
tf_cuda_library(
|
|
name = "runtime",
|
|
srcs = ["runtime.cc"],
|
|
hdrs = ["runtime.h"],
|
|
copts = tf_copts(),
|
|
visibility = ["//tensorflow:internal"],
|
|
deps = select({
|
|
"//tensorflow:android": [
|
|
"//tensorflow/core:android_tensorflow_lib_lite",
|
|
],
|
|
"//conditions:default": [
|
|
"//tensorflow/c:c_api",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
}),
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "runtime_test",
|
|
srcs = ["runtime_test.cc"],
|
|
deps = [
|
|
":runtime",
|
|
"//tensorflow/cc:cc_ops",
|
|
"//tensorflow/cc:client_session",
|
|
"//tensorflow/cc:ops",
|
|
"//tensorflow/cc:scope",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "tape",
|
|
hdrs = ["tape.h"],
|
|
visibility = ["//tensorflow:internal"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
],
|
|
)
|