Home
last modified time | relevance | path

Searched refs:cache_lookup_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/
Drepeat_pass.cc32 … : num_repeats_(1), num_epochs_(1), is_merge_(false), is_cached_(false), cache_lookup_(nullptr) {} in RepeatPass()
101 if (is_merge_ && cache_lookup_) { in VisitAfter()
102 cache_lookup_->SetTotalRepeats(num_repeats_); in VisitAfter()
103 cache_lookup_->SetNumEpochs(num_epochs_); in VisitAfter()
104 cache_lookup_.reset(); in VisitAfter()
178 if (cache_lookup_) { in VisitAfter()
179 cache_lookup_->SetTotalRepeats(num_repeats_); in VisitAfter()
180 cache_lookup_->SetNumEpochs(num_epochs_); in VisitAfter()
184cache_lookup_.reset(); // If we are not repeated then the saved lookup is no longer needed or used in VisitAfter()
203 cache_lookup_ = std::static_pointer_cast<DatasetNode>(node); in VisitAfter()
Drepeat_pass.h126 std::shared_ptr<DatasetNode> cache_lookup_; // A save area for a cache lookup node variable