diff options
author | 2018-01-16 19:02:15 -0800 | |
---|---|---|
committer | 2018-01-16 19:05:41 -0800 | |
commit | baea13831c2d1ffa08c4fcc8944a3870d19826cb (patch) | |
tree | ca046dbc0927c012e0898e0a0bc31044fe7312f2 /tensorflow/c/c_api.h | |
parent | 786dcb0b3ee719ae44ac23ea6ee878bbb04792cd (diff) |
Introduce a new C API entrypoint to set a 'func' attribute on an op
description.
PiperOrigin-RevId: 182146876
Diffstat (limited to 'tensorflow/c/c_api.h')
-rw-r--r-- | tensorflow/c/c_api.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h index 6f1c0606c1..7a3bb47708 100644 --- a/tensorflow/c/c_api.h +++ b/tensorflow/c/c_api.h @@ -511,6 +511,11 @@ TF_CAPI_EXPORT extern void TF_SetAttrTypeList(TF_OperationDescription* desc, const char* attr_name, const TF_DataType* values, int num_values); +// Set a 'func' attribute to the specified name. +// `value` must point to a string of length `length` bytes. +TF_CAPI_EXPORT extern void TF_SetAttrFunc(TF_OperationDescription* desc, + const char* attr_name, + const char* value, size_t length); // Set `num_dims` to -1 to represent "unknown rank". Otherwise, // `dims` points to an array of length `num_dims`. `dims[i]` must be |