aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/eager/c_api.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/c/eager/c_api.h')
-rwxr-xr-xtensorflow/c/eager/c_api.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/c/eager/c_api.h b/tensorflow/c/eager/c_api.h
index 337447eec9..a87d73ec8e 100755
--- a/tensorflow/c/eager/c_api.h
+++ b/tensorflow/c/eager/c_api.h
@@ -311,6 +311,11 @@ TF_CAPI_EXPORT extern void TFE_OpSetAttrFunction(TFE_Op* op,
const char* attr_name,
const TFE_Op* value);
+TF_CAPI_EXPORT extern void TFE_OpSetAttrTensor(TFE_Op* op,
+ const char* attr_name,
+ TF_Tensor* tensor,
+ TF_Status* status);
+
TF_CAPI_EXPORT extern void TFE_OpSetAttrStringList(TFE_Op* op,
const char* attr_name,
const void* const* values,