From 2f97c651e2a3dff205e0bd5f1a239ff3ad368e6b Mon Sep 17 00:00:00 2001 From: zk <694972388@qq.com> Date: Fri, 19 Mar 2021 17:15:57 +0800 Subject: [PATCH] tttt --- .../formats/format_transfers/format_transfer_fractal_z.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ge/common/formats/format_transfers/format_transfer_fractal_z.cc b/ge/common/formats/format_transfers/format_transfer_fractal_z.cc index 69d16842..968dedf7 100644 --- a/ge/common/formats/format_transfers/format_transfer_fractal_z.cc +++ b/ge/common/formats/format_transfers/format_transfer_fractal_z.cc @@ -25,7 +25,7 @@ #include "framework/common/debug/ge_log.h" #include "framework/common/debug/log.h" #include "graph/utils/type_utils.h" - +#include "iostream" namespace ge { namespace formats { namespace { @@ -519,7 +519,8 @@ Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &r return TransFormatNhwcToFz(args, result); } if ((args.src_format == FORMAT_HWCN) && (GetPrimaryFormat(args.dst_format) == FORMAT_FRACTAL_Z)) { - if (GetSubFormat(args.dst_format) >= 1) { + if (GetSubFormat(args.dst_format) > 1) { + std::cout<<"come in groups process"<= 1) { + if (GetSubFormat(dst_format) > 1) { return TransShapeHwcnToFzWithGroups(src_shape, data_type, dst_shape, GetSubFormat(dst_format)); } return TransShapeHwcnToFz(src_shape, data_type, dst_shape);