Home
last modified time | relevance | path

Searched defs:GetBatchSize (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Ddata_preprocessor.h44 size_t GetBatchSize() const { return batch_size_; } in GetBatchSize() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_runtime.h56 int GetBatchSize() { return batch_size_; } in GetBatchSize() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dcontext_util.cc76 std::vector<size_t> GetBatchSize(const std::string &batch_size) { in GetBatchSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/consumers/
Dpull_based_tree_consumer.cc183 Status TreeGetters::GetBatchSize(int64_t *batch_size) { in GetBatchSize() function in mindspore::dataset::TreeGetters
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.cc447 Status BatchOp::GetBatchSize(int32_t *batch_size, CBatchInfo info) { in GetBatchSize() function in mindspore::dataset::BatchOp
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc585 int64_t Dataset::GetBatchSize() { in GetBatchSize() function in mindspore::dataset::Dataset