Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/
Dutils_test.cc59 auto output_kernels = kernel::KernelExecUtil::SubgraphOutputNodes(kernels); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dkernel_exec_util.cc356 std::vector<KernelExec *> &&output_kernels, in CreateCustomSubGraph()
389 std::vector<KernelExec *> output_kernels = SubgraphOutputNodes(kernels); in CreateSubGraphKernel() local
Dlite_mindrt.cc209 auto output_kernels = subgraph_kernel->out_nodes(); in UpdateActorOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/
Dcontrol_flow_scheduler.cc624 auto output_kernels = partial_in_kernel->out_kernels(); in IsolatePartialInputs() local