Searched refs:LoadDataFunc (Results 1 – 3 of 3) sorted by relevance
35 using LoadDataFunc = std::function<int(std::vector<tensor::MSTensor *> inputs, dataset::MSTensorVec… variable90 LoadDataFunc load_func = nullptr) = 0;101 LoadDataFunc load_func = nullptr, int max_steps = INT_MAX) = 0;
58 LoadDataFunc load_func = nullptr) override;59 …int Eval(dataset::Dataset *dataset, std::vector<session::TrainLoopCallBack *> cbs, LoadDataFunc lo…
38 …n(int epochs, Dataset *ds, std::vector<session::TrainLoopCallBack *> cbs, LoadDataFunc load_func) { in Train()109 int TrainLoop::Eval(Dataset *ds, std::vector<session::TrainLoopCallBack *> cbs, LoadDataFunc load_f… in Eval()