Home
last modified time | relevance | path

Searched refs:named_attr (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc139 static LogicalResult VerifyIndexPath(Operation *op, NamedAttribute named_attr) { in VerifyIndexPath() argument
140 auto attr = named_attr.getValue().dyn_cast<ArrayAttr>(); in VerifyIndexPath()
191 NamedAttribute named_attr) { in verifyRegionArgAttribute() argument
192 if (named_attr.getName() == "tf_saved_model.bound_input") { in verifyRegionArgAttribute()
193 if (!named_attr.getValue().isa<FlatSymbolRefAttr>()) { in verifyRegionArgAttribute()
198 named_attr.getValue().cast<FlatSymbolRefAttr>().getValue(); in verifyRegionArgAttribute()
209 if (named_attr.getName() == "tf_saved_model.index_path") { in verifyRegionArgAttribute()
210 return VerifyIndexPath(op, named_attr); in verifyRegionArgAttribute()
214 << named_attr.getName().getValue() << "'"; in verifyRegionArgAttribute()
219 NamedAttribute named_attr) { in verifyRegionResultAttribute() argument
[all …]
Dtf_saved_model.h32 NamedAttribute named_attr) override;
36 NamedAttribute named_attr) override;
38 NamedAttribute named_attr) override;
/external/tensorflow/tensorflow/compiler/mlir/xla/
Doperator_writer_gen.cc42 const mlir::tblgen::NamedAttribute& named_attr) { in GetDefaultAttrExport() argument
43 Attribute attr = named_attr.attr; in GetDefaultAttrExport()
56 return "Convert_" + named_attr.name.str(); in GetDefaultAttrExport()
111 auto named_attr = arg.get<NamedAttribute*>(); in BuildOperator() local
113 << GetDefaultAttrExport(*named_attr) << "(op." << op.getArgName(index) in BuildOperator()
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Droundtrip.cc36 for (auto& named_attr : (*node->mutable_attr())) { in NormalizeNode()
37 AttrValue& attr_val = named_attr.second; in NormalizeNode()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dlift_as_function_call_utils.cc131 for (const auto &named_attr : op->getAttrs()) { in SetAttributeMap() local
132 attr_to_op_map.insert({named_attr, op}); in SetAttributeMap()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc374 for (const mlir::NamedAttribute& named_attr : attrs) { in ConvertAttributes() local
375 auto name_strref = named_attr.getName().str(); in ConvertAttributes()
376 auto attr = named_attr.getValue(); in ConvertAttributes()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dcorert_converter.cc60 for (auto named_attr : derived_attrs) { in MaterializeDerivedAttributes() local
61 op->setAttr(named_attr.getName(), named_attr.getValue()); in MaterializeDerivedAttributes()
Dlower_saved_model.cc290 for (const auto &named_attr : op->getAttrs()) { in FindCalleesRecursive() local
292 named_attr.getValue().dyn_cast<mlir::FlatSymbolRefAttr>()) { in FindCalleesRecursive()
/external/tensorflow/tensorflow/core/ir/importexport/
Dconvert_attributes.cc237 for (const NamedAttribute& named_attr : attrs) { in ConvertAttributes() local
238 std::string name_str = named_attr.getName().str(); in ConvertAttributes()
239 auto attr = named_attr.getValue(); in ConvertAttributes()
Dfunctiondef_import.cc365 for (const auto& named_attr : it->second.attr()) { in ImportGenericFunction() local
367 ConvertAttributeValue(named_attr.second, builder)); in ImportGenericFunction()
368 arg_attr.append(named_attr.first, attr); in ImportGenericFunction()
Dgraphdef_import.cc258 for (const auto &named_attr : node.attr()) in IsGenericFunction() local
259 if (!named_attr.second.placeholder().empty()) return true; in IsGenericFunction()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc261 for (auto& named_attr : inst->getAttrDictionary()) { in GetOpDescriptionForDebug() local
264 os << named_attr.getName().getValue() << " = "; in GetOpDescriptionForDebug()
265 if (auto element_attr = named_attr.getValue().dyn_cast<ElementsAttr>()) { in GetOpDescriptionForDebug()
272 named_attr.getValue().print(os); in GetOpDescriptionForDebug()
1662 for (const auto& named_attr : dict_attr) { in CreateMetadataVector() local
1663 StringRef name = named_attr.getName(); in CreateMetadataVector()
1664 mlir::Attribute attr = named_attr.getValue(); in CreateMetadataVector()
Dflatbuffer_import.cc783 auto named_attr = in AddOpIntermediatesForLstm() local
785 op_state.addAttribute(named_attr.getName(), named_attr.getValue()); in AddOpIntermediatesForLstm()
/external/rust/crates/nom/
DCHANGELOG.md932 - @keeperofdakeys for fixing eol behaviour, writing documentation and adding `named_attr!`
951 - `named_attr!` creates functions like `named!` but can add attributes like documentation