| @@ -34,7 +34,7 @@ bool CheckDataTypeSupported(const DataType &data_type) { | |||||
| Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector<int64_t> &src_shape, | Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector<int64_t> &src_shape, | ||||
| std::vector<int64_t> &dst_shape) { | std::vector<int64_t> &dst_shape) { | ||||
| auto cube_size = GetCubeSizeByDataType(data_type); | |||||
| /*auto cube_size = GetCubeSizeByDataType(data_type); | |||||
| dst_shape.clear(); | dst_shape.clear(); | ||||
| dst_shape.push_back(Ceil(src_shape.at(kHwcnC), static_cast<int64_t>(cube_size))); | dst_shape.push_back(Ceil(src_shape.at(kHwcnC), static_cast<int64_t>(cube_size))); | ||||
| dst_shape.push_back(src_shape.at(kHwcnH)); | dst_shape.push_back(src_shape.at(kHwcnH)); | ||||
| @@ -46,7 +46,7 @@ Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector<int | |||||
| GELOGE(ACL_ERROR_GE_SHAPE_INVALID, "Failed to check dst shape %s", | GELOGE(ACL_ERROR_GE_SHAPE_INVALID, "Failed to check dst shape %s", | ||||
| ShapeToString(dst_shape).c_str()); | ShapeToString(dst_shape).c_str()); | ||||
| return ACL_ERROR_GE_SHAPE_INVALID; | return ACL_ERROR_GE_SHAPE_INVALID; | ||||
| } | |||||
| }*/ | |||||
| return SUCCESS; | return SUCCESS; | ||||
| } | } | ||||
| @@ -89,6 +89,7 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) { | |||||
| } | } | ||||
| Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const int size, const int64_t total_size) { | Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const int size, const int64_t total_size) { | ||||
| return SUCCESS; | |||||
| std::shared_ptr<uint8_t> dst(new (std::nothrow) uint8_t[total_size], std::default_delete<uint8_t[]>()); | std::shared_ptr<uint8_t> dst(new (std::nothrow) uint8_t[total_size], std::default_delete<uint8_t[]>()); | ||||
| if (dst == nullptr) { | 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", | GELOGE(OUT_OF_MEMORY, "Failed to trans format from %s to %s, can not alloc the memory for dst buf %ld, shape %s", | ||||