Searched refs:_iter_slices (Results 1 – 1 of 1) sorted by relevance
758 _iter_slices(shape.as_list(), num_slices, slice_dim)):2572 def _iter_slices(full_shape, num_slices, slice_dim): function