Home
last modified time | relevance | path

Searched refs:input_nodes (Results 1 – 25 of 42) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc33 auto input_nodes = kernel_graph->inputs(); in LoadInputData() local
36 for (size_t i = 0; i < input_nodes.size(); ++i) { in LoadInputData()
38 if (!input_nodes[i]->isa<Parameter>() || !AnfAlgo::OutputAddrExist(input_nodes[i], 0)) { in LoadInputData()
42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData()
62 auto input_nodes = kernel_graph->inputs(); in CompileGraphImpl() local
63 for (size_t i = 0; i < input_nodes.size(); ++i) { in CompileGraphImpl()
64 if (!input_nodes[i]->isa<Parameter>() || !AnfAlgo::OutputAddrExist(input_nodes[i], 0)) { in CompileGraphImpl()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl()
Dgpu_inference_session.cc33 auto input_nodes = kernel_graph->inputs(); in LoadInputData() local
36 for (size_t i = 0; i < input_nodes.size(); ++i) { in LoadInputData()
38 if (!input_nodes[i]->isa<Parameter>() || !AnfAlgo::OutputAddrExist(input_nodes[i], 0)) { in LoadInputData()
42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData()
62 auto input_nodes = kernel_graph->inputs(); in CompileGraphImpl() local
63 for (size_t i = 0; i < input_nodes.size(); ++i) { in CompileGraphImpl()
64 if (!input_nodes[i]->isa<Parameter>() || !AnfAlgo::OutputAddrExist(input_nodes[i], 0)) { in CompileGraphImpl()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl()
Dcpu_session.cc169 auto &input_nodes = kernel_graph->inputs(); in LoadInputData() local
170 if (input_nodes.size() != inputs_const.size()) { in LoadInputData()
172 << input_nodes.size(); in LoadInputData()
174 for (size_t input_idx = 0; input_idx < input_nodes.size(); ++input_idx) { in LoadInputData()
175 auto &input_node = input_nodes[input_idx]; in LoadInputData()
Dgpu_session.cc323 auto &input_nodes = kernel_graph->input_nodes(); in LoadInputData() local
326 if (inputs.size() != input_nodes.size()) { in LoadInputData()
327 …EPTION) << "Tensor input:" << inputs.size() << " is not equal graph inputs:" << input_nodes.size(); in LoadInputData()
332 auto input_node = input_nodes[i]; in LoadInputData()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc137 …ratorInfo::CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateConcateNode() argument
140 if (input_nodes.size() != input_nodes_num) { in CreateConcateNode()
150 (void)std::transform(input_nodes.begin(), input_nodes.end(), std::back_inserter(concat_inputs), in CreateConcateNode()
164 …eratorInfo::CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateReduceNode() argument
167 if (input_nodes.size() != input_nodes_num) { in CreateReduceNode()
177 (void)std::transform(input_nodes.begin(), input_nodes.end(), std::back_inserter(addn_inputs), in CreateReduceNode()
Doperator_info.h67 …AnfNodePtr CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes,
69 AnfNodePtr CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes,
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc207 const auto &input_nodes = graph->input_nodes(); in PrepareDataForDeviceTensorStore() local
209 for (size_t j = 0; j < input_nodes.size(); ++j) { in PrepareDataForDeviceTensorStore()
210 const auto &input_node = input_nodes[j]; in PrepareDataForDeviceTensorStore()
238 const auto &input_nodes = graph->input_nodes(); in PrepareDataForHostTensorQueue() local
240 for (size_t j = 0; j < input_nodes.size(); ++j) { in PrepareDataForHostTensorQueue()
241 const auto &input_node = input_nodes[j]; in PrepareDataForHostTensorQueue()
282 const auto &input_nodes = graph->input_nodes(); in PrepareDataForStepMode() local
284 for (size_t j = 0; j < input_nodes.size(); ++j) { in PrepareDataForStepMode()
285 const auto &input_node = input_nodes[j]; in PrepareDataForStepMode()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc289 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
298 …if (auto in = cnode->input(idx); std::find(input_nodes.begin(), input_nodes.end(), in) == input_no… in TEST_F()
299 input_nodes.push_back(in); in TEST_F()
305 FusionScopeInfo fusion_scope_info(0, full_name, input_nodes, compute_nodes, {}); in TEST_F()
348 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
357 …if (auto in = cnode->input(idx); std::find(input_nodes.begin(), input_nodes.end(), in) == input_no… in TEST_F()
358 input_nodes.push_back(in); in TEST_F()
364 FusionScopeInfo fusion_scope_info(0, full_name, input_nodes, compute_nodes, {}); in TEST_F()
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.cc38 std::vector<kernel::LiteKernel *> input_nodes; in SubgraphInputNodes() local
54 if (kernel_is_input && !lite::IsContain(input_nodes, kernel)) { in SubgraphInputNodes()
55 input_nodes.push_back(kernel); in SubgraphInputNodes()
58 return input_nodes; in SubgraphInputNodes()
88 std::vector<kernel::LiteKernel *> input_nodes = SubgraphInputNodes(kernels); in SubgraphInputTensors() local
89 for (const auto &input_node : input_nodes) { in SubgraphInputTensors()
Dsub_graph_split.cc448 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in OptimizeAfterFusion() local
449 if (input_nodes.size() != 1) continue; in OptimizeAfterFusion()
450 uint32_t input_node_index = input_nodes.at(0); in OptimizeAfterFusion()
537 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in InsertNodeByMid() local
538 if (input_nodes.size() != 1) continue; in InsertNodeByMid()
540 uint32_t input_node = input_nodes.at(0); in InsertNodeByMid()
592 std::vector<uint32_t> input_nodes = tensor->out_nodes_; in InitMiddleSubgraph() local
593 if (input_nodes.empty()) continue; in InitMiddleSubgraph()
594 if (input_nodes.size() != 1) continue; in InitMiddleSubgraph()
595 uint32_t input_node = input_nodes[0]; in InitMiddleSubgraph()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dfusion_tbe_json_creator.cc96 auto input_nodes = fusion_scope_info.input_nodes; in CheckInput() local
108 auto find_iter = std::find(input_nodes.begin(), input_nodes.end(), input); in CheckInput()
109 if (find_iter != input_nodes.end()) { in CheckInput()
114 if (input_nodes.size() != input_size) { in CheckInput()
116 << " ] is not equal to input nodes num: [ " << input_nodes.size() << " ]."; in CheckInput()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_fusion.h34 input_nodes(std::move(in)), in FusionScopeInfo()
39 std::vector<AnfNodePtr> input_nodes; member
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc285 auto &input_nodes = kernel_graph->inputs(); in CreateOutputTensors() local
286 if (input_nodes.size() != inputs.size()) { in CreateOutputTensors()
287 …ON) << "Input size " << inputs.size() << " is not equal to input node size " << input_nodes.size(); in CreateOutputTensors()
291 for (auto &item : input_nodes) { in CreateOutputTensors()
309 auto &input_nodes = kernel_graph.inputs(); in BindInputTensorAddressPtr() local
310 if (input_nodes.size() != inputs.size()) { in BindInputTensorAddressPtr()
311 …ION) << "Input size" << inputs.size() << " is not equal to input node size " << input_nodes.size(); in BindInputTensorAddressPtr()
313 for (size_t input_idx = 0; input_idx < input_nodes.size(); ++input_idx) { in BindInputTensorAddressPtr()
314 auto &item = input_nodes[input_idx]; in BindInputTensorAddressPtr()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc224 std::vector<AnfNodePtr> input_nodes; in ConvertLayers() local
225 status = ConvertBottom(layer, &input_nodes); in ConvertLayers()
243 op_inputs.insert(op_inputs.end(), input_nodes.begin(), input_nodes.end()); in ConvertLayers()
551 …elParser::ConvertBottom(const caffe::LayerParameter &layer, std::vector<AnfNodePtr> *input_nodes) { in ConvertBottom() argument
552 MSLITE_CHECK_PTR(input_nodes); in ConvertBottom()
564 input_nodes->emplace_back(nodes_.find(origin_layer)->second); in ConvertBottom()
Dcaffe_model_parser.h53 STATUS ConvertBottom(const caffe::LayerParameter &layer, std::vector<AnfNodePtr> *input_nodes);
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc251 std::vector<AnfNodePtr> input_nodes; in FetchInputNodeByParameter() local
255 return input_nodes; in FetchInputNodeByParameter()
263 (void)input_nodes.emplace_back(parameter); in FetchInputNodeByParameter()
264 return input_nodes; in FetchInputNodeByParameter()
270 return input_nodes; in FetchInputNodeByParameter()
295 (void)input_nodes.emplace_back(input); in FetchInputNodeByParameter()
299 (void)input_nodes.insert(input_nodes.end(), inputs.begin(), inputs.end()); in FetchInputNodeByParameter()
302 return input_nodes; in FetchInputNodeByParameter()
939 for (const auto &parameter : graph->input_nodes()) { in FetchFrontValueNode()
1010 const auto &inputs = graph->input_nodes(); in FetchControlNodeParameter()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dunsorted_segment_op_info.cc288 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_min, 1), in ComputeReplaceGraph() local
291 std::make_pair(input_nodes, final_output)); in ComputeReplaceGraph()
322 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_max, 1), in ComputeReplaceGraph() local
325 std::make_pair(input_nodes, final_output)); in ComputeReplaceGraph()
Dbroadcast_to_info.cc166 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_broadcast_to, 1)}; in ComputeReplaceGraph() local
168 std::make_pair(input_nodes, new_broadcast_to)); in ComputeReplaceGraph()
Dunique_info.cc151 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(sub, 1), std::make_pair(… in ComputeReplaceGraph() local
153 std::make_pair(input_nodes, make_tuple)); in ComputeReplaceGraph()
Dslice_info.cc234 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(slice, 1)}; in ComputeReplaceGraph() local
236 std::make_pair(input_nodes, slice)); in ComputeReplaceGraph()
Duniform_candidate_sampler_info.cc233 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_sampler_op, 1)}; in ComputeReplaceGraph() local
235 std::make_pair(input_nodes, new_sampler_op)); in ComputeReplaceGraph()
Donehot_info.cc195 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(floor_div, 1), std::make… in ComputeReplaceGraph() local
197 std::make_pair(input_nodes, onehot)); in ComputeReplaceGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc388 auto input_nodes = fusion_scope_info.input_nodes; in GetSpecDataInput() local
403 auto find_iter = std::find(input_nodes.begin(), input_nodes.end(), input); in GetSpecDataInput()
404 if (find_iter != input_nodes.end()) { in GetSpecDataInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc665 CNodePtr CreateCNodeWithGraph(const std::vector<AnfNodePtr> &input_nodes, const BaseRef &graph) { in CreateCNodeWithGraph() argument
667 return std::make_shared<CNode>(input_nodes, utils::cast<FuncGraphPtr>(graph)); in CreateCNodeWithGraph()
670 return std::make_shared<CNode>(input_nodes, utils::cast<VarPtr>(graph)); in CreateCNodeWithGraph()
691 std::vector<AnfNodePtr> input_nodes; in HandleSexpVector() local
696 input_nodes.push_back(node); in HandleSexpVector()
699 return std::make_shared<CNode>(input_nodes, var_ptr); in HandleSexpVector()
704 input_nodes.push_back(node); in HandleSexpVector()
706 return CreateCNodeWithGraph(input_nodes, graph); in HandleSexpVector()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.cc421 std::vector<AnfNodePtr> input_nodes; in Jump() local
422 input_nodes.emplace_back(NewValueNode(target_block->func_graph())); in Jump()
423 (void)std::copy(args.begin(), args.end(), std::back_inserter(input_nodes)); in Jump()
425 CNodePtr jump = func_graph_->NewCNodeInOrder(input_nodes); in Jump()

12