divide_by_n_tensors_or_indexed_slices (reference) in projects: aosp_15_r20

Searched refs:divide_by_n_tensors_or_indexed_slices (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	cross_device_utils_test.py	`63 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n) 72 result = cross_device_utils.divide_by_n_tensors_or_indexed_slices(t, n)`
H A D	cross_device_utils.py	`583 def divide_by_n_tensors_or_indexed_slices(value, n): function`
H A D	cross_device_ops.py	`229 reduced = cross_device_utils.divide_by_n_tensors_or_indexed_slices(`