| /third_party/mindspore/tests/st/control/ |
| D | test_cont_grad.py | 61 graph_output = net(idx, end, x)
63 assert graph_output == 0
96 graph_output = net(idx, end)
99 assert np.allclose(graph_output[0].asnumpy(), expect_one, 0.0001, 0.0001)
100 assert np.allclose(graph_output[1].asnumpy(), expect_two, 0.0001, 0.0001)
133 graph_output = net(idx, end)
136 assert np.allclose(graph_output[0].asnumpy(), expect_one, 0.0001, 0.0001)
137 assert np.allclose(graph_output[1].asnumpy(), expect_two, 0.0001, 0.0001)
168 graph_output = net(idx, end, x)
170 assert np.allclose(graph_output.asnumpy(), expect, 0.0001, 0.0001)
[all …]
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_graph.cc | 264 for (auto graph_output : this->outputs()) { in CreateNPUSubgraphKernel()
267 …if (find(subgraph_op_outputs.begin(), subgraph_op_outputs.end(), graph_output) != subgraph_op_outp… in CreateNPUSubgraphKernel()
268 … find(subgraph_outputs.begin(), subgraph_outputs.end(), graph_output) == subgraph_outputs.end()) { in CreateNPUSubgraphKernel()
269 subgraph_outputs.emplace_back(graph_output); in CreateNPUSubgraphKernel()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_cond.cc | 173 auto graph_output = graph->output(); in CreateBranchGraph() local
174 MS_CHECK_TRUE_RET(graph_output != nullptr, nullptr); in CreateBranchGraph()
175 auto graph_output_cnode = graph_output->cast<CNodePtr>(); in CreateBranchGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 344 for (const auto &graph_output : graph_outputs) { in FetchOutputByCallNode() local
345 if (graph_output->isa<Parameter>()) { in FetchOutputByCallNode()
346 outputs.push_back(graph_output); in FetchOutputByCallNode()
347 } else if (AnfAlgo::CheckPrimitiveType(graph_output, prim::kPrimSwitch)) { in FetchOutputByCallNode()
348 … const auto &switch_outputs = FetchOutputBySwitchNode(graph_output, call_nodes, switch_nodes); in FetchOutputByCallNode()
350 } else if (IsCallNode(graph_output)) { in FetchOutputByCallNode()
351 const auto &call_outputs = FetchOutputByCallNode(graph_output, call_nodes, switch_nodes); in FetchOutputByCallNode()
353 } else if (graph_output->isa<CNode>()) { in FetchOutputByCallNode()
354 (void)outputs.emplace_back(graph_output); in FetchOutputByCallNode()
355 } else if (graph_output->isa<ValueNode>()) { in FetchOutputByCallNode()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 669 bool IsGraphOutputValueNodeOrParameter(const AnfNodePtr &graph_output, const VectorRef &args, Vecto… in IsGraphOutputValueNodeOrParameter() argument
670 MS_EXCEPTION_IF_NULL(graph_output); in IsGraphOutputValueNodeOrParameter()
672 if (graph_output->isa<ValueNode>()) { in IsGraphOutputValueNodeOrParameter()
675 ValuePtr value = GetValueNode(graph_output); in IsGraphOutputValueNodeOrParameter()
687 if (graph_output->isa<Parameter>()) { in IsGraphOutputValueNodeOrParameter()
690 auto func_graph = graph_output->func_graph(); in IsGraphOutputValueNodeOrParameter()
697 auto it = std::find(params.begin(), params.end(), graph_output); in IsGraphOutputValueNodeOrParameter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 155 auto graph_output = output(); in outputs() local
156 if (IsPrimitiveCNode(graph_output, prim::kPrimMakeTuple)) { in outputs()
162 return std::vector<AnfNodePtr>(1, graph_output); in outputs()
|
| D | session_basic.cc | 488 auto graph_output = return_node->input(kReturnInputIdx); in SetReturnNode() local
489 MS_EXCEPTION_IF_NULL(graph_output); in SetReturnNode()
494 if (AnfAlgo::IsTupleOutput(graph_output) && graph_output->isa<ValueNode>()) { in SetReturnNode()
495 return_node->set_input(kReturnInputIdx, graph->TransTupleToMakeTuple(graph_output)); in SetReturnNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 494 auto graph_output = graph->output(); in GetShapeType() local
495 type = graph_output->Type(); in GetShapeType()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 467 std::vector<Operator> graph_output; in BuildSaveCheckpointGraph() local
468 graph_output.emplace_back(save_op); in BuildSaveCheckpointGraph()
471 (void)checkpoint_graph->SetOutputs(graph_output); in BuildSaveCheckpointGraph()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 662 [](auto &graph_output) { return graph_output.name(); }); in ConvertOnnxGraph() argument
|