Searched refs:GetPossibleWorkGroupsConv (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | work_group_picking.h | 54 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info,
|
| H A D | work_group_picking.cc | 293 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroupsConv() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | convolution_transposed_3x3.cc | 354 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| H A D | convolution_transposed.cc | 604 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| H A D | conv_generic.cc | 372 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | converter.cc | 61 GetPossibleWorkGroupsConv(TuningType::kFast, gpu_info_, kernel_.info_, grid, in DispatchKernel()
|