|
|
|
@@ -101,7 +101,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in |
|
|
|
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 c = args.src_shape.at(kHwcnC); |
|
|
|
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 hwncoc0 = h * wncoc0; |
|
|
|
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++) { |
|
|
|
int64_t c1_head_addr = c1_idx * hwncoc0; |
|
|
|
|