Searched refs:divide_by_n_tensors_or_indexed_slices (Results 1 – 3 of 3) sorted by relevance
65 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n)74 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n)
653 def divide_by_n_tensors_or_indexed_slices(value, n): function
217 reduced = cross_device_utils.divide_by_n_tensors_or_indexed_slices(