Searched refs:EpochCtrlNode (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | epoch_ctrl_node.cc | 31 EpochCtrlNode::EpochCtrlNode(std::shared_ptr<DatasetNode> child, int32_t num_epochs) : RepeatNode()… in EpochCtrlNode() function in mindspore::dataset::EpochCtrlNode
37 std::shared_ptr<DatasetNode> EpochCtrlNode::Copy() { in Copy()
38 auto node = std::make_shared<EpochCtrlNode>(repeat_count_); in Copy()
42 void EpochCtrlNode::Print(std::ostream &out) const { in Print()
47 Status EpochCtrlNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
57 Status EpochCtrlNode::ValidateParams() { in ValidateParams()
74 Status EpochCtrlNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
76 return p->Visit(shared_from_base<EpochCtrlNode>(), modified); in Accept()
80 Status EpochCtrlNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
82 return p->VisitAfter(shared_from_base<EpochCtrlNode>(), modified); in AcceptAfter()
|
| D | epoch_ctrl_node.h | 31 class EpochCtrlNode : public RepeatNode {
37 explicit EpochCtrlNode(int32_t num_epochs) : RepeatNode() { repeat_count_ = num_epochs; } in EpochCtrlNode() function
40 EpochCtrlNode(std::shared_ptr<DatasetNode> child, int32_t num_epochs);
43 ~EpochCtrlNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/ |
| D | generator_node_pass.h | 49 Status Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified) override;
67 Status VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified) override;
|
| D | generator_node_pass.cc | 55 Status GeneratorNodePass::Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in Visit()
99 Status GeneratorNodePass::VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in VisitAfter()
|
| D | repeat_pass.h | 50 Status Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified) override;
76 Status VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified) override;
|
| D | repeat_pass.cc | 60 Status RepeatPass::Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in Visit()
122 Status RepeatPass::VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 39 class EpochCtrlNode; variable
179 virtual Status Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified);
180 virtual Status VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified);
|
| D | pass.cc | 170 Status IRNodePass::Visit(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in Visit()
173 Status IRNodePass::VisitAfter(std::shared_ptr<EpochCtrlNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | epoch_ctrl_pass.cc | 90 auto epoch_ctrl_node = std::make_shared<EpochCtrlNode>(num_epochs); in RunOnTree()
|