Searched defs:all_gather (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 130 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/python/distribute/ |
| D | cross_device_utils.py | 437 def all_gather(self, member in CollectiveReplicaLauncher
|
| D | tpu_strategy.py | 1566 def all_gather(self, value, axis, experimental_hints=None): member in _TPUReplicaContext
|
| D | distribute_lib.py | 3295 def all_gather(self, value, axis, options=None): member in ReplicaContext
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 567 auto all_gather = Cast<HloAllGatherInstruction>(instruction); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1105 auto* all_gather = xla::Cast<xla::HloAllGatherInstruction>(instr); in EmitAllGatherOp() local
|