Home
last modified time | relevance | path

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

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc41 void UpdateDynamicSequenceType(const AnfNodePtr &output_node, const kernel::KernelTensorPtr &output… in UpdateDynamicSequenceType()
108 auto &output_node = output_nodes_[i].first; in FreeOutputNodeMem() local
260 TensorPtr OutputActor::CreateOutputTensor(const AnfNodePtr &output_node, size_t output_index, size_… in CreateOutputTensor()
374 auto &output_node = output_nodes_[i].first; in UpdateOutputDeviceAddress() local
Dabstract_actor.cc274 const KernelWithIndex &output_node) { in IsOutputAddressPersisted()
Ddata_source_actor.cc60const AnfNodePtr &output_node, OpContext<DeviceTensor> *const context) { in UpdateOutputData()
Dsuper_kernel_actor.cc114 auto &output_node = output_data_nodes_[i]; in Init() local
126 const auto &output_node = output_with_index.first; in Init() local
Dabstract_actor.h179const AnfNodePtr &output_node, OpContext<DeviceTensor> *const context) {} in UpdateOutputData()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc43 auto output_node = partial_fg->output(); in HasCallAfter() local
50 auto output_node = fg->output(); in GetFgOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dupdate_state_formatter.cc187 auto output_node = func_graph->output()->cast<CNodePtr>(); in FilterIndexes() local
195 auto output_node = func_graph->output()->cast<CNodePtr>(); in FindAllOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Dgraph_output_name_keeper.cc46 auto output_node = original_outputs_[j]; in SaveOriginalOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Ddefault_graph_compiler.cc293 auto output_node = graph_outputs[i]; in CreateExecPlanOutputs() local
316 auto &output_node = segments_outputs[i][j]; in CreateExecPlanOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc605 auto &output_node = output_with_index.first; in AllocOutputHostMemory() local
777 auto &output_node = output_with_index.first; in BuildOutputDataGeTensor() local
809 auto output_node = output_with_index.first; in CreateOutputDeviceAddress() local
873 auto &output_node = output_with_index.first; in AllocOutputMemory() local
885 auto &output_node = output_with_index.first; in AllocOutputMemory() local
1162 void SetOutput(GeDeviceResManager *res_manager, GeTensor *ge_output, const AnfNodePtr &output_node,… in SetOutput()
1582 auto output_node = iter->second.graph_outputs[idx].first.lock(); in GenerateOutputGeTensor() local
Dge_utils.cc119 auto &output_node = output_with_index.first; in GetComputeGraphReuseOptions() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dgraph_split_api.cc223 int DetermineOutputFormat(const api::AnfNodePtr &output_node, Format *format) { in DetermineOutputFormat()
383 for (auto &output_node : subgraph_outputs) { in FillSubgraphOutputsFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Dlink_custom_op.cc172 auto output_node = return_node->input(kFirstDataInputIndex); in AttachDependNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc161 auto output_node = node_index_set.first; in GetMaxSubGraph() local
190 auto output_node = node_index_set.first; in GetOriginRecomputeAndTargetNodes() local
481 for (const auto &output_node : tuple_getitem_output_nodes) { in SetRecomputedAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dswitch_not_cut.cc64 const auto &output_node = sub_graph->return_node()->input(1); in IsValidInlinePartial() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc277 auto output_node = output_pair.first; in InitSubGraphOutNode() local
375 auto output_node = belong_anf_->output(); in FindAfterSubGraphInBelongAnf() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc66 auto output_node = return_node->input(kInputNodeIndex); in SetGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dsplit_matmul_comm_elementwise_fp.cc66 auto output_node = output_node_set.front().first; in PatternFilter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/gsm/
Dmem_usage_analyzer.cc89 auto output_node = output_with_index.first; in AddOutputNodeInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/jit/
Djit_dfunctor.cc164 const auto &output_node = make_tuple_node->input(forward_output_index); in GetForwardOutNodeAndBpropGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dexec_order_builder.cc258 for (const auto &output_node : it->second) { in EnqueueReadyNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/
Dom_model_parser.cc123 CNodePtr output_node = nullptr; in CreateGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc547 for (auto output_node : tflite_subgraph->outputs) { in ConvertGraphOutputs() local
625 auto output_node = output_nodes.at(i); in ConvertGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_shape_fusion_pass.cc120 auto output_node = candidate_nodes.front(); in FusePostNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_recompute.cc50 auto output_node = func_graph->output()->cast<CNodePtr>(); in GetOutput() local
646 auto output_node = new_funcgraph->output()->cast<CNodePtr>(); in CloneGraph() local

123