| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.h | 99 void HandlePushReq(const DataPtr &data, size_t size, const VectorPtr &res);
100 void HandlePullReq(const DataPtr &data, size_t size, const VectorPtr &res);
101 void HandleInitWeights(const DataPtr &data, size_t size, const VectorPtr &res);
102 void HandleInitWeightToOptimId(const DataPtr &data, size_t size, const VectorPtr &res);
103 void HandleInitInputsShape(const DataPtr &data, size_t size, const VectorPtr &res);
104 void HandleInitEmbeddings(const DataPtr &data, size_t size, const VectorPtr &res);
105 void HandleCheckReadyForPush(const DataPtr &data, size_t size, const VectorPtr &res);
106 void HandleCheckReadyForPull(const DataPtr &data, size_t size, const VectorPtr &res);
107 void HandleEmbeddingLookup(const DataPtr &data, size_t size, const VectorPtr &res);
108 void HandleUpdateEmbeddings(const DataPtr &data, size_t size, const VectorPtr &res);
[all …]
|
| D | parameter_server.cc | 627 …meterServer::ServerHandler::HandlePushReq(const DataPtr &data, size_t size, const VectorPtr &res) { in HandlePushReq()
638 …meterServer::ServerHandler::HandlePullReq(const DataPtr &data, size_t size, const VectorPtr &res) { in HandlePullReq()
656 …rServer::ServerHandler::HandleInitWeights(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleInitWeights()
681 …:ServerHandler::HandleInitWeightToOptimId(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleInitWeightToOptimId()
699 …ver::ServerHandler::HandleInitInputsShape(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleInitInputsShape()
716 …rServer::ServerHandler::HandleInitEmbeddings(const DataPtr &data, size_t size, const VectorPtr &) { in HandleInitEmbeddings() argument
753 …r::ServerHandler::HandleCheckReadyForPush(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleCheckReadyForPush()
772 …r::ServerHandler::HandleCheckReadyForPull(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleCheckReadyForPull()
790 …ver::ServerHandler::HandleEmbeddingLookup(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleEmbeddingLookup()
811 …er::ServerHandler::HandleUpdateEmbeddings(const DataPtr &data, size_t size, const VectorPtr &res) { in HandleUpdateEmbeddings()
[all …]
|
| D | constants.h | 196 using VectorPtr = std::shared_ptr<std::vector<unsigned char>>; variable
|
| D | worker.cc | 309 std::vector<VectorPtr> resp; in DoPSEmbeddingLookup()
989 std::vector<VectorPtr> resp; in SendForPull()
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | node.cc | 151 VectorPtr received_data = std::make_shared<std::vector<unsigned char>>(size, 0); in ProcessSendDataResp()
162 std::unordered_map<uint32_t, VectorPtr> res; in ProcessSendDataResp()
168 VectorPtr received_data = std::make_shared<std::vector<unsigned char>>(size, 0); in ProcessSendDataResp()
179 std::unordered_map<uint32_t, VectorPtr> res; in ProcessSendDataResp()
|
| D | abstract_node.h | 65 using VectorPtr = std::shared_ptr<std::vector<unsigned char>>; variable
97 VectorPtr *output, const uint32_t &timeout = kTimeoutInSeconds);
99 const std::vector<size_t> &data_lens, int command, std::vector<VectorPtr> *output,
104 VectorPtr *output);
|
| D | node.h | 135 std::unordered_map<uint64_t, std::unordered_map<uint32_t, VectorPtr>> receive_messages_;
137 std::unordered_map<uint64_t, std::unordered_map<uint32_t, VectorPtr>> workder_receive_messages_;
|
| D | scheduler_node.cc | 766 std::unordered_map<uint32_t, VectorPtr> outputs; in ProcessNewInstance()
820 std::unordered_map<uint32_t, VectorPtr> outputs; in ProcessQueryInstance()
874 std::unordered_map<uint32_t, VectorPtr> outputs; in ProcessEnableFLS()
930 std::unordered_map<uint32_t, VectorPtr> outputs; in ProcessDisableFLS()
|
| D | abstract_node.cc | 241 int command, VectorPtr *output, const uint32_t &timeout) { in Send()
282 std::vector<VectorPtr> *output, const uint32_t &timeout) { in Send()
354 VectorPtr *output) { in CollectiveReceiveAsync()
|
| /third_party/skia/third_party/externals/swiftshader/third_party/llvm-10.0/llvm/include/llvm/CodeGen/PBQP/ |
| D | Graph.h | 55 using VectorPtr = typename CostAllocator::VectorPtr; variable
68 NodeEntry(VectorPtr Costs) : Costs(std::move(Costs)) {} in NodeEntry()
94 VectorPtr Costs;
377 VectorPtr AllocatedCosts = CostAlloc.getVector(std::move(Costs)); in addNode()
467 VectorPtr AllocatedCosts = CostAlloc.getVector(std::move(Costs)); in setNodeCosts()
481 const VectorPtr& getNodeCostsPtr(NodeId NId) const { in getNodeCostsPtr()
|
| D | CostAllocator.h | 115 using VectorPtr = typename VectorCostPool::PoolRef; variable
118 template <typename VectorKeyT> VectorPtr getVector(VectorKeyT v) { in getVector()
|