Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.cc134 int total_work_items, in GenerateUploadByThreads() argument
139 const int groups = elements_to_upload / total_work_items; in GenerateUploadByThreads()
140 const int reminder = elements_to_upload % total_work_items; in GenerateUploadByThreads()
143 std::to_string(total_work_items * i) + "] = " + global_ptr_name + "[" + in GenerateUploadByThreads()
144 offset + lid_name + " + " + std::to_string(total_work_items * i) + in GenerateUploadByThreads()
150 std::to_string(total_work_items * groups) + "] = " + global_ptr_name + in GenerateUploadByThreads()
152 std::to_string(total_work_items * groups) + "];\n"; in GenerateUploadByThreads()
388 const int total_work_items = params.work_group_size.x * in GenerateConvolution() local
394 total_work_items, local_mem_size); in GenerateConvolution()
Dconv_powervr.cc37 int total_work_items, in GenerateUploadByThreads() argument
42 const int groups = elements_to_upload / total_work_items; in GenerateUploadByThreads()
43 const int reminder = elements_to_upload % total_work_items; in GenerateUploadByThreads()
46 std::to_string(total_work_items * i) + "] = " + global_ptr_name + "[" + in GenerateUploadByThreads()
47 offset + lid_name + " + " + std::to_string(total_work_items * i) + in GenerateUploadByThreads()
53 std::to_string(total_work_items * groups) + "] = " + global_ptr_name + in GenerateUploadByThreads()
55 std::to_string(total_work_items * groups) + "];\n"; in GenerateUploadByThreads()
903 const int total_work_items = in GenerateConv() local
914 total_work_items, local_mem_size); in GenerateConv()
985 "DST_S", "lid", total_work_items, in GenerateConv()