Home
last modified time | relevance | path

Searched defs:KernelCallBack (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/
Dentrance_subgraph_kernel.cc21 int EntranceSubGraphKernel::Execute(const KernelCallBack &, const KernelCallBack &) { return lite::… in Execute()
Dexit_subgraph_kernel.cc21 int ExitSubGraphKernel::Execute(const KernelCallBack &, const KernelCallBack &) { return lite::RET_… in Execute()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/infer/
Dkernel_callback.h22 using KernelCallBack = mindspore::lite::KernelCallBack; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.h44 using KernelCallBack = std::function<bool(std::vector<lite::Tensor *> inputs, std::vector<lite::Ten… variable