diff options
Diffstat (limited to 'tensorflow/core/framework/attr_value_util.h')
-rw-r--r-- | tensorflow/core/framework/attr_value_util.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/framework/attr_value_util.h b/tensorflow/core/framework/attr_value_util.h index 08cc3b7158..08d813bb6f 100644 --- a/tensorflow/core/framework/attr_value_util.h +++ b/tensorflow/core/framework/attr_value_util.h @@ -16,8 +16,10 @@ limitations under the License. #ifndef TENSORFLOW_FRAMEWORK_ATTR_VALUE_UTIL_H_ #define TENSORFLOW_FRAMEWORK_ATTR_VALUE_UTIL_H_ +#include <functional> #include <string> #include <vector> + #include "tensorflow/core/framework/attr_value.pb.h" // TODO(62899350): Remove #include "tensorflow/core/framework/partial_tensor_shape.h" #include "tensorflow/core/framework/tensor.h" @@ -100,8 +102,8 @@ bool HasPlaceHolder(const AttrValue& val); // SubstituteFunc is given a placeholder string. If the placeholder is // unknown, SubstituteFunc returns false. Otherwise, overwrites the // attr value and returns true. -typedef std::function<bool(const string&, AttrValue*)> SubstituteFunc; -bool SubstitutePlaceholders(SubstituteFunc substitute, AttrValue* value); +using SubstituteFunc = std::function<bool(const string&, AttrValue*)>; +bool SubstitutePlaceholders(const SubstituteFunc& substitute, AttrValue* value); } // namespace tensorflow |