Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.cc58 std::vector<std::string> local_ids; member
87 params.local_sizes[0] + " + " + params.local_ids[0] + ";\n"; in GlobalIdsGen()
98 params.local_sizes[1] + " + " + params.local_ids[1] + ") * " + in GlobalIdsGen()
107 params.local_sizes[0] + " + " + params.local_ids[0] + ") * " + in GlobalIdsGen()
115 params.local_sizes[1] + " + " + params.local_ids[1] + ") * " + in GlobalIdsGen()
123 params.local_sizes[2] + " + " + params.local_ids[2] + ") * " + in GlobalIdsGen()
164 ids_params.local_ids = {"tid3d.x", "tid3d.y", "tid3d.z"}; in GenerateConvolution()
Dreduce.cc303 const std::vector<std::string> local_ids = {"local_x", "local_y", "local_z"}; in GetReduceKernelCode() local
308 const std::string first = index < wg_dims ? local_ids[index] : "0"; in GetReduceKernelCode()