From: A. Unique TensorFlower <gardener@tensorflow.org>
Date: Wed, 17 Jan 2018 03:02:15 +0000 (-0800)
Subject: Introduce a new C API entrypoint to set a 'func' attribute on an op
X-Git-Tag: v1.6.0-rc0~221^2^2~24
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=baea13831c2d1ffa08c4fcc8944a3870d19826cb;p=platform%2Fupstream%2Ftensorflow.git

Introduce a new C API entrypoint to set a 'func' attribute on an op
description.

PiperOrigin-RevId: 182146876
---

diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc
index bc19044fa2..ead36506f0 100644
--- a/tensorflow/c/c_api.cc
+++ b/tensorflow/c/c_api.cc
@@ -1201,6 +1201,13 @@ void TF_SetAttrTypeList(TF_OperationDescription* desc, const char* attr_name,
                      reinterpret_cast<const DataType*>(values), num_values));
 }
 
+void TF_SetAttrFunc(TF_OperationDescription* desc, const char* attr_name,
+                    const char* value, size_t length) {
+  tensorflow::NameAttrList func_name;
+  func_name.set_name(std::string(value, value + length));
+  desc->node_builder.Attr(attr_name, func_name);
+}
+
 void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name,
                      const int64_t* dims, int num_dims) {
   PartialTensorShape shape;
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