Home
last modified time | relevance | path

Searched refs:IsNStepEndNeeded (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/callback/
Dds_callback.h91 virtual bool IsNStepEndNeeded() = 0;
Dpy_ds_callback.h106 bool IsNStepEndNeeded() override;
Dpy_ds_callback.cc45 bool PyDSCallback::IsNStepEndNeeded() { return step_end_needed_; } in IsNStepEndNeeded() function in mindspore::dataset::PyDSCallback
Dcallback_manager.cc150 …if (callbacks_[ind]->IsNStepEndNeeded() && (cb_param.cur_epoch_step_num_ - 1) % callbacks_[ind]->s… in StepEnd()
/third_party/mindspore/tests/ut/cpp/dataset/
Dir_callback_test.cc96 bool IsNStepEndNeeded() override { return step_end_; } in IsNStepEndNeeded() function in mindspore::dataset::test::TestCallback