Searched refs:GetTreeBatchSize (Results 1 – 5 of 5) sorted by relevance
434 int64_t DatasetOp::GetTreeBatchSize() { in GetTreeBatchSize() function in mindspore::dataset::DatasetOp436 return child_[0]->GetTreeBatchSize(); in GetTreeBatchSize()441 return child_[child_.size() - 1]->GetTreeBatchSize(); in GetTreeBatchSize()
210 int64_t GetTreeBatchSize() override;
167 virtual int64_t GetTreeBatchSize();
569 int64_t BatchOp::GetTreeBatchSize() { in GetTreeBatchSize() function in mindspore::dataset::BatchOp
618 *batch_size = root->GetTreeBatchSize(); in GetBatchSize()