Searched defs:tensor_dims_map (Results 1 – 1 of 1) sorted by relevance
136 def grid_partitioner(full_grid, grid_blocks, tensor_dims_map): argument161 def launch_kernel(kernel, tensor_dims_map, full_grid, grid_blocks=None): argument