Searched defs:return_inputs (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/utils/ |
| D | build_func_graph.cc | 30 CNodePtr AddReturn(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &return_inputs) { in AddReturn()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/ |
| D | control_node_parser_test.cc | 58 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), backend_node}; in BuildKernelGraph() local
104 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), root_call_node}; in BuildGraphs() local
|
| D | graph_compiler_test.cc | 69 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), sub_node}; in TEST_F() local
|
| D | graph_scheduler_test.cc | 85 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), root_call_node}; in BuildGraphs() local
191 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), add_node}; in BuildAnyTypeGraph() local
|
| D | any_type_kernel_actor_test.cc | 50 std::vector<AnfNodePtr> return_inputs{NewValueNode(prim::kPrimReturn), add_node}; in BuildAnyTypeGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/ |
| D | fusion_inout_test.cc | 93 …FusionInoutTest::AddReturn(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &return_inputs… in AddReturn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/ |
| D | pytorch_model_parser.cc | 191 STATUS BuildReturnNode(const FuncGraphPtr &anf_graph, const std::vector<AnfNodePtr> &return_inputs)… in BuildReturnNode()
212 std::vector<AnfNodePtr> return_inputs; in ConvertGraphOutputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/adapter/acl/mapper/ |
| D | clip_mapper_test.cc | 40 CNodePtr AddReturn(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &return_inputs) { in AddReturn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 216 STATUS BuildReturnNode(const FuncGraphPtr &anf_graph, const std::vector<AnfNodePtr> &return_inputs)… in BuildReturnNode()
420 std::vector<AnfNodePtr> return_inputs; in ConvertGraphOutputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.cc | 269 std::vector<AnfNodePtr> return_inputs; in CreateReturnNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | graph_kernel_splitter.cc | 124 AnfNodePtrList return_inputs = {NewValueNode(prim::kPrimReturn)}; in CreateReturnNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | graph_kernel_recompute.cc | 596 auto return_inputs = return_cnode->inputs(); in FindNodeRecomputeCandidates() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | kernel_graph_mgr.cc | 1476 std::vector<AnfNodePtr> return_inputs = { in ProcessNodeRetFunc() local
|