[mlir][linalg] Fix depthwise conv C++ symbol to be consistent
authorLei Zhang <antiagainst@google.com>
Tue, 9 Feb 2021 15:04:43 +0000 (10:04 -0500)
committerLei Zhang <antiagainst@google.com>
Tue, 9 Feb 2021 15:06:32 +0000 (10:06 -0500)
The assembly mnemonic includes information about input/filter
data format. The C++ symbol should be consistent.

mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc

index 6764e55..6c6c923 100644 (file)
@@ -92,7 +92,7 @@ def conv_3d_ncdhw(I: f32(N, C, D, H, W), K: f32(F, C, KD, KH, KW)) -> (O: f32(N,
     I(n, c, d + kd, h + kh, w + kw), K(f, c, kd, kh, kw)));
 }
 
-ods_def<DepthwiseConvNHWCOp>:
+ods_def<DepthwiseConvInputNHWCFilterHWCOp>:
 def depthwise_conv_2d_input_nhwc_filter_hwc
       (I: f32(N, IH, IW, C), K: f32(KH, KW, C))
    -> (O: f32(N, OH, OW, C))