Home
last modified time | relevance | path

Searched defs:output_node (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Doutput_actor.cc23 TensorPtr CreateOutputTensor(const AnfNodePtr &output_node, size_t output_index, size_t output_posi… in CreateOutputTensor()
85 auto &output_node = output_nodes_[i].first; in UpdateOutputDeviceAddress() local
108 void OutputActor::CollectOutput(const AnfNodePtr &output_node, size_t output_index, size_t output_p… in CollectOutput()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc38 auto output_node = partial_fg->output(); in HasCallAfter() local
45 auto output_node = fg->output(); in GetFgOutput() local
Dmindspore_importer.cc127 auto output_node = return_node->input(i); in GetFuncGraphOutputName() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc148 auto output_node = node_index_set.first; in GetMaxSubGraph() local
177 auto output_node = node_index_set.first; in GetOriginRecomputeAndTargetNodes() local
318 for (const auto &output_node : tuple_getitem_output_nodes) { in SetRecomputedAttr() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc180 auto output_node = func_graph->output()->cast<CNodePtr>(); in FilterIndexes() local
188 auto output_node = func_graph->output()->cast<CNodePtr>(); in FindAllOutputs() local
Dgraph_kernel_helper.cc849 auto output_node = func_graph->output(); in AnfGraph2LiteGraph() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc62 auto output_node = return_node->input(kInputNodeIndex); in SetGraphOutput() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.cc360 py::object StructureOutput(const AnfNodePtr &output_node, const py::tuple &data, size_t *count) { in StructureOutput()
436 AnfNodePtr output_node = graph->get_return()->input(1); in DoExecGraph() local
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc275 auto output_node = output_pair.first; in InitSubGraphOutNode() local
374 auto output_node = belong_anf_->output(); in FindAfterSubGraphInBelongAnf() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dpynative_dfunctor.cc108 const auto &output_node = make_tuple_node->input(forward_output_index); in GetForwardOutNodeAndBpropGraph() local
Ddfunctor.cc432 auto output_node = primal_graph_->output(); in MapMorphism() local
Dkpynative.cc441 void KPynativeCellImpl::UpdateOutputNodeOfTopCell(const AnfNodePtr &output_node) { in UpdateOutputNodeOfTopCell()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc467 for (auto output_node : tflite_subgraph->outputs) { in ConvertGraphOutputs() local
536 auto output_node = output_nodes.at(i); in ConvertGraphOutputs() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc197 TensorPtr CreateOutputTensor(const AnfNodePtr &output_node, size_t output_index) { in CreateOutputTensor()
239 auto &output_node = item_with_index.first; in UpdateOutputDeviceAddress() local
871 void MindRTBackend::ConstructOutputs(const AnfNodePtr &output_node, in ConstructOutputs()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dtf_bidirection_gru_cf_fusion.cc213 auto output_node = GetPostProcessNode(func_graph, get_item_node, gru_node->fullname_with_scope()); in Process() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dincorporate_getitem.h67 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator() local
119 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator() local
Dbranch_culling.cc265 …eSwitchDependNode(const FuncGraphPtr &graph, const AnfNodePtr &cond, const AnfNodePtr &output_node, in GenerateSwitchDependNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc426 for (const auto &output_node : caffeInspector.GetGraphOutput()) { in ConvertGraphOutputs() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dcoco_op.cc256 …CocoOp::SearchNodeInJson(const nlohmann::json &input_tree, std::string node_name, T *output_node) { in SearchNodeInJson()
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc980 const mind_ir::ValueInfoProto &output_node = importProto.output(out_size); in BuildReturnForFuncGraph() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_bidirection_gru_fusion.cc825 auto output_node = GetPostProcessNode(func_graph, get_item_node, gru_node->fullname_with_scope()); in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc922 for (auto &output_node : outputs) { in ReplaceNode() local
Dascend_session.cc192 const auto &output_node = output_kernel_with_index.first; in GenOpOutputStubTensor() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc517 auto output_node = output_node_with_index.first; in RunOpAssignOutputNodeMemory() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc1535 for (const auto &output_node : output_nodes) { in CalOutputSize() local

12