Home
last modified time | relevance | path

Searched refs:attr_def (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dop_def_library.py52 def _SatisfiesTypeConstraint(dtype, attr_def, param_name): argument
53 if attr_def.HasField("allowed_values"):
54 allowed_list = attr_def.allowed_values.list.type
172 def _MakeType(v, attr_def): argument
177 (attr_def.name, repr(v)))
179 _SatisfiesTypeConstraint(i, attr_def, param_name=attr_def.name)
381 for attr_def in op_def.attr:
382 if attr_def.type != "type":
384 key = attr_def.name
385 if attr_def.HasField("default_value"):
[all …]
Dimporter.py206 for attr_def in op_def.attr:
207 if attr_name == attr_def.name:
208 return attr_def
234 attr_def = _FindAttrInOpDef(key, producer_op_def)
235 if (attr_def and attr_def.HasField('default_value') and
236 node.attr[key] == attr_def.default_value):
390 for attr_def in op_def.attr:
391 key = attr_def.name
392 if attr_def.HasField('default_value'):
395 node_def.attr[key].CopyFrom(attr_def.default_value)
Dmeta_graph.py452 for attr_def in op_def.attr:
453 if attr_def.name == attr_name:
454 if not attr_def.HasField("default_value"):
460 attr_def.default_value.SerializeToString())
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc615 for (const auto& attr_def : op_def->attr()) { in AddDefaultAttrs() local
616 if (attr_def.has_default_value() && !attr_slice.Find(attr_def.name())) { in AddDefaultAttrs()
617 if (!attrs->insert({attr_def.name(), attr_def.default_value()}).second) { in AddDefaultAttrs()
618 return errors::Internal("Somehow duplicated: ", attr_def.name()); in AddDefaultAttrs()
1215 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def, in Create() argument
1224 for (const auto& a : attr_def) b.Attr(a); in Create()
1246 gtl::ArraySlice<string> attr_def, in Define() argument
1252 for (const auto& a : attr_def) b.Attr(a); in Define()
1313 gtl::ArraySlice<string> attr_def, in Define() argument
1315 return Define("_", arg_def, ret_def, attr_def, node_def); in Define()
Dop_def_util_test.cc36 OpDef::AttrDef attr_def; in ADef() local
37 EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &attr_def)); in ADef()
38 return attr_def; in ADef()
Dnode_def_util.cc536 for (const auto& attr_def : op_def.attr()) { in AddDefaultsToNodeDef() local
538 if (attr_def.has_default_value() && !attrs.Find(attr_def.name())) { in AddDefaultsToNodeDef()
539 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def); in AddDefaultsToNodeDef()
Dfunction.h125 gtl::ArraySlice<string> attr_def,
134 gtl::ArraySlice<string> attr_def,
140 gtl::ArraySlice<string> attr_def,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc174 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in RegisterCompilationKernels() local
175 if (attr_def.type() == "type" || attr_def.type() == "list(type)") { in RegisterCompilationKernels()
176 type_attrs.insert(attr_def.name()); in RegisterCompilationKernels()