Searched refs:graph_out (Results 1 – 16 of 16) sorted by relevance
| /third_party/mindspore/tests/st/control/inner/ |
| D | test_break_continue.py | 69 graph_out = forward_net(x, y)
70 assert graph_out == Tensor(np.array(3), mstype.int32)
329 graph_out = forward_net(x, y)
331 assert graph_out == Tensor(np.array(54), mstype.int32)
375 graph_out = forward_net(x)
377 assert graph_out == Tensor(np.array(1), mstype.int32)
405 graph_out = forward_net(x, y, z)
406 print("test_for_in_for_break graph out:", graph_out)
|
| D | test_012_if_in_for_break.py | 62 graph_out = forward_net(x, y)
64 assert graph_out == Tensor(np.array(9), mstype.int32)
|
| D | test_022_for_while_normal.py | 65 graph_out = forward_net(x, y)
67 assert graph_out == Tensor(np.array(27), mstype.int32)
|
| D | test_021_while_while_normal.py | 67 graph_out = forward_net(x, y)
69 assert graph_out == Tensor(np.array(27), mstype.int32)
|
| D | test_202_while_n_while.py | 68 graph_out = forward_net(x, y)
70 assert graph_out == Tensor(np.array(18), mstype.int32)
|
| D | test_301_while_normal_for.py | 66 graph_out = forward_net(x, y)
68 assert graph_out == Tensor(np.array(18), mstype.int32)
|
| D | test_222_for_while_while.py | 69 graph_out = forward_net(x, y)
71 assert graph_out == Tensor(np.array(36), mstype.int32)
|
| D | test_321_while_while_in_while.py | 71 graph_out = forward_net(x, y)
73 assert graph_out == Tensor(np.array(36), mstype.int32)
|
| D | test_221_while_while_while.py | 71 graph_out = forward_net(x, y)
73 assert graph_out == Tensor(np.array(36), mstype.int32)
|
| D | test_201_for_n_while.py | 66 graph_out = forward_net(x, y)
68 assert graph_out == Tensor(np.array(18), mstype.int32)
|
| D | test_311_while_if_for.py | 70 graph_out = forward_net(x, y)
72 assert graph_out == Tensor(np.array(13), mstype.int32)
|
| D | test_232_for_for_while.py | 68 graph_out = forward_net(x, y)
70 assert graph_out == Tensor(np.array(36), mstype.int32)
|
| D | test_011_if_in_while_break.py | 123 graph_out = forward_net(x, y)
125 assert graph_out == Tensor(np.array(21), mstype.int32)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 742 auto graph_out = graph->output(); in IsParameterGraph() local
743 MS_EXCEPTION_IF_NULL(graph_out); in IsParameterGraph()
744 auto actual_op = ActualOp(graph_out); in IsParameterGraph()
775 auto graph_out = ActualOp(graph->output()); in HandleParameterGraph() local
776 MS_EXCEPTION_IF_NULL(graph_out); in HandleParameterGraph()
778 auto param_iter = std::find(parameter_list.begin(), parameter_list.end(), graph_out); in HandleParameterGraph()
789 argument = graph_out; in HandleParameterGraph()
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 130 graph_out = self.forward_mindspore_impl()
135 allclose_nparray(graph_out[0], pynative_out[0], self.loss, self.loss)
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 333 for (const auto &graph_out : onnx_graph.output()) { in ConvertGraphOutputs() local
334 if (anf_nodes_map.find(graph_out.name()) == anf_nodes_map.end()) { in ConvertGraphOutputs()
338 auto cnode = anf_nodes_map.at(graph_out.name()); in ConvertGraphOutputs()
353 const auto &graph_out = onnx_graph.output(0); in ConvertGraphOutputs() local
354 if (anf_nodes_map.find(graph_out.name()) == anf_nodes_map.end()) { in ConvertGraphOutputs()
358 auto cnode = anf_nodes_map.at(graph_out.name()); in ConvertGraphOutputs()
|