Move TF_TensorToTensor and TF_TensorFromTensor from c_api_internal.h to
tf_tensor_internal.h since these functions are implemented in tf_tensor.cc and not c_api.cc. PiperOrigin-RevId: 306344465 Change-Id: I1c613ae2fccf86be271ff870b257db13b4e19125
This commit is contained in:
parent
f255fc6854
commit
b0726c4787
@ -186,10 +186,6 @@ struct TF_Server {
|
|||||||
|
|
||||||
namespace tensorflow {
|
namespace tensorflow {
|
||||||
|
|
||||||
Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
|
|
||||||
|
|
||||||
TF_Tensor* TF_TensorFromTensor(const Tensor& src, Status* status);
|
|
||||||
|
|
||||||
Status MessageToBuffer(const tensorflow::protobuf::MessageLite& in,
|
Status MessageToBuffer(const tensorflow::protobuf::MessageLite& in,
|
||||||
TF_Buffer* out);
|
TF_Buffer* out);
|
||||||
|
|
||||||
|
@ -119,6 +119,9 @@ inline Tensor& TensorFromInterface(AbstractTensorInterface* tensor) {
|
|||||||
return down_cast<TensorInterface*>(tensor)->Tensor();
|
return down_cast<TensorInterface*>(tensor)->Tensor();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
|
||||||
|
|
||||||
|
TF_Tensor* TF_TensorFromTensor(const Tensor& src, Status* status);
|
||||||
} // namespace tensorflow
|
} // namespace tensorflow
|
||||||
|
|
||||||
#endif // TENSORFLOW_C_TF_TENSOR_INTERNAL_H_
|
#endif // TENSORFLOW_C_TF_TENSOR_INTERNAL_H_
|
||||||
|
Loading…
Reference in New Issue
Block a user