Searched refs:trans_matrix_data_size (Results 1 – 3 of 3) sorted by relevance
111 auto trans_matrix_data_size = in Init() local113 set_workspace_size(trans_matrix_data_size); in Init()204 auto trans_matrix_data_size = in MallocWeightBiasData() local208 packed_weight_ = malloc(trans_matrix_data_size); in MallocWeightBiasData()214 memset(packed_weight_, 0, trans_matrix_data_size); in MallocWeightBiasData()
43 …auto trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * col_tile_ * … in MallocWeightBiasData() local46 packed_weight_ = malloc(trans_matrix_data_size); in MallocWeightBiasData()52 memset(packed_weight_, 0, trans_matrix_data_size); in MallocWeightBiasData()156 …auto trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * col_tile_ * … in Init() local157 set_workspace_size(trans_matrix_data_size); in Init()
133 int trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * oc_block; in InitWeightBias() local135 …allocator_->Malloc(kNumberTypeFloat32, trans_matrix_data_size * sizeof(float), kOfflinePackWeight)… in InitWeightBias()137 …int ret = memset_s(trans_weight_, trans_matrix_data_size * sizeof(float), 0, trans_matrix_data_siz… in InitWeightBias()