| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel_util.cc | 67 std::vector<kernel::LiteKernel *> output_nodes; in SubgraphOutputNodes() local
72 if (!lite::IsContain(output_nodes, kernel)) { in SubgraphOutputNodes()
73 output_nodes.push_back(kernel); in SubgraphOutputNodes()
79 !lite::IsContain(output_nodes, kernel)) { in SubgraphOutputNodes()
80 output_nodes.push_back(kernel); in SubgraphOutputNodes()
83 return output_nodes; in SubgraphOutputNodes()
121 std::vector<kernel::LiteKernel *> output_nodes = SubgraphOutputNodes(kernels); in SubgraphOutputTensors() local
122 for (const auto &output_kernel : output_nodes) { in SubgraphOutputTensors()
|
| D | sub_graph_split.cc | 310 std::vector<uint32_t> output_nodes; in IsNodeSubGraphHead() local
313 output_nodes.insert(output_nodes.end(), cur_nodes.begin(), cur_nodes.end()); in IsNodeSubGraphHead()
315 if (output_indexes.size() == 1 && output_nodes.size() == 1) { in IsNodeSubGraphHead()
318 for (uint32_t out_n : output_nodes) { in IsNodeSubGraphHead()
329 std::vector<uint32_t> output_nodes; in IsNodeSubGraphHeadWithRoot() local
332 output_nodes.insert(output_nodes.end(), cur_nodes.begin(), cur_nodes.end()); in IsNodeSubGraphHeadWithRoot()
334 for (uint32_t out_n : output_nodes) { in IsNodeSubGraphHeadWithRoot()
898 … const std::map<int, OpParameter *> *op_parameters, std::vector<size_t> *output_nodes) in SearchSubGraph() argument
899 …: output_nodes_(output_nodes), context_(context), src_tensors_(src_tensors), op_parameters_(op_par… in SearchSubGraph()
|
| D | sub_graph_split.h | 86 … const std::map<int, OpParameter *> *op_parameters, std::vector<size_t> *output_nodes);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.h | 36 output_nodes(std::move(out)) {} in FusionScopeInfo()
41 std::vector<AnfNodePtr> output_nodes; member
|
| D | kernel_fusion.cc | 90 …if (!TbeKernelBuild::GetIOSize(fusion_op["op_list"], fusion_scope_iter.output_nodes, &input_size_l… in KernelFusion()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_model_parser.cc | 1077 STATUS TFModelParser::GetGraphOutputNames(std::vector<AnfNodePtr> *output_nodes) { in GetGraphOutputNames() argument
1095 output_nodes->push_back(anf_node); in GetGraphOutputNames()
1120 std::vector<AnfNodePtr> output_nodes; in ConvertRootGraphOutputs() local
1121 auto status = GetGraphOutputNames(&output_nodes); in ConvertRootGraphOutputs()
1131 status = MakeAnfGraphOutputs(output_nodes, func_graph); in ConvertRootGraphOutputs()
1140 STATUS TFModelParser::MakeAnfGraphOutputs(const std::vector<AnfNodePtr> &output_nodes, const FuncGr… in MakeAnfGraphOutputs() argument
1141 if (output_nodes.empty() || anf_graph == nullptr) { in MakeAnfGraphOutputs()
1145 if (output_nodes.size() > 1) { in MakeAnfGraphOutputs()
1146 std::vector<AnfNodePtr> make_tuple_inputs = output_nodes; in MakeAnfGraphOutputs()
1179 std::vector<AnfNodePtr> op_inputs{value_node, output_nodes.front()}; in MakeAnfGraphOutputs()
|
| D | tf_model_parser.h | 74 STATUS GetGraphOutputNames(std::vector<AnfNodePtr> *output_nodes);
97 …static STATUS MakeAnfGraphOutputs(const std::vector<AnfNodePtr> &output_nodes, const FuncGraphPtr …
|
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | func_graph_subgraph.cc | 471 std::vector<AnfNodePtr> output_nodes; in SetFuncGraphOutput() local
472 output_nodes.insert(output_nodes.end(), outputs.begin(), outputs.end()); in SetFuncGraphOutput()
473 return lite::SetFuncGraphOutput(graph, output_nodes); in SetFuncGraphOutput()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 465 std::vector<AnfNodePtr> output_nodes; in ConvertGraphOutputs() local
474 output_nodes.emplace_back(cnode); in ConvertGraphOutputs()
483 std::vector<AnfNodePtr> make_tuple_inputs = output_nodes; in ConvertGraphOutputs()
518 output_nodes.emplace_back(cnode); in ConvertGraphOutputs()
535 for (size_t i = 0; i < output_nodes.size(); i++) { in ConvertGraphOutputs()
536 auto output_node = output_nodes.at(i); in ConvertGraphOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.h | 48 …c bool GetIOSize(const nlohmann::json &fusion_op_list, const std::vector<AnfNodePtr> &output_nodes,
52 const std::vector<mindspore::AnfNodePtr> &output_nodes,
|
| D | tbe_kernel_build.cc | 1531 const std::vector<mindspore::AnfNodePtr> &output_nodes, in CalOutputSize() argument
1535 for (const auto &output_node : output_nodes) { in CalOutputSize()
1567 const std::vector<mindspore::AnfNodePtr> &output_nodes, in GetIOSize() argument
1576 return CalOutputSize(fusion_op_list, output_nodes, output_size_list); in GetIOSize()
|
| D | ascend_kernel_compile.cc | 673 …if (!TbeKernelBuild::GetIOSize(fusion_op[kOpList], fusion_scope_iter.output_nodes, &input_size_lis… in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 224 void UpdateOutput(const std::vector<session::KernelWithIndex> &output_nodes, VectorRef *const outpu… in UpdateOutput() argument
226 for (auto &item_with_index : output_nodes) { in UpdateOutput()
236 void UpdateOutputDeviceAddress(const std::vector<session::KernelWithIndex> &output_nodes, in UpdateOutputDeviceAddress() argument
238 for (auto &item_with_index : output_nodes) { in UpdateOutputDeviceAddress()
1080 const auto &output_nodes = graph_compiler_->GetGraphOutputNodes(graph->graph_id()); in RunGraph() local
1082 UpdateOutput(output_nodes, outputs); in RunGraph()
1102 UpdateOutputDeviceAddress(output_nodes, graph_compiler_info.device_contexts_.front()); in RunGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 416 auto output_nodes = graph->outputs(); in CompileGraph() local
418 for (auto &node : output_nodes) { in CompileGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 298 auto output_nodes = kernel_graph->outputs(); in CreateOutputTensors() local
299 for (const auto &item : output_nodes) { in CreateOutputTensors()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 509 auto output_nodes = graph.outputs(); in RunOpAssignOutputNodeMemory() local
510 if (pre_output_tensors.size() != output_nodes.size()) { in RunOpAssignOutputNodeMemory()
512 … << "] is not equal to the size of output nodes of graph [" << output_nodes.size() << "]"; in RunOpAssignOutputNodeMemory()
515 for (size_t i = 0; i < output_nodes.size(); ++i) { in RunOpAssignOutputNodeMemory()
516 auto output_node_with_index = AnfAlgo::VisitKernel(output_nodes[i], 0); in RunOpAssignOutputNodeMemory()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 832 std::vector<AnfNodePtr> output_nodes; in GetOutputNodes() local
834 …(void)std::transform(it->second.begin(), it->second.end(), std::back_inserter(output_nodes), trans… in GetOutputNodes()
835 return output_nodes; in GetOutputNodes()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 651 const auto &output_nodes = AnfAlgo::GetAllOutput(graph->output(), {prim::kPrimTupleGetItem}); in SaveGraphOutputNode() local
652 for (const auto &node : output_nodes) { in SaveGraphOutputNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 53 for (auto &node : info.output_nodes) { in DumpFusionScopeInfo()
|