Searched refs:GetMaxWorkGroupSizeForX (Results 1 – 6 of 6) sorted by relevance
54 if (x <= gpu_info.GetMaxWorkGroupSizeForX() && in GenerateWorkGroupSizesXYMultipleOf()80 if (x <= gpu_info.GetMaxWorkGroupSizeForX() && in GenerateWorkGroupSizesXMultipleOf()96 max_wg_size.x = gpu_info.GetMaxWorkGroupSizeForX(); in GetWorkGroupsAlignedToGrid()
32 while (wg->x > gpu_info.GetMaxWorkGroupSizeForX()) { in MaybeShrinkWorkgroup()
100 std::min(tensor_slices, gpu_info.GetMaxWorkGroupSizeForX()); in MeanStdDevNormalization()
99 const uint32_t max_wg_size_x = ctx.gpu_info->GetMaxWorkGroupSizeForX(); in GenerateSubgroupBasedMean()
350 int GetMaxWorkGroupSizeForX() const;
524 int GpuInfo::GetMaxWorkGroupSizeForX() const { in GetMaxWorkGroupSizeForX() function in tflite::gpu::GpuInfo