Browse Source

add fractal zn lstm

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

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

@@ -101,7 +101,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
return OUT_OF_MEMORY; return OUT_OF_MEMORY;
} }


auto h = args.src_shape.at(kHwcnH);
/*auto h = args.src_shape.at(kHwcnH);
auto w = args.src_shape.at(kHwcnW); auto w = args.src_shape.at(kHwcnW);
auto c = args.src_shape.at(kHwcnC); auto c = args.src_shape.at(kHwcnC);
auto n = args.src_shape.at(kHwcnN); auto n = args.src_shape.at(kHwcnN);
@@ -114,7 +114,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in
int64_t wncoc0 = w * ncoc0; int64_t wncoc0 = w * ncoc0;
int64_t hwncoc0 = h * wncoc0; int64_t hwncoc0 = h * wncoc0;
int64_t cn = c * n; int64_t cn = c * n;
int64_t wcn = w * cn;
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; int64_t c1_head_addr = c1_idx * hwncoc0;


Loading…
Cancel
Save