From b5594e6121e902f8dd2d5127653a1ec5f97daccd Mon Sep 17 00:00:00 2001 From: Mingsheng Hong Date: Fri, 14 Sep 2018 14:15:05 -0700 Subject: Added TFE_OpSetAttrTensor() to eager C API. Also added some experimental C APIs for facilitate the use of eager C APIs in S4TF compiler. PiperOrigin-RevId: 213041780 --- tensorflow/c/eager/c_api.h | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'tensorflow/c/eager/c_api.h') 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, -- cgit v1.2.3