Moves TensorCApi from c_api.cc to c_api_internal.h, where it can be used

by other code that require access to the underlying TensorBuffers.

PiperOrigin-RevId: 160323362

1 files changed, 13 insertions, 0 deletions

diff --git a/tensorflow/c/c_api_internal.h b/tensorflow/c/c_api_internal.h
index f17ac26ad9..f2773ae20f 100644
--- a/tensorflow/c/c_api_internal.h
+++ b/tensorflow/c/c_api_internal.h
@@ -117,3 +117,16 @@ struct TF_ImportGraphDefOptions {
 struct TF_DeviceList {
   std::vector<tensorflow::DeviceAttributes> response;
 };
+
+namespace tensorflow {
+
+class TensorCApi {
+ public:
+  static TensorBuffer* Buffer(const Tensor& tensor) { return tensor.buf_; }
+  static Tensor MakeTensor(TF_DataType type, const TensorShape& shape,
+                           TensorBuffer* buf) {
+    return Tensor(static_cast<DataType>(type), shape, buf);
+  }
+};
+
+}  // end namespace tensorflow