Searched defs:AllReduceWrapper (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/distribution/ | ||
| D | distribution_collective.cc | 33 int DistributionCollective::AllReduceWrapper(const void *input_addr, void *output_addr, size_t coun… in AllReduceWrapper() function in mindspore::lite::DistributionCollective |
| D | distribution_collective_impl.cc | 55 int DistributionCollective::AllReduceWrapper(const void *input_addr, void *output_addr, size_t coun… in AllReduceWrapper() function in mindspore::lite::DistributionCollective |