Home
last modified time | relevance | path

Searched refs:possible_work_groups (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc210 std::vector<int3> possible_work_groups; in Tune() local
212 &possible_work_groups); in Tune()
213 if (possible_work_groups.empty()) { in Tune()
217 if (possible_work_groups.size() == 1) { in Tune()
218 operation_->work_group_size_ = possible_work_groups[0]; in Tune()
224 std::vector<int3> work_groups_count(possible_work_groups.size()); in Tune()
228 possible_work_groups[i], operation_->work_group_launch_order_); in Tune()
233 kernel_, gpu_info, work_groups_count, possible_work_groups, in Tune()
235 operation_->work_group_size_ = possible_work_groups[best_work_group_index]; in Tune()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc243 std::vector<int3> possible_work_groups; in Tune() local
248 kernel_info, &possible_work_groups); in Tune()
249 if (possible_work_groups.empty()) { in Tune()
253 operation_->work_group_size_ = possible_work_groups[0]; in Tune()