Searched defs:OperatorInfoPtr (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.h | 32 using OperatorInfoPtr = std::shared_ptr<mindspore::parallel::OperatorInfo>; variable
|
| D | node_info.cc | 465 std::vector<std::pair<OperatorInfoPtr, int64_t>> *next_ops_index, in FindReshapeNextNodeStraCosts()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 37 using OperatorInfoPtr = std::shared_ptr<mindspore::parallel::OperatorInfo>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.h | 45 using OperatorInfoPtr = std::shared_ptr<OperatorInfo>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | graph_costmodel.cc | 94 void CostGraph::StrategyPropagate(const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &ops_… in StrategyPropagate()
150 … const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &configured_ops) { in CheckConfiguredSuccEdgeConsistency()
175 … const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &configured_ops) { in CheckConfiguredPrevEdgeConsistency()
197 const std::shared_ptr<Edge> &edge, std::map<OperatorInfoPtr, bool> *visited, int64_t curr_depth, in BFSPreNode()
198 …const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &configured_ops, const OperatorInfoPtr… in BFSPreNode()
199 …std::queue<std::pair<std::pair<OperatorInfoPtr, std::pair<StrategyPtr, int64_t>>, int64_t>> *next_… in BFSPreNode()
249 const std::shared_ptr<Edge> &edge, std::map<OperatorInfoPtr, bool> *visited, int64_t curr_depth, in BFSNextNode()
250 …const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &configured_ops, const OperatorInfoPtr… in BFSNextNode()
251 …std::queue<std::pair<std::pair<OperatorInfoPtr, std::pair<StrategyPtr, int64_t>>, int64_t>> *next_… in BFSNextNode()
300 const std::map<OperatorInfoPtr, StrategyPtr, OpsPtrCompare> &configured_ops, in BFS()
[all …]
|
| D | edge_costmodel.cc | 211 void Edge::EdgeEliminationSetNewCost(OperatorInfoPtr, const std::vector<EdgePtr> &edges, OperatorIn… in EdgeEliminationSetNewCost() argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 138 TensorInfo PipelineTransformer::GetTensorInfo(const std::pair<OperatorInfoPtr, int> &op_info_pair, … in GetTensorInfo()
|