error message add format

4 years ago · 0d484181af
parent 01934a69f2
commit 0d484181af
2 changed files with 2 additions and 2 deletions
--- a/ge/common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc
+++ b/ge/common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc
@ -148,7 +148,7 @@ Status FormatTransferDhwcnFractalZ3D::TransFormat(const TransArgs &args, TransRe
  if (ret != SUCCESS) {
    return ret;
  }
-  if (!IsTransShapeCorrect(args, expect_shape)) {
+  if (!IsTransShapeDstCorrect(args, expect_shape)) {
    return PARAM_INVALID;
  }

--- a/ge/common/formats/format_transfers/format_transfer_nchw_fz_c04.cc
+++ b/ge/common/formats/format_transfers/format_transfer_nchw_fz_c04.cc
@ -280,7 +280,7 @@ Status FormatTransferNchwToFZC04::TransFormat(const TransArgs &args, TransResult
    return ret;
  }

-  if (!IsTransShapeCorrect(args_tmp, expect_shape)) {
+  if (!IsTransShapeDstCorrect(args_tmp, expect_shape)) {
    return PARAM_INVALID;
  }