From baa288f6ef1059917deccf5d0d1ae459ec9ae191 Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Mon, 12 Apr 2021 19:50:41 +0800 Subject: [PATCH] add fractal zn index --- .../format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc index 279c840b..b046df1f 100644 --- a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc +++ b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc @@ -34,7 +34,7 @@ bool CheckDataTypeSupported(const DataType &data_type) { Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector &src_shape, std::vector &dst_shape) { - auto cube_size = GetCubeSizeByDataType(data_type); + /*auto cube_size = GetCubeSizeByDataType(data_type); dst_shape.clear(); dst_shape.push_back(Ceil(src_shape.at(kHwcnC), static_cast(cube_size))); dst_shape.push_back(src_shape.at(kHwcnH)); @@ -46,7 +46,7 @@ Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector dst(new (std::nothrow) uint8_t[total_size], std::default_delete()); if (dst == nullptr) { GELOGE(OUT_OF_MEMORY, "Failed to trans format from %s to %s, can not alloc the memory for dst buf %ld, shape %s",