提交 b2e2a83e 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Fix wrong error message in `legalize_hlo.cc`

PiperOrigin-RevId: 561126894
上级 50dd11b0
......@@ -937,7 +937,7 @@ class ConvertNonTrivialConvOp
if (dnums.getKernelInputFeatureDimension() != num_spatial_dims + 1 ||
dnums.getKernelOutputFeatureDimension() != num_spatial_dims)
return rewriter.notifyMatchFailure(conv_op,
"requires kernel format [b, 0, 1, f]");
"requires kernel format [0, 1, o, i]");
auto kernel_spatial_dimensions = dnums.getKernelSpatialDimensions();
for (auto p : llvm::enumerate(kernel_spatial_dimensions)) {
if (p.value() != p.index())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册