Searched refs:_iter_slices (Results 1 – 1 of 1) sorted by relevance
737 for i, (var_offset, var_shape) in enumerate(_iter_slices(2450 def _iter_slices(full_shape, num_slices, slice_dim): function