Home
last modified time | relevance | path

Searched refs:origin_inputs (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/tests/st/cpp/model/
Dtest_dynamic_batch_size.cc57 std::vector<MSTensor> origin_inputs = tensor_add.GetInputs(); in TEST_F() local
58 ASSERT_EQ(origin_inputs.size() - 1, 2); in TEST_F()
65 …inputs.emplace_back(origin_inputs[0].Name(), origin_inputs[0].DataType(), origin_inputs[0].Shape()… in TEST_F()
67 …inputs.emplace_back(origin_inputs[1].Name(), origin_inputs[1].DataType(), origin_inputs[1].Shape()… in TEST_F()
69 …inputs.emplace_back(origin_inputs[2].Name(), origin_inputs[2].DataType(), origin_inputs[2].Shape()… in TEST_F()
Dtest_tensor_add.cc43 std::vector<MSTensor> origin_inputs = tensor_add.GetInputs(); in TEST_F() local
44 ASSERT_EQ(origin_inputs.size(), 2); in TEST_F()
49 …inputs.emplace_back(origin_inputs[0].Name(), origin_inputs[0].DataType(), origin_inputs[0].Shape(), in TEST_F()
51 …inputs.emplace_back(origin_inputs[1].Name(), origin_inputs[1].DataType(), origin_inputs[1].Shape(), in TEST_F()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model.cc102 auto origin_inputs = GetInputs(); in Resize() local
103 if (inputs.size() != origin_inputs.size()) { in Resize()
104 …"Invalid inputs size " << inputs.size() << " not match model inputs size " << origin_inputs.size(); in Resize()
120 if (inputs[i].Name() != origin_inputs[i].Name()) { in Resize()
122 << origin_inputs[i].Name(); in Resize()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc212 auto origin_inputs = cnode->inputs(); in GetIntVecInput() local
213 std::vector<AnfNodePtr> specify_inputs = {origin_inputs[0], origin_inputs[index]}; in GetIntVecInput()
217 cnode->set_inputs(origin_inputs); in GetIntVecInput()
220 cnode->set_inputs(origin_inputs); in GetIntVecInput()
244 auto origin_inputs = cnode->inputs(); in GetCNodeInputTensors() local
252 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
259 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
282 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
Dinfershape_pass.cc27 auto origin_inputs = cnode->inputs(); in GetCNodeCertainInputFormat() local
33 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
37 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
45 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
336 auto origin_inputs = return_node->inputs(); in SetSubGraphAbstract() local
366 return_node->set_inputs(origin_inputs); in SetSubGraphAbstract()
Ddecrease_transpose_algo.cc59 auto origin_inputs = cur_node->inputs(); in FindAreaSurroundedByTranspose() local
77 cur_node->set_inputs(origin_inputs); in FindAreaSurroundedByTranspose()
316 auto origin_inputs = cnode->inputs(); in InsertPreTransNode() local
320 cnode->set_inputs(origin_inputs); in InsertPreTransNode()
323 cnode->set_inputs(origin_inputs); in InsertPreTransNode()
584 auto origin_inputs = return_node->inputs(); in SetSubGraphAbstract() local
614 return_node->set_inputs(origin_inputs); in SetSubGraphAbstract()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/ms/
Dms_model.cc119 auto origin_inputs = GetInputs(); in Resize() local
120 if (inputs.size() != origin_inputs.size()) { in Resize()
121 …"Invalid inputs size " << inputs.size() << " not match model inputs size " << origin_inputs.size(); in Resize()
/third_party/mindspore/mindspore/lite/tools/converter/
Dexport_model.cc42 auto origin_inputs = origin->get_inputs(); in CloneGraphInputs() local
43 for (auto &input : origin_inputs) { in CloneGraphInputs()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkprim.cc190 auto &origin_inputs = origin_cnode->inputs(); in GetPythonOps() local
192 (void)std::copy(origin_inputs.begin() + 1, origin_inputs.end(), std::back_inserter(new_inputs)); in GetPythonOps()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc690 auto origin_inputs = cnode->inputs(); in GetNewCNodeInputs() local
693 for (size_t input_idx = 1; input_idx < origin_inputs.size(); input_idx++) { in GetNewCNodeInputs()
694 auto anf = origin_inputs[input_idx]; in GetNewCNodeInputs()