Home
last modified time | relevance | path

Searched refs:padding_attr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dvalidators.h97 auto padding_attr = op->getAttrOfType<StringAttr>("padding"); in TFPaddingIsSameOrValid() local
98 if (padding_attr.getValue() != "SAME" && padding_attr.getValue() != "VALID") in TFPaddingIsSameOrValid()
100 *padding = padding_attr; in TFPaddingIsSameOrValid()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc107 auto& padding_attr = padding_attr_and_idx.value(); in GetRemappedPaddings() local
108 auto padding = padding_attr.dyn_cast<StringAttr>(); in GetRemappedPaddings()
Dtpu_rewrite_pass.cc183 auto& padding_attr = padding_and_idx.value(); in SetMetadataProtoPaddingMap() local
184 auto padding_attr_str = padding_attr.dyn_cast<StringAttr>(); in SetMetadataProtoPaddingMap()
Dlegalize_hlo.cc1182 auto padding_attr = DenseIntElementsAttr::get(attr_type, padding); in ConvertPadOp() local
1183 auto padding_op = rewriter.create<ConstantOp>(loc, attr_type, padding_attr); in ConvertPadOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc386 const tflite::Padding padding_attr = in GetTflitePadding() local
390 if (padding_attr == tflite::Padding_SAME) { in GetTflitePadding()
393 if (padding_attr == tflite::Padding_VALID) { in GetTflitePadding()