Searched refs:_ragged_split (Results 1 – 1 of 1) sorted by relevance
160 def _ragged_split(tensor, pieces): function611 _ragged_split(input_tensors[d], num_gather_devices))