Browse Source

add fractal zn lstm

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

+ 7
- 8
ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc View File

@@ -87,12 +87,6 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) {
}

Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const int size, const int64_t total_size) {

auto ret = memcpy_s(dst.get(), static_cast<size_t>(total_size), args.data, static_cast<size_t>(total_size));
result.data = dst;
result.length = static_cast<size_t>(total_size);
return SUCCESS;

std::shared_ptr<uint8_t> dst(new (std::nothrow) uint8_t[total_size], std::default_delete<uint8_t[]>());
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",
@@ -101,6 +95,11 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
return OUT_OF_MEMORY;
}

auto ret = memcpy_s(dst.get(), static_cast<size_t>(total_size), args.data, static_cast<size_t>(total_size));
result.data = dst;
result.length = static_cast<size_t>(total_size);
return SUCCESS;

/*auto h = args.src_shape.at(kHwcnH);
auto w = args.src_shape.at(kHwcnW);
auto c = args.src_shape.at(kHwcnC);
@@ -163,10 +162,10 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
}
}
}
}*/
}
result.data = dst;
result.length = static_cast<size_t>(total_size);
return SUCCESS;
return SUCCESS;*/
}
} // namespace



Loading…
Cancel
Save