Searched defs:all_gather (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 437 def all_gather(self, member in CollectiveReplicaLauncher
537 def all_gather(): function
|
| D | tpu_strategy.py | 1553 def all_gather(self, value, axis, experimental_hints=None): member in _TPUReplicaContext
|
| D | distribute_lib.py | 3156 def all_gather(self, value, axis, options=None): member in ReplicaContext
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 123 def all_gather(t, function
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 48 all_gather = _collective_ops.all_gather variable in CollectiveOpsV1
64 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): member in CollectiveOpsV2
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1065 auto* all_gather = xla::Cast<xla::HloAllGatherInstruction>(instr); in EmitAllGatherOp() local
|