all_reduce_v2 (reference) in projects: external

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

/external/tensorflow/tensorflow/python/ops/
D	collective_ops.py	`74 def all_reduce_v2(t, function`
/external/tensorflow/tensorflow/python/distribute/
D	cross_device_utils.py	`357 return collective_ops.all_reduce_v2(`
/external/tensorflow/tensorflow/python/kernel_tests/
D	collective_ops_test.py	`60 return _collective_ops.all_reduce_v2(t, group_size, group_key, instance_key,`