Browse Source

Add transfer HWCN to Fractal_Zn_Lstm

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

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

@@ -111,11 +111,11 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
int index = 0;
for (int64_t y_src = y_src_head; y_src < y_src_head + K; y_src++) {
int64_t x_dst = x_src_head + index;
int64_t y_dst = x_src + y_src - x_dst;
++index;
int64_t src_idx = x_src * n + y_src;
auto src_offset = src_idx * size;

int64_t y_dst = x_src + y_src - x_dst;
auto src_idx = x_src * n + y_src;
auto src_offset = src_idx * size;
auto dst_idx = x_dst * n + y_dst;
auto dst_offset = dst_idx * size;

@@ -142,6 +142,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in

result.data = dst;
result.length = static_cast<size_t>(total_size);
return SUCCESS;
}
} // namespace



Loading…
Cancel
Save