Searched defs:CreateBucket (Results 1 – 5 of 5) sorted by relevance
118 …virtual std::shared_ptr<Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) const { ret… in CreateBucket() function
509 std::shared_ptr<Bucket> GPUDeviceContext::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) co… in CreateBucket() function in mindspore::device::gpu::GPUDeviceContext
303 …virtual std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { r… in CreateBucket() function
737 std::shared_ptr<device::Bucket> GPUSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { in CreateBucket() function in mindspore::session::gpu::GPUSession
1942 std::shared_ptr<device::Bucket> AscendSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_siz… in CreateBucket() function in mindspore::session::AscendSession