Browse Source

add fractal zn lstm

pull/1502/head
dongduo5@huawei.com 4 years ago
parent
commit
ab463188e6
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc

+ 5
- 1
ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc View File

@@ -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;


Loading…
Cancel
Save