Searched refs:num_epochs_ (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/ |
| D | repeat_pass.cc | 32 … : num_repeats_(1), num_epochs_(1), is_merge_(false), is_cached_(false), cache_lookup_(nullptr) {} in RepeatPass()
64 num_epochs_ = node->Count(); in Visit()
69 num_repeats_ *= num_epochs_; in Visit()
103 cache_lookup_->SetNumEpochs(num_epochs_); in VisitAfter()
111 node->SetNumEpochs(num_epochs_); in VisitAfter()
127 node->SetNumEpochs(num_epochs_); in VisitAfter()
144 node->SetNumEpochs(num_epochs_); in VisitAfter()
168 node->SetNumEpochs(num_epochs_); in VisitAfter()
180 cache_lookup_->SetNumEpochs(num_epochs_); in VisitAfter()
183 node->SetNumEpochs(num_epochs_); in VisitAfter()
[all …]
|
| D | repeat_pass.h | 124 int32_t num_epochs_; // To save the total number of epochs variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | root_node.cc | 33 num_epochs_ = 0; in RootNode()
38 node->SetNumEpochs(num_epochs_); in Copy()
54 if (num_epochs_ <= 0 && num_epochs_ != -1) { in ValidateParams()
56 …e: num_epochs should be either -1 or positive integer, num_epochs: " + std::to_string(num_epochs_); in ValidateParams()
|
| D | root_node.h | 32 RootNode() : DatasetNode(), num_epochs_(0) {} in RootNode()
58 int32_t num_epochs() { return num_epochs_; } in num_epochs()
61 void SetNumEpochs(int32_t num_epochs) { num_epochs_ = num_epochs; } in SetNumEpochs()
80 int32_t num_epochs_;
|
| D | dataset_node.h | 311 void SetNumEpochs(int32_t num_epochs) { num_epochs_ = num_epochs; } in SetNumEpochs()
319 int32_t GetNumEpochs() const { return num_epochs_; } in GetNumEpochs()
323 int32_t GetNumRepeatsPerEpoch() const { return total_repeats_ / num_epochs_; } in GetNumRepeatsPerEpoch()
334 int32_t num_epochs_; // Number of epochs variable
|
| D | dataset_node.cc | 235 num_epochs_(1) {
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | epoch_ctrl_pass.h | 75 int32_t num_epochs() { return num_epochs_; } in num_epochs()
79 int32_t num_epochs_; variable
|
| D | epoch_ctrl_pass.cc | 28 : injection_point_(nullptr), num_epochs_(-1) {} in InjectionFinder()
38 num_epochs_ = node->num_epochs(); in Visit()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | tree_consumer.h | 64 explicit IteratorConsumer(int32_t num_epochs = -1) : TreeConsumer(), num_epochs_(num_epochs) {} in TreeConsumer()
91 int32_t num_epochs_;
157 explicit ToDevice(int32_t num_epochs = -1) : TreeConsumer(), num_epochs_(num_epochs) {} in TreeConsumer()
187 int32_t num_epochs_;
|
| D | tree_consumer.cc | 47 return tree_adapter_->Compile(std::move(d), num_epochs_); in Init()
151 …:Init(std::shared_ptr<DatasetNode> d) { return tree_adapter_->Compile(std::move(d), num_epochs_); } in Init()
|