Home
last modified time | relevance | path

Searched refs:bucket_map_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.h316 std::map<uint32_t, std::vector<std::shared_ptr<device::Bucket>>> bucket_map_; variable
Dsession_basic.cc2499 auto bucket_ret = bucket_map_.try_emplace(graph->graph_id(), bucket_list); in InitAllBucket()
2519 auto iter = bucket_map_.find(graph_id); in AddGradAddrToBucket()
2520 if (iter == bucket_map_.end()) { in AddGradAddrToBucket()
2547 auto iter = bucket_map_.find(graph_id); in ClearAllBucket()
2548 if (iter != bucket_map_.end()) { in ClearAllBucket()