|
|
|
@@ -65,7 +65,7 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) { |
|
|
|
GELOGE(PARAM_INVALID, "Failed to check src shape %s", ShapeToString(args.src_shape).c_str()); |
|
|
|
return PARAM_INVALID; |
|
|
|
} |
|
|
|
if (!CheckShapeValid(args.dst_shape, kFracZnLstmDimsNum)) { |
|
|
|
if (!CheckShapeValid(args.dst_shape, kFracZDimsNum)) { |
|
|
|
GELOGE(PARAM_INVALID, "Failed to check dst shape %s", ShapeToString(args.dst_shape).c_str()); |
|
|
|
return PARAM_INVALID; |
|
|
|
} |
|
|
|
@@ -96,6 +96,10 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in |
|
|
|
} |
|
|
|
|
|
|
|
auto ret = memcpy_s(dst.get(), static_cast<size_t>(total_size), args.data, static_cast<size_t>(total_size)); |
|
|
|
if (ret != EOK) { |
|
|
|
GELOGE(INTERNAL_ERROR, "Failed to copy data ==="); |
|
|
|
return INTERNAL_ERROR; |
|
|
|
} |
|
|
|
result.data = dst; |
|
|
|
result.length = static_cast<size_t>(total_size); |
|
|
|
return SUCCESS; |
|
|
|
|