wehub-resource-sync
68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_cc_test",
|
|
)
|
|
load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
default_visibility =
|
|
[
|
|
"//tensorflow/python/saved_model:__subpackages__",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "save",
|
|
srcs = ["save.cc"],
|
|
hdrs = ["save.h"],
|
|
deps = [
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core/platform:status",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "save_test",
|
|
size = "small",
|
|
srcs = ["save_test.cc"],
|
|
deps = [
|
|
":save",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core/platform:path",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "load",
|
|
srcs = ["load.cc"],
|
|
hdrs = ["load.h"],
|
|
deps = [
|
|
"//tensorflow/cc/saved_model:constants",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core/platform:env",
|
|
"//tensorflow/core/platform:errors",
|
|
"//tensorflow/core/platform:path",
|
|
"//tensorflow/core/platform:protobuf",
|
|
"//tensorflow/core/platform:statusor",
|
|
"//tensorflow/core/util/tensor_bundle",
|
|
"@com_google_absl//absl/container:flat_hash_map",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "load_test",
|
|
size = "small",
|
|
srcs = ["load_test.cc"],
|
|
deps = [
|
|
":load",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core/platform:path",
|
|
],
|
|
)
|