Home
last modified time | relevance | path

Searched refs:work_groups_count (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_command_queue.cc62 const int3& work_groups_count, in Dispatch() argument
69 global[i] = work_groups_count[i] * work_group_size[i]; in Dispatch()
87 const int3& work_groups_count, in Dispatch() argument
89 return Dispatch(kernel, work_groups_count, work_group_size, nullptr); in Dispatch()
208 const int3& work_groups_count, in Dispatch() argument
212 RETURN_IF_ERROR(CLCommandQueue::Dispatch(kernel, work_groups_count, in Dispatch()
220 const CLKernel& kernel, const int3& work_groups_count, in DispatchNTimes() argument
225 RETURN_IF_ERROR(CLCommandQueue::Dispatch(kernel, work_groups_count, in DispatchNTimes()
232 RETURN_IF_ERROR(CLCommandQueue::Dispatch(kernel, work_groups_count, in DispatchNTimes()
237 CLCommandQueue::Dispatch(kernel, work_groups_count, work_group_size)); in DispatchNTimes()
[all …]
Dcl_command_queue.h53 const int3& work_groups_count,
56 absl::Status Dispatch(const CLKernel& kernel, const int3& work_groups_count,
91 absl::Status Dispatch(const CLKernel& kernel, const int3& work_groups_count,
96 const int3& work_groups_count,
103 const std::vector<int3>& work_groups_count,
Dcl_operation.cc208 work_groups_counts[i] = possible_dispatches[i].work_groups_count; in Tune()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dutil.cc211 int3 work_groups_count; in GetWorkGroupsCount() local
212 work_groups_count.x = DivideRoundUp(grid_size.x, work_group_size.x); in GetWorkGroupsCount()
213 work_groups_count.y = DivideRoundUp(grid_size.y, work_group_size.y); in GetWorkGroupsCount()
214 work_groups_count.z = DivideRoundUp(grid_size.z, work_group_size.z); in GetWorkGroupsCount()
215 return work_groups_count; in GetWorkGroupsCount()
Dgpu_operation.cc36 int3 work_groups_count; in GetWorkGroupsCountInternal() local
38 work_groups_count.x = DivideRoundUp(grid_size.x, work_group_size.x); in GetWorkGroupsCountInternal()
39 work_groups_count.y = 1; in GetWorkGroupsCountInternal()
40 work_groups_count.z = 1; in GetWorkGroupsCountInternal()
45 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCountInternal()
46 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCountInternal()
47 work_groups_count.z = 1; in GetWorkGroupsCountInternal()
53 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCountInternal()
54 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCountInternal()
55 work_groups_count.z = wgs[work_group_launch_order[2]]; in GetWorkGroupsCountInternal()
[all …]
Dserialization_base.fbs208 work_groups_count:Int3;
Dgpu_operation.h133 int3 work_groups_count; member
Dserialization_base_generated.h1722 const tflite::gpu::data::Int3 *work_groups_count() const { in work_groups_count() function
1752 verifier.VerifyTable(work_groups_count()) && in Verify()
1797 void add_work_groups_count(flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count) { in add_work_groups_count()
1798 fbb_.AddOffset(GPUOperation::VT_WORK_GROUPS_COUNT, work_groups_count); in add_work_groups_count()
1832 flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count = 0) {
1835 builder_.add_work_groups_count(work_groups_count);
1868 flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count = 0) {
1877 dst_tensors_names__, work_groups_count);
Dserialization_base.cc672 op->work_groups_count_.x = fb_op->work_groups_count()->x(); in Decode()
673 op->work_groups_count_.y = fb_op->work_groups_count()->y(); in Decode()
674 op->work_groups_count_.z = fb_op->work_groups_count()->z(); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc64 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in DispatchKernel() local
65 return queue_->Dispatch(kernel_, work_groups_count, work_group_size); in DispatchKernel()
204 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in Convert() local
205 return queue_->Dispatch(kernel_, work_groups_count, work_group_size); in Convert()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_generic.cc1210 int work_groups_count; member
1224 wg.work_groups_count = in CreateWorkGroupSizeOption()
1228 wg.work_groups_count = in CreateWorkGroupSizeOption()
1231 wg.work_groups_count = in CreateWorkGroupSizeOption()
1305 float optimum = options[0].work_groups_count * options[0].penalty * in GetConvParamsForA7A8()
1310 float local_optimum = options[i].work_groups_count * options[i].penalty * in GetConvParamsForA7A8()