Searched refs:slices_per_thread (Results 1 – 2 of 2) sorted by relevance
123 if (conv_params.slices_per_thread != 1) { in GenerateConvolution()124 c += " DST_S *= " + std::to_string(conv_params.slices_per_thread) + ";\n"; in GenerateConvolution()131 const int dst_x4_slices = conv_params.slices_per_thread; in GenerateConvolution()446 DivideRoundUp(dst_[0]->Slices(), params_.slices_per_thread); in GetGridSize()472 desc.params_.slices_per_thread = 4; in CreateConvolutionMetalSimd()498 attr.weights, weights_type, desc.params_.slices_per_thread / 2); in CreateConvolutionMetalSimd()
58 int slices_per_thread; member