Home
last modified time | relevance | path

Searched defs:SetBatchSize (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_runtime.h54 void SetBatchSize(int batch_size) { batch_size_ = batch_size; } in SetBatchSize() function
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/acl/
Dmodel_process.cc342 Status ModelProcess::SetBatchSize(const std::vector<MSTensor> &inputs) { in SetBatchSize() function in mindspore::ModelProcess
/third_party/mindspore/mindspore/lite/src/runtime/kernel/ascend310/src/
Dmodel_process.cc324 STATUS ModelProcess::SetBatchSize(const std::vector<mindspore::MSTensor> &inputs) { in SetBatchSize() function in mindspore::kernel::acl::ModelProcess