Home
last modified time | relevance | path

Searched refs:output_node (Results 1 – 25 of 33) 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() argument
24 MS_EXCEPTION_IF_NULL(output_node); in CreateOutputTensor()
25 MS_LOG(INFO) << "Create output tensor, output node: " << output_node->fullname_with_scope() in CreateOutputTensor()
30 auto type_id = AnfAlgo::GetOutputInferDataType(output_node, output_index); in CreateOutputTensor()
32 auto shape = AnfAlgo::GetOutputInferShape(output_node, output_index); in CreateOutputTensor()
35 tensor->set_padding_type(AnfAlgo::GetOutputReshapeType(output_node, output_index)); in CreateOutputTensor()
38 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor()
85 auto &output_node = output_nodes_[i].first; in UpdateOutputDeviceAddress() local
87 if ((output_node != nullptr) && (!IsPersistentDeviceTensor(output_node))) { in UpdateOutputDeviceAddress()
88 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress()
[all …]
Doutput_actor.h62 void CollectOutput(const AnfNodePtr &output_node, size_t output_index, size_t output_position,
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindspore_importer.cc127 auto output_node = return_node->input(i); in GetFuncGraphOutputName() local
128 if (output_node == nullptr) { in GetFuncGraphOutputName()
131 } else if (output_node->isa<mindspore::CNode>()) { in GetFuncGraphOutputName()
132 if (opt::CheckPrimitiveType(output_node, prim::kPrimUpdateState) || in GetFuncGraphOutputName()
133 opt::CheckPrimitiveType(output_node, prim::kPrimLoad)) { in GetFuncGraphOutputName()
136 auto output_cnode = utils::cast<CNodePtr>(output_node); in GetFuncGraphOutputName()
137 if (opt::CheckPrimitiveType(output_node, prim::kPrimMakeTuple)) { in GetFuncGraphOutputName()
145 if (opt::CheckPrimitiveType(output_node, prim::kPrimUpdateState) || in GetFuncGraphOutputName()
146 opt::CheckPrimitiveType(output_node, prim::kPrimLoad)) { in GetFuncGraphOutputName()
154 } else if (opt::CheckPrimitiveType(output_node, prim::kPrimDepend)) { in GetFuncGraphOutputName()
Dmindir_control_flow_adjust.cc38 auto output_node = partial_fg->output(); in HasCallAfter() local
39 return IsCall(output_node); in HasCallAfter()
45 auto output_node = fg->output(); in GetFgOutput() local
46 if (output_node == nullptr) { in GetFgOutput()
51 auto output_cnode = output_node->cast<CNodePtr>(); in GetFgOutput()
54 return {output_node}; in GetFgOutput()
56 if (!IsMakeTuple(output_node)) { in GetFgOutput()
58 return {output_node}; in GetFgOutput()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc148 auto output_node = node_index_set.first; in GetMaxSubGraph() local
149 MS_EXCEPTION_IF_NULL(output_node); in GetMaxSubGraph()
150 if (output_node->isa<CNode>()) { in GetMaxSubGraph()
151 auto output_cnode = output_node->cast<CNodePtr>(); in GetMaxSubGraph()
177 auto output_node = node_index_set.first; in GetOriginRecomputeAndTargetNodes() local
178 MS_EXCEPTION_IF_NULL(output_node); in GetOriginRecomputeAndTargetNodes()
179 if (!IsBpropNode(output_node)) { in GetOriginRecomputeAndTargetNodes()
182 target_nodes->insert(output_node->cast<CNodePtr>()); in GetOriginRecomputeAndTargetNodes()
318 for (const auto &output_node : tuple_getitem_output_nodes) { in SetRecomputedAttr() local
319 auto output_cnode = output_node->cast<CNodePtr>(); in SetRecomputedAttr()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc180 auto output_node = func_graph->output()->cast<CNodePtr>(); in FilterIndexes() local
183 … [&output_node](size_t i) { return IsSideEffectNode(output_node->input(i + 1)); }), in FilterIndexes()
188 auto output_node = func_graph->output()->cast<CNodePtr>(); in FindAllOutputs() local
189 auto index_node = output_node->input(index); in FindAllOutputs()
206 for (size_t i = 1; i < output_node->size(); i++) { in FindAllOutputs()
207 auto out = output_node->input(i); in FindAllOutputs()
Dgraph_kernel_helper.cc849 auto output_node = func_graph->output(); in AnfGraph2LiteGraph() local
850 if (AnfAlgo::CheckPrimitiveType(output_node, prim::kPrimMakeTuple)) { in AnfGraph2LiteGraph()
852 auto mt = output_node->cast<CNodePtr>(); in AnfGraph2LiteGraph()
857 gb.SetOutputs({node_map[output_node]}); in AnfGraph2LiteGraph()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc197 TensorPtr CreateOutputTensor(const AnfNodePtr &output_node, size_t output_index) { in CreateOutputTensor() argument
198 MS_EXCEPTION_IF_NULL(output_node); in CreateOutputTensor()
201 auto type_id = AnfAlgo::GetOutputInferDataType(output_node, output_index); in CreateOutputTensor()
203 const auto &shape = AnfAlgo::GetOutputInferShape(output_node, output_index); in CreateOutputTensor()
206 tensor->set_padding_type(AnfAlgo::GetOutputReshapeType(output_node, output_index)); in CreateOutputTensor()
209 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor()
239 auto &output_node = item_with_index.first; in UpdateOutputDeviceAddress() local
241 if (output_node != nullptr) { in UpdateOutputDeviceAddress()
242 if (!AnfAlgo::OutputAddrExist(output_node, output_index, false)) { in UpdateOutputDeviceAddress()
245 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress()
[all …]
Dbackend.h131 …void ConstructOutputs(const AnfNodePtr &output_node, const std::vector<tensor::TensorPtr> &output_…
/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() argument
361 MS_EXCEPTION_IF_NULL(output_node); in StructureOutput()
363 if (output_node->isa<ValueNode>()) { in StructureOutput()
364 return ValueToPyData(GetValueNode(output_node)); in StructureOutput()
367 if (output_node->isa<Parameter>()) { in StructureOutput()
375 auto output_c = output_node->cast<CNodePtr>(); in StructureOutput()
378 << output_node->ToString(); in StructureOutput()
436 AnfNodePtr output_node = graph->get_return()->input(1); in DoExecGraph() local
437 MS_EXCEPTION_IF_NULL(output_node); in DoExecGraph()
439 py::object oj = StructureOutput(output_node, outputs, &count); in DoExecGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc62 auto output_node = return_node->input(kInputNodeIndex); in SetGraphOutput() local
63 if (output_node == nullptr) { in SetGraphOutput()
67 auto output_cnode = output_node->cast<CNodePtr>(); in SetGraphOutput()
74 if (CheckPrimitiveType(output_node, prim::kPrimMakeTuple)) { in SetGraphOutput()
88 …auto make_tuple_cnode = func_graph->NewCNode({make_tuple_vnode, output_node, tensor_array_write_no… in SetGraphOutput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Drectify_do_mask_kernel_info.cc50 for (const auto &output_node : *gen_mask_output_nodes) { in Process()
51 if (AnfAlgo::GetCNodeName(output_node.first) == prim::kPrimDropoutDoMask->name()) { in Process()
52 MS_EXCEPTION_IF_NULL(output_node.first); in Process()
53 auto output_cnode = output_node.first->cast<CNodePtr>(); in Process()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc275 auto output_node = output_pair.first; in InitSubGraphOutNode() local
276 if (output_node == nullptr || !utils::isa<CNodePtr>(output_node)) { in InitSubGraphOutNode()
280 if (opt::CheckPrimitiveType(output_node, prim::kPrimReturn)) { in InitSubGraphOutNode()
283 auto output_cnode = utils::cast<CNodePtr>(output_node); in InitSubGraphOutNode()
374 auto output_node = belong_anf_->output(); in FindAfterSubGraphInBelongAnf() local
375 if (!utils::isa<CNodePtr>(output_node)) { in FindAfterSubGraphInBelongAnf()
376 MS_LOG(ERROR) << "Output node of anf should be a cnode: " << output_node->fullname_with_scope(); in FindAfterSubGraphInBelongAnf()
379 q.push(utils::cast<CNodePtr>(output_node)); in FindAfterSubGraphInBelongAnf()
/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
109 MS_EXCEPTION_IF_NULL(output_node); in GetForwardOutNodeAndBpropGraph()
110 if (!output_node->isa<CNode>()) { in GetForwardOutNodeAndBpropGraph()
123 *forward_node = output_node->cast<CNodePtr>(); in GetForwardOutNodeAndBpropGraph()
Dkpynative.cc273 void UpdateOutputNodeOfTopCell(const AnfNodePtr &output_node) override;
441 void KPynativeCellImpl::UpdateOutputNodeOfTopCell(const AnfNodePtr &output_node) { in UpdateOutputNodeOfTopCell() argument
442 MS_EXCEPTION_IF_NULL(output_node); in UpdateOutputNodeOfTopCell()
443 MS_LOG(DEBUG) << "Real output node of top cell is " << output_node->DebugString(); in UpdateOutputNodeOfTopCell()
444 last_node_ = output_node; in UpdateOutputNodeOfTopCell()
448 if (IsPrimitiveCNode(output_node, prim::kPrimTupleGetItem) || in UpdateOutputNodeOfTopCell()
449 IsPrimitiveCNode(output_node, prim::kPrimListGetItem)) { in UpdateOutputNodeOfTopCell()
450 MS_LOG(DEBUG) << "Build cnode adjoint for anfnode: " << output_node->DebugString(); in UpdateOutputNodeOfTopCell()
451 auto cnode = output_node->cast<CNodePtr>(); in UpdateOutputNodeOfTopCell()
454 } else if (output_node->isa<ValueNode>()) { in UpdateOutputNodeOfTopCell()
[all …]
Dkpynative.h30 virtual void UpdateOutputNodeOfTopCell(const AnfNodePtr &output_node) = 0;
Ddfunctor.cc432 auto output_node = primal_graph_->output(); in MapMorphism() local
433 if (IsPrimitiveCNode(output_node, prim::kPrimHookBackward)) { in MapMorphism()
434 auto output_cnode = output_node->cast<CNodePtr>(); in MapMorphism()
437 output_node = output_cnode->input(1); in MapMorphism()
440 (void)MapMorphism(output_node); in MapMorphism()
443 auto output_adjoint = anfnode_to_adjoin_.find(output_node); in MapMorphism()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc467 for (auto output_node : tflite_subgraph->outputs) { in ConvertGraphOutputs() local
468 … auto output_idx = output_node < 0 ? output_node + tflite_subgraph->tensors.size() : output_node; in ConvertGraphOutputs()
536 auto output_node = output_nodes.at(i); in ConvertGraphOutputs() local
538 if (utils::isa<CNodePtr>(output_node)) { in ConvertGraphOutputs()
539 …output_node->cast<CNodePtr>()->set_fullname_with_scope(subgraph_name + "_output_" + std::to_string… in ConvertGraphOutputs()
541 } else if (utils::isa<ParameterPtr>(output_node)) { in ConvertGraphOutputs()
542 …output_node->cast<ParameterPtr>()->set_name(subgraph_name + "_output_" + std::to_string(i) + "_par… in ConvertGraphOutputs()
/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
214 MS_CHECK_TRUE_MSG(output_node != nullptr, nullptr, "output_node is nullptr."); in Process()
216 return output_node; in Process()
/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
68 output_node->set_abstract(node->abstract()); in operator()
69 new_fg->set_output(output_node); in operator()
119 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator() local
120 output_node->set_abstract(node->abstract()); in operator()
121 new_fg->set_output(output_node); in operator()
Dbranch_culling.cc265 …eSwitchDependNode(const FuncGraphPtr &graph, const AnfNodePtr &cond, const AnfNodePtr &output_node, in GenerateSwitchDependNode() argument
281 AnfNodePtrList inputs = {NewValueNode(prim::kPrimDepend), square_op, 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
427 if (nodes_.find(output_node) == nodes_.end()) { in ConvertGraphOutputs()
431 auto cnode = nodes_.find(output_node)->second; in ConvertGraphOutputs()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dcoco_op.h269 Status SearchNodeInJson(const nlohmann::json &input_tree, std::string node_name, T *output_node);
/third_party/node/test/fixtures/wpt/resources/
Dtestharness.js2597 this.output_node = null;
2632 if (this.output_node) {
2675 this.output_node = node;
2690 if (this.output_node) {
2694 this.output_node.textContent = "Running, " +
2708 if (!this.output_node) {
2713 var log = this.output_node;
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc517 auto output_node = output_node_with_index.first; in RunOpAssignOutputNodeMemory() local
518 MS_EXCEPTION_IF_NULL(output_node); in RunOpAssignOutputNodeMemory()
519 if (!output_node->isa<CNode>()) { in RunOpAssignOutputNodeMemory()
520 if (output_node->isa<Parameter>()) { in RunOpAssignOutputNodeMemory()
521 auto param = output_node->cast<ParameterPtr>(); in RunOpAssignOutputNodeMemory()
528 auto real_output_cnode = output_node->cast<CNodePtr>(); in RunOpAssignOutputNodeMemory()

12