Searched refs:divide_by_n_tensors_or_indexed_slices (Results 1 – 3 of 3) sorted by relevance
66 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n)75 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n)
582 def divide_by_n_tensors_or_indexed_slices(value, n): function
213 reduced = cross_device_utils.divide_by_n_tensors_or_indexed_slices(