Searched refs:FuncGraphTransaction (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | manager.h | 54 class FuncGraphTransaction; variable
305 FuncGraphTransaction Transact();
377 class FuncGraphTransaction {
379 explicit FuncGraphTransaction(FuncGraphManager *manager) : manager_(manager) {} in FuncGraphTransaction() function
380 FuncGraphTransaction() : manager_(nullptr) {} in FuncGraphTransaction() function
381 ~FuncGraphTransaction() = default;
383 FuncGraphTransaction(const FuncGraphTransaction &other) = delete;
384 FuncGraphTransaction &operator=(const FuncGraphTransaction &other) = delete;
386 FuncGraphTransaction(FuncGraphTransaction &&other) = default;
387 FuncGraphTransaction &operator=(FuncGraphTransaction &&other) = default;
[all …]
|
| D | manager.cc | 775 void FuncGraphTransaction::SetParameters(FuncGraphPtr fg, const std::vector<AnfNodePtr> ¶ms) { in SetParameters()
779 void FuncGraphTransaction::AddParameter(FuncGraphPtr fg, const AnfNodePtr ¶m) { in AddParameter()
783 void FuncGraphTransaction::InsertFrontParameter(FuncGraphPtr fg, const AnfNodePtr ¶m) { in InsertFrontParameter()
787 bool FuncGraphTransaction::Replace(const AnfNodePtr &old_node, const AnfNodePtr &new_node) { in Replace()
802 void FuncGraphTransaction::SetEdge(const AnfNodePtr &src_node, int k, const AnfNodePtr &v) { in SetEdge()
814 void FuncGraphTransaction::AddEdge(const AnfNodePtr &src_node, const AnfNodePtr &v) { in AddEdge()
823 void FuncGraphTransaction::Commit() { manager_->CommitChanges(std::move(changes_)); } in Commit()
|
| D | func_graph_cloner.h | 99 void SetEdges(const FuncGraphPtr &func_graph, FuncGraphTransaction *tx);
|
| D | func_graph_cloner.cc | 452 void Cloner::SetEdges(const FuncGraphPtr &func_graph, FuncGraphTransaction *tx) { in SetEdges()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 986 std::shared_ptr<FuncGraphTransaction> tr = std::make_shared<FuncGraphTransaction>(manager.get()); in PreposeWithMatmul()
1144 std::shared_ptr<FuncGraphTransaction> tr = std::make_shared<FuncGraphTransaction>(manager.get()); in PreposeWithFullConnection()
1244 std::shared_ptr<FuncGraphTransaction> tr = std::make_shared<FuncGraphTransaction>(manager.get()); in PreposeWithArithmetic()
1446 std::shared_ptr<FuncGraphTransaction> tr = std::make_shared<FuncGraphTransaction>(manager.get()); in MergeParallelSlice()
|
| D | slice_prepose_pass.h | 33 using TransactionPtr = std::shared_ptr<mindspore::FuncGraphTransaction>;
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.cc | 420 const FuncGraphManagerPtr &manager_ptr, FuncGraphTransaction *const tr, const FuncGraphSet &fgs, in TraverseGraphMap()
490 FuncGraphTransaction tr = manager_ptr->Transact(); in WrapPrimitives()
|