experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF] Expose C API in libtensorflow_framework.

Browse Source 
While at it, expose the associated header files from tensorflow/c/ in pip package.

Note, we expose the subset of C API that doesn't require tensorflow/cc linkage;
specifically the core operations that exclude building while loops and gradient ops,
and also excluding the experimental API.

The experimental API can also be added in the future, by factoring it into
"core" and "non-core" targets.  Similarly for the C eager API.

PiperOrigin-RevId: 301601988
Change-Id: I97eac79e684fc42ce90e67ee901cdcf6f7e91cbe
This commit is contained in:
Gunhan Gulsoy 2020-03-18 09:14:38 -07:00 committed by TensorFlower Gardener
parent fe84fb19eb
commit 37dfcf6e2a
12 changed files with 3509 additions and 3739 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/BUILD
View File

@ -644,7 +644,6 @@ tf_cc_shared_object(
"//tensorflow/core:lib_internal_impl",
"//tensorflow/core/profiler:profiler_impl",
"//tensorflow/stream_executor:stream_executor_impl",
"//tensorflow/c:c_core_api_no_xla",
"//tensorflow:tf_framework_version_script.lds",
] + tf_additional_binary_deps(),
)

76
tensorflow/c/BUILD
View File

@ -23,7 +23,6 @@ filegroup(
srcs = [
"c_api.h",
"c_api_experimental.h",
"c_core_api.h",
"tf_attrtype.h",
"tf_datatype.h",
"tf_file_statistics.h",
@ -74,7 +73,6 @@ tf_cuda_library(
hdrs = [
"c_api.h",
"c_api_internal.h",
"c_core_api.h",
"tf_datatype.h",
"tf_tensor.h",
],
@ -118,41 +116,10 @@ cc_library(
visibility = ["//visibility:public"],
)
tf_cuda_library(
name = "c_core_api",
hdrs = [
"c_core_api.h",
"tf_attrtype.h",
"tf_datatype.h",
"tf_file_statistics.h",
"tf_status.h",
"tf_tensor.h",
],
copts = tf_copts(),
visibility = [
"//visibility:public",
],
deps = [
":c_core_api_no_xla",
":c_api_internal",
":tf_attrtype",
":tf_status_internal",
":tf_file_statistics",
":tf_tensor_internal",
] + select({
"//tensorflow:with_xla_support": [
"//tensorflow/compiler/tf2xla:xla_compiler",
"//tensorflow/compiler/jit",
],
"//conditions:default": [],
}),
)
tf_cuda_library(
name = "c_api",
hdrs = [
"c_api.h",
"c_core_api.h",
"tf_attrtype.h",
"tf_datatype.h",
"tf_file_statistics.h",
@ -162,7 +129,6 @@ tf_cuda_library(
copts = tf_copts(),
visibility = ["//visibility:public"],
deps = [
":c_core_api",
":c_api_no_xla",
":c_api_internal",
":tf_attrtype",
@ -178,48 +144,11 @@ tf_cuda_library(
}),
)
tf_cuda_library(
name = "c_core_api_no_xla",
srcs = [
"c_api_function.cc",
"c_core_api.cc",
],
hdrs = [
"c_core_api.h",
],
copts = tf_copts(),
visibility = ["//tensorflow:__subpackages__"],
deps = [
":c_api_internal",
":tf_attrtype",
":tf_datatype",
":tf_status_internal",
] + select({
"//tensorflow:android": [
"//tensorflow/core:android_tensorflow_lib_lite",
],
"//conditions:default": [
":tf_status",
":tf_tensor",
"@com_google_absl//absl/strings",
"//tensorflow/cc/saved_model:loader_lite",
"//tensorflow/core:core_cpu",
"//tensorflow/core:core_cpu_internal",
"//tensorflow/core:framework",
"//tensorflow/core:op_gen_lib",
"//tensorflow/core:protos_all_cc",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core/distributed_runtime:server_lib",
],
}),
alwayslink = 1,
)
tf_cuda_library(
name = "c_api_no_xla",
srcs = [
"c_api.cc",
"c_api_function.cc",
],
hdrs = [
"c_api.h",
@ -230,7 +159,6 @@ tf_cuda_library(
"//third_party/llvm/llvm-project:__subpackages__",
],
deps = [
":c_core_api_no_xla",
":c_api_internal",
":tf_attrtype",
":tf_datatype",
@ -256,6 +184,8 @@ tf_cuda_library(
"//tensorflow/core:protos_all_cc",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core/distributed_runtime:server_lib",
"//tensorflow/core/kernels:logging_ops",
],
}),
alwayslink = 1,

2083
tensorflow/c/c_api.cc
View File

File diff suppressed because it is too large Load Diff

1423
tensorflow/c/c_api.h
View File

File diff suppressed because it is too large Load Diff

8
tensorflow/c/c_api_internal.h
View File

@ -16,14 +16,14 @@ limitations under the License.
#ifndef TENSORFLOW_C_C_API_INTERNAL_H_
#define TENSORFLOW_C_C_API_INTERNAL_H_
#include "tensorflow/c/c_api.h"
#include <list>
#include <set>
#include <string>
#include <unordered_map>
#include <vector>
#include "tensorflow/c/c_core_api.h"
// clang-format off
// Required for IS_MOBILE_PLATFORM
#include "tensorflow/core/platform/platform.h"
@ -217,10 +217,6 @@ bool ExtendSessionGraphHelper(TF_Session* session, TF_Status* status)
std::string getTF_OutputDebugString(TF_Output node);
TF_Operation* ToOperation(Node* node);
TensorId ToTensorId(const TF_Output& output);
} // end namespace tensorflow
#endif // TENSORFLOW_C_C_API_INTERNAL_H_

2193
tensorflow/c/c_core_api.cc
View File

File diff suppressed because it is too large Load Diff

1456
tensorflow/c/c_core_api.h
View File

File diff suppressed because it is too large Load Diff

2
tensorflow/c/eager/BUILD
View File

@ -42,7 +42,7 @@ tf_cuda_library(
"//conditions:default": [
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/container:fixed_array",
"//tensorflow/c:c_core_api",
"//tensorflow/c:c_api",
"//tensorflow/c:c_api_internal",
"//tensorflow/c:tf_tensor_internal",
"//tensorflow/core:core_cpu",

2
tensorflow/c/eager/c_api.cc
View File

@ -28,7 +28,7 @@ limitations under the License.
#include "absl/algorithm/container.h"
#include "absl/memory/memory.h"
#include "tensorflow/c/c_core_api.h"
#include "tensorflow/c/c_api.h"
#include "tensorflow/c/c_api_internal.h"
#include "tensorflow/c/eager/tensor_handle_interface.h"
#include "tensorflow/c/tf_tensor_internal.h"

2
tensorflow/c/eager/c_api.h
View File

@ -20,7 +20,7 @@ limitations under the License.
// WARNING: Unlike tensorflow/c/c_api.h, the API here is not guaranteed to be
// stable and can change without notice.
#include "tensorflow/c/c_core_api.h"
#include "tensorflow/c/c_api.h"
// Macro to control visibility of exported symbols in the shared library (.so,
// .dylib, .dll).

1
tensorflow/tools/pip_package/BUILD
View File

@ -36,7 +36,6 @@ transitive_hdrs(
"//tensorflow/cc/saved_model:loader",
"//tensorflow/cc/saved_model:reader",
"//tensorflow/cc/saved_model:bundle_v2",
"//tensorflow/c:c_core_api_no_xla",
# WARNING: None of the C/C++ code under python/ has any API guarantees, and TF team
# reserves the right to change APIs and other header-level interfaces. If your custom
# op uses these headers, it may break when users upgrade their version of tensorflow.

1
tensorflow/tools/pip_package/setup.py
View File

@ -246,7 +246,6 @@ headers = (
list(find_files('*.proto', 'tensorflow/compiler')) +
list(find_files('*.proto', 'tensorflow/core')) +
list(find_files('*.proto', 'tensorflow/python')) +
list(find_files('*.h', 'tensorflow/c')) +
list(find_files('*.h', 'tensorflow/cc')) +
list(find_files('*.h', 'tensorflow/compiler')) +
list(find_files('*.h', 'tensorflow/core')) +