Searched refs:graph_outputs (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | getnext_tensor_move_elimination.cc | 75 auto graph_outputs = AnfAlgo::GetAllOutput(graph->output(), {prim::kPrimTupleGetItem}); in Process() local
76 auto iter = std::find(graph_outputs.begin(), graph_outputs.end(), next_node); in Process()
77 if (iter != graph_outputs.end()) { in Process()
|
| /third_party/mindspore/mindspore/lite/micro/coder/generator/component/ |
| D | common_component.cc | 38 std::vector<Tensor *> outputs = ctx->graph_outputs(); in CodeSessionCompileGraph()
130 std::vector<Tensor *> outputs = ctx->graph_outputs(); in CodeCopyOutputsImplement()
197 std::vector<Tensor *> graph_outputs = ctx->graph_outputs(); in CodeGraphQuantArgsImplement() local
198 if (graph_outputs.empty()) { in CodeGraphQuantArgsImplement()
202 Tensor *out_tensor = graph_outputs.at(kOutputIndex); in CodeGraphQuantArgsImplement()
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_transform_pass.cc | 80 std::vector<mindspore::MSTensor> graph_outputs) { in InsertPostNodes() argument
83 find(graph_outputs.begin(), graph_outputs.end(), op->outputs()[0]) != graph_outputs.end()) { in InsertPostNodes()
163 auto graph_outputs = subgraph->outputs(); in Run() local
188 ret = InsertPostNodes(op, &post_ops, graph_outputs); in Run()
|
| D | npu_transform_pass.h | 35 …tNodes(NPUOp *op, std::vector<NPUOp *> *trans_ops, std::vector<mindspore::MSTensor> graph_outputs);
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute_ge.cc | 162 std::vector<GeOperator> graph_outputs{*op}; in BuildSingleOpGraph() local
165 (void)graph->SetInputs(graph_input_nodes).SetOutputs(graph_outputs); in BuildSingleOpGraph()
307 std::vector<MeTensorPtr> graph_outputs = ConvertOutputTensors(op_exec_info, ge_outputs); in RunOpInGE() local
308 size_t output_size = graph_outputs.size(); in RunOpInGE()
311 MS_EXCEPTION_IF_NULL(graph_outputs[i]); in RunOpInGE()
312 result[i] = *graph_outputs[i]; in RunOpInGE()
|
| /third_party/mindspore/mindspore/lite/micro/coder/ |
| D | context.h | 56 …void set_graph_outputs(const std::vector<Tensor *> &graph_outputs) { graph_outputs_ = graph_output… in set_graph_outputs() argument
59 std::vector<Tensor *> graph_outputs() const { return graph_outputs_; } in graph_outputs() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_converter.cc | 37 void CopyGraphOutputTypeAndShape(const std::vector<session::KernelWithIndex> &graph_outputs, CNodeP… in CopyGraphOutputTypeAndShape() argument
40 for (const auto &item : graph_outputs) { in CopyGraphOutputTypeAndShape()
50 const std::vector<session::KernelWithIndex> &graph_outputs) { in NewTrtNode() argument
61 CopyGraphOutputTypeAndShape(graph_outputs, trt_node); in NewTrtNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.h | 86 VectorRef *graph_outputs; member
|
| D | session_basic.cc | 1339 MS_EXCEPTION_IF_NULL(graph_output_info->graph_outputs); in HandleOpOutputs()
1357 const VectorRef *cur_vector_ref = graph_output_info->graph_outputs; in HandleOpOutputs()
2279 graph_output_info.graph_outputs = outputs; in RunOpsInGraphImpl()
2280 …CreateOutputPlaceholder(kernel_graph, inputs, graph_output_info.graph_outputs, &graph_output_info.… in RunOpsInGraphImpl()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 343 const std::vector<AnfNodePtr> graph_outputs = FetchFuncGraphOutput(func_graph, &sub_call_nodes); in FetchOutputByCallNode() local
344 for (const auto &graph_output : graph_outputs) { in FetchOutputByCallNode()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 722 graph_output_info.graph_outputs = outputs; in RunGraphBySingleOp()
|