Lines Matching full:linalg

10 #include "mlir/Dialect/Linalg/IR/Linalg.h"
25 namespace linalg {
32 ///   %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
41 ///    %1 = linalg.fill ins(%cst : f32) outs(%0 : tensor<2x2x6x8xf32>) -> tensor<2x2x6x8xf32>
42 ///    %transposed = linalg.transpose ins(%arg1 : tensor<8x2x2x6xf32>) outs(%1 : tensor<2x2x6x8xf32>)
44 ///    %2 = linalg.conv_2d_nhwc_hwcf {dilations = dense<1> : tensor<2xi64>, strides = dense<2> : tensor<2xi64>}
66   // Because linalg.transpose expects an "out" parameter we need to pass it a
87       rewriter.create<linalg::TransposeOp>(loc, filter, input, filterPerm);
128                                        linalg::Conv2DNhwcFhwcOp op) {
130   return transposeConv2DHelper<linalg::Conv2DNhwcFhwcOp,
131                                linalg::Conv2DNhwcHwcfOp>(rewriter, op);
135                                        linalg::Conv2DNhwcFhwcQOp op) {
137   return transposeConv2DHelper<linalg::Conv2DNhwcFhwcQOp,
138                                linalg::Conv2DNhwcHwcfQOp>(rewriter, op);
144       ConvConverter<linalg::Conv2DNhwcFhwcOp, linalg::Conv2DNhwcHwcfOp>,
145       ConvConverter<linalg::Conv2DNhwcFhwcQOp, linalg::Conv2DNhwcHwcfQOp>>(
148 } // namespace linalg