Home
last modified time | relevance | path

Searched refs:GetWorkGroupsCount (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.h76 return queue->Dispatch(kernel_, operation_->GetWorkGroupsCount(), in AddToQueue()
87 operation_->GetWorkGroupsCount()[i] * operation_->work_group_size_[i]; in AddToCommanBuffer()
103 operation_->GetWorkGroupsCount(), in AddToQueue()
111 return queue->DispatchNTimes(kernel_, operation_->GetWorkGroupsCount(),
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc280 groupsCount.width = operation_->GetWorkGroupsCount().x; in EncodeToICB()
281 groupsCount.height = operation_->GetWorkGroupsCount().y; in EncodeToICB()
282 groupsCount.depth = operation_->GetWorkGroupsCount().z; in EncodeToICB()
319 groupsCount.width = operation_->GetWorkGroupsCount().x; in Encode()
320 groupsCount.height = operation_->GetWorkGroupsCount().y; in Encode()
321 groupsCount.depth = operation_->GetWorkGroupsCount().z; in Encode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dutil.h62 int3 GetWorkGroupsCount(const int3& grid_size, const int3& work_group_size);
Dgpu_operation.h147 const int3& GetWorkGroupsCount() const { return work_groups_count_; } in GetWorkGroupsCount() function
Dutil.cc210 int3 GetWorkGroupsCount(const int3& grid_size, const int3& work_group_size) { in GetWorkGroupsCount() function
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc64 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in DispatchKernel()
204 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in Convert()