all_gather_v2 (reference) in projects: external

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

/external/tensorflow/tensorflow/python/ops/
D	collective_ops.py	`162 def all_gather_v2(t, function`
/external/tensorflow/tensorflow/python/distribute/
D	cross_device_utils.py	`381 return collective_ops.all_gather_v2(`
/external/tensorflow/tensorflow/python/kernel_tests/
D	collective_ops_test.py	`68 return _collective_ops.all_gather_v2(t, group_size, group_key, instance_key,`