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

!45701 fix conv3dtranspose error log 

Merge pull request !45701 from 胡彬/conv3d-transpose-fix
This commit is contained in:
i-robot 2022-11-18 08:13:07 +00:00 committed by Gitee
parent 26aef12ff9 131dbb19af
commit deb8beb524
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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;