Home
last modified time | relevance | path

Searched refs:kPaddingMapAttr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc43 constexpr char kPaddingMapAttr[] = "padding_map"; variable
93 kPaddingMapAttr, index, arg_type, arg_index) in GetRemappedPaddings()
97 Attribute padding_map_attr = cluster_func->getAttr(kPaddingMapAttr); in GetRemappedPaddings()
103 << "requires '" << kPaddingMapAttr << "' array attribute"; in GetRemappedPaddings()
112 kPaddingMapAttr, padding_attr_and_idx.index())); in GetRemappedPaddings()
119 kPaddingMapAttr, idx, padding.getValue())); in GetRemappedPaddings()
139 kPaddingMapAttr, idx, padding_arg_index)); in GetRemappedPaddings()
Dtpu_rewrite_pass.cc69 constexpr char kPaddingMapAttr[] = "padding_map"; variable
178 auto padding_map = op->getAttrOfType<ArrayAttr>(kPaddingMapAttr); in SetMetadataProtoPaddingMap()
180 return op.emitOpError(CreateMissingAttributeMsg(kPaddingMapAttr)); in SetMetadataProtoPaddingMap()
187 kBadStringArrayElementMsg, kPaddingMapAttr, padding_and_idx.index())); in SetMetadataProtoPaddingMap()
193 kBadArrayElementMsg, kPaddingMapAttr, padding_and_idx.index(), in SetMetadataProtoPaddingMap()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc76 constexpr char kPaddingMapAttr[] = "mhlo.padding_map"; variable
1590 attr_name, kPaddingMapAttr, index) in PaddingMapBadArrayAttrMsg()
1600 kShapeIndicesAttr, kPaddingArgIndicesAttr, kPaddingMapAttr, in PaddingMapMismatchedArraySizeMsg()
1610 element_index, attr_name, kPaddingMapAttr, arg_index) in PaddingMapBadIntAttrMsg()
1619 element_index, attr_name, kPaddingMapAttr, arg_index, max, value) in PaddingMapBadIndexMsg()
1628 element_index, kShapeIndicesAttr, kPaddingMapAttr, arg_index, in PaddingMapNegativeShapeIndexMsg()
1638 kShapeIndicesAttr, kPaddingMapAttr, arg_index, value, in PaddingMapUniqueShapeIndexMsg()
1672 auto padding_map_attr = entry_func.getArgAttr(i, kPaddingMapAttr); in AddDynamicParameterBindings()
1676 return entry_func.emitError() << "requires '" << kPaddingMapAttr in AddDynamicParameterBindings()