Home
last modified time | relevance | path

Searched defs:SetupVirtualBatch (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/nnacl/
Dnnacl_batchnorm.cc25 int BatchNormKernel::SetupVirtualBatch(int virtual_batch_multiplier, int momentum) { in SetupVirtualBatch() function in mindspore::nnacl::BatchNormKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dlite_kernel.h163 virtual int SetupVirtualBatch(int, int) { return mindspore::lite::RET_OK; } in SetupVirtualBatch() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model/
Dmodel.cc421 Status Model::SetupVirtualBatch(int virtual_batch_multiplier, float lr, float momentum) { in SetupVirtualBatch() function in mindspore::Model
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/model/
Dmodel.cc647 Status Model::SetupVirtualBatch(int virtual_batch_multiplier, float lr, float momentum) { in SetupVirtualBatch() function in mindspore::Model
Dmodel_impl.cc977 Status ModelImpl::SetupVirtualBatch(int virtual_batch_multiplier, float lr, float momentum) { in SetupVirtualBatch() function in mindspore::ModelImpl
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_session.cc1077 int TrainSession::SetupVirtualBatch(int virtual_batch_multiplier, float lr, float momentum) { in SetupVirtualBatch() function in mindspore::lite::TrainSession
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/js_api/
Dmslite_model_napi.cc2429 napi_value MSLiteModelNapi::SetupVirtualBatch(napi_env env, napi_callback_info info) { in SetupVirtualBatch() function in mindspore::MSLiteModelNapi