Searched refs:wg_total_size (Results 1 – 5 of 5) sorted by relevance
78 const int wg_total_size = work_group_size_.x * work_group_size_.y; in GetFullyConnectedKernelCode() local80 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GetFullyConnectedKernelCode()
108 const int wg_total_size = in GenerateConvolutionTransposedCode() local111 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConvolutionTransposedCode()
122 const int wg_total_size = in GenerateConvolutionTransposedCode() local125 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConvolutionTransposedCode()
207 const int wg_total_size = in GetReduceKernelCode() local211 c += " __local float4 accum[" + std::to_string(wg_total_size) + "];\n"; in GetReduceKernelCode()
432 const int wg_total_size = in GenerateConv() local435 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConv()