Home
last modified time | relevance | path

Searched defs:ReduceScatterWrapper (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/distribution/
Ddistribution_collective.cc27 int DistributionCollective::ReduceScatterWrapper(const void *input_addr, void *output_addr, size_t … in ReduceScatterWrapper() function in mindspore::lite::DistributionCollective
Ddistribution_collective_impl.cc36 int DistributionCollective::ReduceScatterWrapper(const void *input_addr, void *output_addr, size_t … in ReduceScatterWrapper() function in mindspore::lite::DistributionCollective