Home
last modified time | relevance | path

Searched refs:AllocateAllReduceAddr (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.h33 void AllocateAllReduceAddr() override;
Dgpu_bucket.cc47 void GPUBucket::AllocateAllReduceAddr() { in AllocateAllReduceAddr() function in mindspore::device::gpu::GPUBucket
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.h33 void AllocateAllReduceAddr() override;
Dascend_bucket.cc32 void AscendBucket::AllocateAllReduceAddr() { in AllocateAllReduceAddr() function in mindspore::device::ascend::AscendBucket
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dbucket.h79 virtual void AllocateAllReduceAddr() = 0;
Dbucket.cc49 AllocateAllReduceAddr(); in Launch()