Browse Source

Transdata

tags/v1.3.0
zk 4 years ago
parent
commit
b3082e8faf
1 changed files with 7 additions and 10 deletions
  1. +7
    -10
      ge/common/formats/format_transfers/format_transfer_fractal_z.cc

+ 7
- 10
ge/common/formats/format_transfers/format_transfer_fractal_z.cc View File

@@ -319,16 +319,13 @@ Status TransFormatHwcnToFzWithGroups(const TransArgs &args, TransResult &result,
int64_t src_co = g * cout_ori + n; int64_t src_co = g * cout_ori + n;
int64_t tempory = dst_ci % cube_k; int64_t tempory = dst_ci % cube_k;
int64_t srx_inx = 0; int64_t srx_inx = 0;
int64_t dst_inx =
(g / e_mult) * kDim * c1_dim * h_dim * w_dim * cout_opt *
cube_k +
d * c1_dim * h_dim * w_dim * cout_opt * cube_k +
(dst_ci / cube_k) * h_dim * w_dim * cout_opt * cube_k +
h * w_dim * cout_opt * cube_k + w * cout_opt * cube_k +
dst_co * cube_k + tempory;
srx_inx = d * h_dim * w_dim * c_dim * n_dim +
h * w_dim * c_dim * n_dim + w * c_dim * n_dim +
c * n_dim + src_co;
int64_t dst_inx = (g / e_mult) * kDim * c1_dim * h_dim * w_dim * cout_opt * cube_k +
d * c1_dim * h_dim * w_dim * cout_opt * cube_k +
(dst_ci / cube_k) * h_dim * w_dim * cout_opt * cube_k +
h * w_dim * cout_opt * cube_k + w * cout_opt * cube_k +
dst_co * cube_k + tempory;
srx_inx = d * h_dim * w_dim * c_dim * n_dim + h * w_dim * c_dim * n_dim +
w * c_dim * n_dim + c * n_dim + src_co;
char *dst_data = reinterpret_cast<char *>(dst.get() + dst_inx * data_size); char *dst_data = reinterpret_cast<char *>(dst.get() + dst_inx * data_size);
const char *src_data = reinterpret_cast<const char *>(args.data + srx_inx * data_size); const char *src_data = reinterpret_cast<const char *>(args.data + srx_inx * data_size);
for (int64_t index = 0; index < data_size; index++) { for (int64_t index = 0; index < data_size; index++) {


Loading…
Cancel
Save