53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
# Description:
|
|
# JavaScript/TypeScript code generation for TensorFlow.js
|
|
|
|
visibility = [
|
|
"//tensorflow:internal",
|
|
]
|
|
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_cc_test",
|
|
)
|
|
|
|
package(
|
|
default_visibility = visibility,
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
cc_library(
|
|
name = "ts_op_gen",
|
|
srcs = [
|
|
"ops/ts_op_gen.cc",
|
|
],
|
|
hdrs = [
|
|
"ops/ts_op_gen.h",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:op_gen_lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "ts_op_gen_test",
|
|
srcs = [
|
|
"ops/ts_op_gen.cc",
|
|
"ops/ts_op_gen.h",
|
|
"ops/ts_op_gen_test.cc",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:op_gen_lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
],
|
|
)
|