Home
last modified time | relevance | path

Searched refs:tree_state_ (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dexecution_tree.h172 bool IsEpochEnd() const { return tree_state_ == TreeState::kDeTStateEpochEnd; } in IsEpochEnd()
175 void SetEpochEnd() { tree_state_ = TreeState::kDeTStateEpochEnd; } in SetEpochEnd()
178 void SetExecuting() { tree_state_ = TreeState::kDeTStateExecuting; } in SetExecuting()
181 void SetFinished() { tree_state_ = TreeState::kDeTStateFinished; } in SetFinished()
185 bool isFinished() const { return tree_state_ == TreeState::kDeTStateFinished; } in isFinished()
190 …return tree_state_ == TreeState::kDeTStatePrepared || tree_state_ == TreeState::kDeTStateExecuting… in isPrepared()
191 tree_state_ == TreeState::kDeTStateFinished; in isPrepared()
212 TreeState tree_state_; // Tracking the current tree state variable
Dexecution_tree.cc34 ExecutionTree::ExecutionTree() : id_count_(0), tree_state_(kDeTStateInit) { in ExecutionTree()
76 if (tree_state_ != kDeTStateInit && tree_state_ != kDeTStateBuilding) { in AssociateNode()
78 …id tree state for adding a node. Current state: " + std::to_string(static_cast<int>(tree_state_)) + in AssociateNode()
85 tree_state_ = kDeTStateBuilding; in AssociateNode()
100 if (tree_state_ != kDeTStateBuilding) { in AssignRoot()
102 …state for assigning a root node. Current state: " + std::to_string(static_cast<int>(tree_state_)) + in AssignRoot()
184 if (tree_state_ != kDeTStatePrepared) { in Launch()
186 …d tree state for launching tree. Current state: " + std::to_string(static_cast<int>(tree_state_)) + in Launch()
216 tree_state_ = kDeTStateExecuting; in Launch()
282 tree_state_ = kDeTStatePrepared; in Prepare()
Dtree_adapter.cc41 TreeAdapter::TreeAdapter(UsageFlag usage) : usage_(usage), launched_(false), tree_state_(kCompileSt… in TreeAdapter()
178 tree_state_ = kCompileStateIRGraphBuilt; in Compile()
194 tree_state_ = kCompileStateIRTreeCloned; in Compile()
208 tree_state_ = kCompileStateOptimized; in Compile()
214 tree_state_ = kCompileStateReady; in Compile()
Dtree_adapter.h111 CompileState tree_state_; variable