Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc25 int3 work_groups_count; in GetWorkGroupsCount() local
27 work_groups_count.x = DivideRoundUp(grid_size.x, work_group_size.x); in GetWorkGroupsCount()
28 work_groups_count.y = 1; in GetWorkGroupsCount()
29 work_groups_count.z = 1; in GetWorkGroupsCount()
34 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
35 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
36 work_groups_count.z = 1; in GetWorkGroupsCount()
42 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
43 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
44 work_groups_count.z = wgs[work_group_launch_order[2]]; in GetWorkGroupsCount()
[all …]
Dcl_command_queue.h53 const int3& work_groups_count,
56 absl::Status Dispatch(const CLKernel& kernel, const int3& work_groups_count,
89 absl::Status Dispatch(const CLKernel& kernel, const int3& work_groups_count,
95 const std::vector<int3>& work_groups_count,
Dcl_command_queue.cc61 const int3& work_groups_count, in Dispatch() argument
68 global[i] = work_groups_count[i] * work_group_size[i]; in Dispatch()
86 const int3& work_groups_count, in Dispatch() argument
88 return Dispatch(kernel, work_groups_count, work_group_size, nullptr); in Dispatch()
198 const int3& work_groups_count, in Dispatch() argument
201 RETURN_IF_ERROR(CLCommandQueue::Dispatch(kernel, work_groups_count, in Dispatch()
221 const std::vector<int3>& work_groups_count, in GetBestWorkGroupIndex() argument
228 RETURN_IF_ERROR(CLCommandQueue::Dispatch(kernel, work_groups_count[i], in GetBestWorkGroupIndex()
Dserialization.cc835 op->work_groups_count_.x = fb_op->work_groups_count()->x(); in Decode()
836 op->work_groups_count_.y = fb_op->work_groups_count()->y(); in Decode()
837 op->work_groups_count_.z = fb_op->work_groups_count()->z(); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc40 int3 work_groups_count; in GetWorkGroupsCount() local
42 work_groups_count.x = DivideRoundUp(grid_size.x, work_group_size.x); in GetWorkGroupsCount()
43 work_groups_count.y = 1; in GetWorkGroupsCount()
44 work_groups_count.z = 1; in GetWorkGroupsCount()
49 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
50 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
51 work_groups_count.z = 1; in GetWorkGroupsCount()
57 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
58 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
59 work_groups_count.z = wgs[work_group_launch_order[2]]; in GetWorkGroupsCount()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dutil.cc157 int3 work_groups_count; in GetWorkGroupsCount() local
158 work_groups_count.x = DivideRoundUp(grid_size.x, work_group_size.x); in GetWorkGroupsCount()
159 work_groups_count.y = DivideRoundUp(grid_size.y, work_group_size.y); in GetWorkGroupsCount()
160 work_groups_count.z = DivideRoundUp(grid_size.z, work_group_size.z); in GetWorkGroupsCount()
161 return work_groups_count; in GetWorkGroupsCount()
Dserialization_base.fbs242 work_groups_count:Int3;
Dserialization_base_generated.h2178 const tflite::gpu::data::Int3 *work_groups_count() const { in work_groups_count() function
2214 verifier.VerifyTable(work_groups_count()) && in Verify()
2287 flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count) { in add_work_groups_count()
2288 fbb_.AddOffset(GPUOperation::VT_WORK_GROUPS_COUNT, work_groups_count); in add_work_groups_count()
2331 flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count = 0,
2337 builder_.add_work_groups_count(work_groups_count);
2374 flatbuffers::Offset<tflite::gpu::data::Int3> work_groups_count = 0,
2399 src_tensors_names__, dst_tensors_names__, work_groups_count,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc60 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in DispatchKernel() local
61 return queue_->Dispatch(kernel_, work_groups_count, work_group_size); in DispatchKernel()
191 const int3 work_groups_count = GetWorkGroupsCount(grid, work_group_size); in Convert() local
192 return queue_->Dispatch(kernel_, work_groups_count, work_group_size); in Convert()