| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_subgraph.cc | 155 if (runtime_->GetBatchSize() == 0) { in SetTensorRTNetworkInput()
157 MS_LOG(INFO) << "batch size init as " << runtime_->GetBatchSize(); in SetTensorRTNetworkInput()
162 if (input_dims.d[n] == runtime_->GetBatchSize()) { in SetTensorRTNetworkInput()
428 if (runtime_->GetBatchSize() <= 0) { in Execute()
457 if (runtime_->GetBatchSize() != new_shape[output_batchsize_index_]) { in Execute()
458 new_shape[output_batchsize_index_] = runtime_->GetBatchSize(); in Execute()
|
| D | tensorrt_runtime.h | 52 int GetBatchSize() { return batch_size_; } in GetBatchSize() function
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | c_api_datasets_test.cc | 136 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
139 EXPECT_EQ(ds->GetBatchSize(), 2); in TEST_F()
142 EXPECT_EQ(ds->GetBatchSize(), 3); in TEST_F()
|
| D | c_api_dataset_cifar_test.cc | 149 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
161 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
224 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
|
| D | c_api_dataset_sbu_test.cc | 145 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
157 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
|
| D | c_api_dataset_usps_test.cc | 214 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
225 EXPECT_EQ(ds->GetBatchSize(), 1); in TEST_F()
|
| D | c_api_dataset_cityscapes_test.cc | 139 EXPECT_EQ(ds1->GetBatchSize(), 1); in TEST_F()
144 EXPECT_EQ(ds2->GetBatchSize(), 1); in TEST_F()
|
| D | c_api_dataset_album_test.cc | 129 int64_t batch_size = ds->GetBatchSize(); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 107 RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(0, 0, 0))); in operator ()()
117 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()()
133 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()()
326 Status BatchOp::GetBatchSize(int32_t *batch_size, CBatchInfo info) { in GetBatchSize() function in mindspore::dataset::BatchOp
583 RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(0, batch_num_, batch_cnt_))); in GetNextRowPullMode()
|
| D | batch_op.h | 249 Status GetBatchSize(int32_t *batch_size, CBatchInfo info);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/consumer/ |
| D | bindings.cc | 81 THROW_IF_ERROR(self.GetBatchSize(&batch_size)); in __anona1cf9bfd0602()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | tree_consumer.h | 199 Status GetBatchSize(int64_t *batch_size);
|
| D | tree_consumer.cc | 613 Status TreeGetters::GetBatchSize(int64_t *batch_size) { in GetBatchSize() function in mindspore::dataset::TreeGetters
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 94 int64_t GetBatchSize();
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 526 int64_t Dataset::GetBatchSize() { in GetBatchSize() function in mindspore::dataset::Dataset
531 RETURN_SECOND_IF_ERROR(tree_getters_->GetBatchSize(&batch_size), -1); in GetBatchSize()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 114 int64_t GetBatchSize();
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 1805 self._batch_size = runtime_getter[0].GetBatchSize()
|