Searched refs:IsWaveSizeEqualTo32 (Results 1 – 6 of 6) sorted by relevance
80 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GetFullyConnectedKernelCode()
111 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConvolutionTransposedCode()
125 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConvolutionTransposedCode()
435 wg_total_size == 32 && gpu_info.IsWaveSizeEqualTo32() in GenerateConv()
340 bool IsWaveSizeEqualTo32() const;
457 bool GpuInfo::IsWaveSizeEqualTo32() const { in IsWaveSizeEqualTo32() function in tflite::gpu::GpuInfo