Searched refs:_padded_split (Results 1 – 1 of 1) sorted by relevance
78 def _padded_split(tensor, pieces): function332 splits, split_pad_len = _padded_split(input_tensors[d], num_chunks)