diff options
Diffstat (limited to 'tensorflow/c/eager/c_api.h')
| -rw-r--r-- | tensorflow/c/eager/c_api.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/c/eager/c_api.h b/tensorflow/c/eager/c_api.h index 476c9288f8..88a0dd343f 100644 --- a/tensorflow/c/eager/c_api.h +++ b/tensorflow/c/eager/c_api.h @@ -43,7 +43,7 @@ extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, // placed in memory of different devices or remote address spaces. typedef struct TFE_TensorHandle TFE_TensorHandle; -extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t); +extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t, TF_Status* status); extern void TFE_DeleteTensorHandle(TFE_TensorHandle* h); extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h); extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h); |