Searched refs:batch_reduce_implementation (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_ops.py | 443 return self.batch_reduce_implementation(reduce_op, value_destination_pairs,
498 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in CrossDeviceOps
637 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in ReductionToOneDevice
868 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in AllReduceCrossDeviceOps
1254 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in CollectiveAllReduce
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.distribute.-cross-device-ops.pbtxt | 14 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 15 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 16 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 16 name: "batch_reduce_implementation"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.distribute.-cross-device-ops.pbtxt | 14 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 15 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 16 name: "batch_reduce_implementation"
|
| H A D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 16 name: "batch_reduce_implementation"
|