Home
last modified time | relevance | path

Searched defs:work_groups_count (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dcl_command_queue.cc62 const int3& work_groups_count, in Dispatch()
87 const int3& work_groups_count, in Dispatch()
208 const int3& work_groups_count, in Dispatch()
220 const CLKernel& kernel, const int3& work_groups_count, in DispatchNTimes()
275 const std::vector<int3>& work_groups_count, in GetBestWorkGroupIndex()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dconverter.cc64 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in DispatchKernel() local
204 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in Convert() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dutil.cc211 int3 work_groups_count; in GetWorkGroupsCount() local
H A Dgpu_operation.h133 int3 work_groups_count; member
H A Dgpu_operation.cc36 int3 work_groups_count; in GetWorkGroupsCountInternal() local
H A Dserialization_base_generated.h1722 const tflite::gpu::data::Int3 *work_groups_count() const { in work_groups_count() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_generic.cc1210 int work_groups_count; member