Skip to content
Snippets Groups Projects
Conv2DNchwFchw.mlir 406 B
Newer Older
Joejiong's avatar
Joejiong committed
// generate from iree processed alexnet or lenet mlir file
func @conv_2d_nchw_fchw(%input: memref<?x?x?x?xf32>, %filter: memref<?x?x?x?xf32>, %output: memref<?x?x?x?xf32>) {
Joejiong's avatar
Joejiong committed
    linalg.conv_2d_nchw_fchw 
    {dilations = dense<1> : tensor<2xi64>, strides = dense<1> : tensor<2xi64>} 
Joejiong's avatar
Joejiong committed
    ins(%input, %filter : memref<?x?x?x?xf32>, memref<?x?x?x?xf32>) 
    outs(%output : memref<?x?x?x?xf32>)
Joejiong's avatar
Joejiong committed
    return
Joejiong's avatar
Joejiong committed
}