Searched defs:SetupVirtualBatch (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/nnacl/ |
| D | nnacl_batchnorm.cc | 25 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/ |
| D | lite_kernel.h | 163 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/ |
| D | model.cc | 421 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/ |
| D | model.cc | 647 Status Model::SetupVirtualBatch(int virtual_batch_multiplier, float lr, float momentum) { in SetupVirtualBatch() function in mindspore::Model
|
| D | model_impl.cc | 977 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/ |
| D | train_session.cc | 1077 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/ |
| D | mslite_model_napi.cc | 2429 napi_value MSLiteModelNapi::SetupVirtualBatch(napi_env env, napi_callback_info info) { in SetupVirtualBatch() function in mindspore::MSLiteModelNapi
|