Home
last modified time | relevance | path

Searched defs:graph_outputs (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/npu/pass/
Dnpu_transform_pass.cc80 std::vector<mindspore::MSTensor> graph_outputs) { in InsertPostNodes()
163 auto graph_outputs = subgraph->outputs(); in Run() local
/third_party/mindspore/mindspore/lite/micro/coder/
Dcontext.h56 …void set_graph_outputs(const std::vector<Tensor *> &graph_outputs) { graph_outputs_ = graph_output… in set_graph_outputs()
59 std::vector<Tensor *> graph_outputs() const { return graph_outputs_; } in graph_outputs() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dgetnext_tensor_move_elimination.cc75 auto graph_outputs = AnfAlgo::GetAllOutput(graph->output(), {prim::kPrimTupleGetItem}); in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_converter.cc37 void CopyGraphOutputTypeAndShape(const std::vector<session::KernelWithIndex> &graph_outputs, CNodeP… in CopyGraphOutputTypeAndShape()
50 const std::vector<session::KernelWithIndex> &graph_outputs) { in NewTrtNode()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute_ge.cc162 std::vector<GeOperator> graph_outputs{*op}; in BuildSingleOpGraph() local
307 std::vector<MeTensorPtr> graph_outputs = ConvertOutputTensors(op_exec_info, ge_outputs); in RunOpInGE() local
/third_party/mindspore/mindspore/lite/micro/coder/generator/component/
Dcommon_component.cc197 std::vector<Tensor *> graph_outputs = ctx->graph_outputs(); in CodeGraphQuantArgsImplement() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.h86 VectorRef *graph_outputs; member
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc343 const std::vector<AnfNodePtr> graph_outputs = FetchFuncGraphOutput(func_graph, &sub_call_nodes); in FetchOutputByCallNode() local