Searched refs:batch_num_ (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | davinci_model.h | 43 batch_num_(batch_num), in task_info_list_()
57 uint32_t GetBatchNum() const { return batch_num_; } in GetBatchNum()
82 uint32_t batch_num_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.h | 139 : epoch_num_(ep), batch_num_(bat), total_batch_num_(cur), ctrl_(ctrl) {} in CBatchInfo()
144 int64_t batch_num_; // i-th batch since the start of current epoch. i starts from 0 member
147 const int64_t get_batch_num() const { return batch_num_; } in get_batch_num()
279 int64_t batch_num_; variable
|
| D | batch_op.cc | 64 batch_num_(0), in BatchOp()
85 batch_num_(0), in BatchOp()
583 RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(0, batch_num_, batch_cnt_))); in GetNextRowPullMode()
601 batch_num_++; in GetNextRowPullMode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/datasetops/ |
| D | bindings.cc | 31 return py::make_tuple(p.epoch_num_, p.batch_num_, p.total_batch_num_); in __anon29e9d4ff0102()
|