Searched refs:attr_def (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | op_def_library.py | 52 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 …]
|
D | importer.py | 206 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)
|
D | meta_graph.py | 452 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/ |
D | function.cc | 615 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()
|
D | op_def_util_test.cc | 36 OpDef::AttrDef attr_def; in ADef() local 37 EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &attr_def)); in ADef() 38 return attr_def; in ADef()
|
D | node_def_util.cc | 536 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()
|
D | function.h | 125 gtl::ArraySlice<string> attr_def, 134 gtl::ArraySlice<string> attr_def, 140 gtl::ArraySlice<string> attr_def,
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_op_registry.cc | 174 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()
|