Searched defs:_all_reduce (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_ops.py | 552 def _all_reduce(self, reduce_op, value, replica_id, options): member in CrossDeviceOps
1113 def _all_reduce(self, reduce_op, value, replica_id, options): member in CollectiveAllReduce
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nccl_ops.py | 215 def _all_reduce(): function
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | _functional_collectives_impl.py | 23 def _all_reduce(input, reduce_op, tag, ranks, group_size): function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 6309 def _all_reduce(inp, reduce_op, group_name): function
|