Home
last modified time | relevance | path

Searched defs:CreateBucket (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h118 …virtual std::shared_ptr<Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) const { ret… in CreateBucket() function
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc509 std::shared_ptr<Bucket> GPUDeviceContext::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) co… in CreateBucket() function in mindspore::device::gpu::GPUDeviceContext
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.h303 …virtual std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { r… in CreateBucket() function
Dgpu_session.cc737 std::shared_ptr<device::Bucket> GPUSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { in CreateBucket() function in mindspore::session::gpu::GPUSession
Dascend_session.cc1942 std::shared_ptr<device::Bucket> AscendSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_siz… in CreateBucket() function in mindspore::session::AscendSession