mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix w_shape to kernel_size

This commit is contained in:
hbhu_bin 2022-11-18 09:48:24 +08:00
parent 9e375c7b44
commit 131dbb19af
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/core/ops/conv3d_transpose.cc
View File

@ -265,8 +265,8 @@ class Conv3DTransposeInfer : public abstract::OpInferBase {
MS_EXCEPTION(ValueError) << "For '" << prim_name << "', the dimension 'DHW' of input 'weight' must be "
<< " equal to the shape of 'kernel size', but got 'DHW' of input 'weight': ("
<< w_shape[kAxis2] << ", " << w_shape[kAxis3] << ", " << w_shape[kAxis4]
<< "), and 'kernel size': (" << kernel_size[kAxis0] << ", " << w_shape[kAxis1] << ", "
<< w_shape[kAxis2] << ").";
<< "), and 'kernel size': (" << kernel_size[kAxis0] << ", " << kernel_size[kAxis1]
<< ", " << kernel_size[kAxis2] << ").";
}
int64_t d_out = abstract::Shape::kShapeDimAny;