Browse Source

add fractal zn lstm

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

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

@@ -92,7 +92,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
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",
@@ -116,7 +116,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
int64_t cn = c * n;
int64_t wcn = w * cn;

for (int64_t c1_idx = 0; c1_idx < c1; c1_idx++) {
/*for (int64_t c1_idx = 0; c1_idx < c1; c1_idx++) {
int64_t c1_head_addr = c1_idx * hwncoc0;
for (int64_t h_idx = 0; h_idx < h; h_idx++) {
int64_t h_head_addr = c1_head_addr + h_idx * wncoc0;
@@ -163,7 +163,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
}
}
}
}
}*/
result.data = dst;
result.length = static_cast<size_t>(total_size);
return SUCCESS;


Loading…
Cancel
Save