Home
last modified time | relevance | path

Searched refs:FindByString (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc175 const AttrValue* AttrSlice::FindByString(const string& attr_name) const { in FindByString() function in tensorflow::AttrSlice
205 Status AttrSlice::FindByString(const string& attr_name, in FindByString() function in tensorflow::AttrSlice
207 *attr_value = FindByString(attr_name); in FindByString()
498 .FindByString(arg_def.type_attr(), &attr_value)); in AddArgToSig()
504 .FindByString(arg_def.type_list_attr(), &attr_value)); in AddArgToSig()
Dnode_def_util.h157 const AttrValue* FindByString(const std::string& attr_name) const;
162 Status FindByString(const std::string& attr_name,
Dfunction.cc75 const AttrValue* v = attrs.FindByString(arg_def.type_list_attr()); in ArgNumType()
90 const AttrValue* v = attrs.FindByString(arg_def.number_attr()); in ArgNumType()
103 const AttrValue* v = attrs.FindByString(arg_def.type_attr()); in ArgNumType()
124 const AttrValue* attr_value = attr_values.FindByString(attr.name()); in ValidateSignatureWithAttrs()
785 if (const AttrValue* v = attr_values.FindByString(name)) { in InstantiateFunction()
Dop_kernel.cc1362 const AttrValue* attr_value = node_attrs.FindByString(kKernelAttr); in GetKernelLabelAttr()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc66 const AttrValue* attr_value = attr.FindByString("_tftrt_convert_function"); in ShouldConvertFunction()
Dconvert_nodes.cc512 if (attrs.FindByString(dtype_attr_name) != nullptr) { in CreateBroadcastableScalarConstant()
1730 if (attrs.FindByString(type_attr_name) == nullptr) { in GetNodeDefTfType()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_bridge_pass.cc138 auto attr = node->attrs().FindByString(kXlaMustCompile); in HasQualifiedNonTPUOp()