Searched refs:local_sizes (Results 1 – 3 of 3) sorted by relevance
57 std::vector<std::string> local_sizes; member87 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()165 ids_params.local_sizes = {"lsize.x", "lsize.y", "lsize.z"}; in GenerateConvolution()
304 const std::vector<std::string> local_sizes = {wg_x, wg_y, wg_z}; in GetReduceKernelCode() local309 const std::string step = index < wg_dims ? local_sizes[index] : "1"; in GetReduceKernelCode()
509 isl_union_set *local_sizes; in extract_sizes() local522 local_sizes = isl_union_set_apply(isl_union_set_from_set(dom), in extract_sizes()524 isl_union_set_foreach_set(local_sizes, &extract_size_of_type, &data); in extract_sizes()525 isl_union_set_free(local_sizes); in extract_sizes()