|
@@ -293,7 +293,7 @@ Status TransFormatHwcnToFzWithGroups(const TransArgs &args, TransResult &result, |
|
|
std::cout<<"cout_opt:"<<cout_opt<<std::endl; |
|
|
std::cout<<"cout_opt:"<<cout_opt<<std::endl; |
|
|
std::cout<<"c1_dim:"<<c1_dim<<std::endl; |
|
|
std::cout<<"c1_dim:"<<c1_dim<<std::endl; |
|
|
std::cout<<"g_dim:"<<g_dim<<std::endl; |
|
|
std::cout<<"g_dim:"<<g_dim<<std::endl; |
|
|
int64_t data_size = GetCubeSizeByDataType(args.src_data_type); |
|
|
|
|
|
|
|
|
int64_t data_size = GetSizeByDataType(args.src_data_type); |
|
|
std::cout<<"data_size:"<<data_size<<std::endl; |
|
|
std::cout<<"data_size:"<<data_size<<std::endl; |
|
|
int64_t size_output_data = |
|
|
int64_t size_output_data = |
|
|
g_dim * kDim * dim_cin * h_dim * w_dim * cout_opt * cube_k * data_size; |
|
|
g_dim * kDim * dim_cin * h_dim * w_dim * cout_opt * cube_k * data_size; |
|
|