Searched refs:explicit_pad (Results 1 – 4 of 4) sorted by relevance
82 PatternRewriter& rewriter, ArrayAttr& explicit_pad);86 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,95 ArrayAttr& explicit_pad);
284 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromExplicitPadAttr() argument293 pad_before = explicit_pad[dim * 2].template cast<IntegerAttr>().getInt(); in getPaddingValuesFromExplicitPadAttr()294 pad_after = explicit_pad[dim * 2 + 1].template cast<IntegerAttr>().getInt(); in getPaddingValuesFromExplicitPadAttr()
1685 ArrayAttr explicit_pad = in inferReturnTypes() local1687 if (!explicit_pad) { in inferReturnTypes()1688 explicit_pad = ::mlir::Builder(context).getI64ArrayAttr({}); in inferReturnTypes()1690 explicit_padding = explicit_pad.getValue(); in inferReturnTypes()1882 ArrayAttr explicit_pad = in inferReturnTypes() local1884 if (!explicit_pad) { in inferReturnTypes()1885 explicit_pad = ::mlir::Builder(context).getI64ArrayAttr({}); in inferReturnTypes()1887 explicit_padding = explicit_pad.getValue(); in inferReturnTypes()
138 vector<int64_t> get_padding_values_from_explict_pad_attr(vector<int64_t> explicit_pad,146 pad_before = explicit_pad[dim * 2]147 pad_after = explicit_pad[dim * 2 + 1]