Home
last modified time | relevance | path

Searched refs:ExecuteVirtualBatch (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dsgd.cc126 error_code = sgd_kernel->ExecuteVirtualBatch(task_id); in SgdRun()
128 error_code = sgd_kernel->ExecuteVirtualBatch(task_id); in SgdRun()
144 error_code = sgd_kernel->ExecuteVirtualBatch(task_id); in SgdRunInit()
Dapply_momentum.cc78 error_code = applyMomentum_kernel->ExecuteVirtualBatch(task_id); in ApplyMomentumRun()
80 error_code = applyMomentum_kernel->ExecuteVirtualBatch(task_id); in ApplyMomentumRun()
Dadam.cc97 error_code = adam_kernel->ExecuteVirtualBatch(task_id); in AdamRun()
99 error_code = adam_kernel->ExecuteVirtualBatch(task_id); in AdamRun()
/third_party/mindspore/mindspore/lite/src/train/
Doptimizer_kernel.h150 int ExecuteVirtualBatch(int task_id) { in ExecuteVirtualBatch() function