Searched refs:explicit_padding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 239 PatternRewriter& rewriter, ArrayAttr& explicit_padding) { in getPaddingValuesFromPadType() argument
269 explicit_padding = rewriter.getI64ArrayAttr(computed_paddings); in getPaddingValuesFromPadType()
308 ArrayAttr& explicit_padding) { in getTransposeConv2dPaddingValues() argument
341 explicit_padding = rewriter.getI64ArrayAttr(computed_paddings); in getTransposeConv2dPaddingValues()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize.td | 54 $strides, $use_cudnn, $padding, $explicit_padding,
62 $strides, $use_cudnn, $padding, $explicit_padding, $data_format,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1570 ArrayRef<Attribute> explicit_padding) { in inferConvReturnTypes() argument
1609 if (explicit_padding.size() == 0) { in inferConvReturnTypes()
1614 if (explicit_padding.size() != num_dims * 2) { in inferConvReturnTypes()
1622 if (llvm::any_of(explicit_padding, is_negative)) in inferConvReturnTypes()
1657 pad_low = get_int(explicit_padding[2 * dim]); in inferConvReturnTypes()
1658 pad_high = get_int(explicit_padding[2 * dim + 1]); in inferConvReturnTypes()
1684 ArrayRef<Attribute> explicit_padding; in inferReturnTypes() local
1690 explicit_padding = explicit_pad.getValue(); in inferReturnTypes()
1693 explicit_padding); in inferReturnTypes()
1881 ArrayRef<Attribute> explicit_padding; in inferReturnTypes() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 167 for (int64 explicit_padding : explicit_paddings) { in operator ()() local
168 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
205 for (int64 explicit_padding : explicit_paddings) { in operator ()() local
206 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4145 for (Attribute explicit_padding : explicit_paddings_attr) in matchAndRewrite() local
4147 explicit_padding.cast<IntegerAttr>().getInt()); in matchAndRewrite()
4328 for (Attribute explicit_padding : explicit_paddings_attr) in matchAndRewrite() local
4330 explicit_padding.cast<IntegerAttr>().getInt()); in matchAndRewrite()
|