Home
last modified time | relevance | path

Searched defs:KernelExec (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_utils.h26 class KernelExec; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dtranspose_strategy.cc98 TransAxisFunc kNoNeedTransAxisFunc = [](kernel::KernelExec *, const TransInfoPair &) { return RET_O… in __anond23b4b690502()
99 TransAxisFunc kNotImplementedTransAxisFunc = [](kernel::KernelExec *, const TransInfoPair &) { retu… in __anond23b4b690602()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.h86 KernelExec() { in KernelExec() function
91 explicit KernelExec(std::shared_ptr<MSKernel> kernel) : kernel_(std::move(kernel)) { in KernelExec() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/
Dcontrol_flow_scheduler.cc744 std::unordered_map<kernel::KernelExec *, size_t> *partial_kernel_subgraph_index_map, in SetSubgraphForPartialNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dscheduler.cc1675std::map<const kernel::KernelExec *, bool> *is_kernel_finish) { in ConstructNormalSubGraphs()