Searched refs:tfl_conv_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1094 auto tfl_conv_op = cast<TFL::TransposeConvOp>(op); in matchAndRewrite() local
1096 ShapedType input_type = tfl_conv_op.input().getType().dyn_cast<ShapedType>(); in matchAndRewrite()
1098 tfl_conv_op.weights().getType().dyn_cast<ShapedType>(); in matchAndRewrite()
1100 tfl_conv_op.getResult().getType().dyn_cast<ShapedType>(); in matchAndRewrite()
1124 int64_t stride_h = tfl_conv_op.stride_h(); in matchAndRewrite()
1125 int64_t stride_w = tfl_conv_op.stride_w(); in matchAndRewrite()
1131 if (!GetPaddingFromString(tfl_conv_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
1144 if (matchPattern(tfl_conv_op.output_shape(), in matchAndRewrite()
1196 rewriter, op->getLoc(), output_type.clone(bias_ety), tfl_conv_op.input(), in matchAndRewrite()
1197 tfl_conv_op.weights(), zero_bias.getValue(), outpad, stride, in matchAndRewrite()
|