build_shuffle_all_reduce (reference) in projects: external

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

/external/tensorflow/tensorflow/python/distribute/
D	all_reduce.py	`552 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): function 792 return build_shuffle_all_reduce(x, gather_devices, shuffle_red_op, un_op) 855 return build_shuffle_all_reduce(tensors, second_gather_devices,`
D	all_reduce_test.py	`194 return lambda x, un_op: ar.build_shuffle_all_reduce(`
D	cross_device_utils.py	`409 summed_grads = all_reduce.build_shuffle_all_reduce(`