src_data, dest_data, is_forward);
}
} else {
- // We are at the last two dimensions, which are stored continuously in memory
- // With (N,C,H,W)
- // (0,1,2,3) cur_dim -> H
- // cur_dim+1 -> W
+ // We are at the last two dimensions, which are stored continuously in
+ // memory With (N,C,H,W)
+ // (0,1,2,3) cur_dim -> H
+ // cur_dim+1 -> W
const int lines = top[0]->shape(cur_dim);
const int height = top[0]->shape(cur_dim);
const int width = top[0]->shape(cur_dim+1);