| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 65 std::vector<AnfNodePtr> node_inputs{node->inputs()}; in ExtractInputParameterByNode() local
67 if ((node_inputs.size() == 2) && in ExtractInputParameterByNode()
68 (IsValueNode<ValueList>(node_inputs[1]) || IsValueNode<ValueTuple>(node_inputs[1]))) { in ExtractInputParameterByNode()
70 if (IsValueNode<ValueList>(node_inputs[1])) { in ExtractInputParameterByNode()
71 inputs_seq = node_inputs[1]->cast<ValueNodePtr>()->value()->cast<ValueListPtr>()->value(); in ExtractInputParameterByNode()
73 inputs_seq = node_inputs[1]->cast<ValueNodePtr>()->value()->cast<ValueTuplePtr>()->value(); in ExtractInputParameterByNode()
77 if ((node_inputs.size() == 2) && in ExtractInputParameterByNode()
78 …(AnfNodeIsPrimitive(node_inputs[1], MAKE_TUPLE) || AnfNodeIsPrimitive(node_inputs[1], MAKE_LIST)))… in ExtractInputParameterByNode()
79 node_inputs = node_inputs[1]->cast<CNodePtr>()->inputs(); in ExtractInputParameterByNode()
81 for (size_t i = 1; i < node_inputs.size(); ++i) { in ExtractInputParameterByNode()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | bce_with_logits_loss_fusion.cc | 34 std::vector<AnfNodePtr> node_inputs = { in AddReduceNode() local
36 (void)node_inputs.insert(node_inputs.end(), cnode->inputs().begin() + 1, cnode->inputs().end()); in AddReduceNode()
37 CNodePtr new_cnode = func_graph->NewCNode(node_inputs); in AddReduceNode()
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/parser/ |
| D | model_parser_test.cc | 93 auto node_inputs = model_layers_info_[node_name]; in BuildGraphNodes() local
94 if (node_inputs.empty()) { in BuildGraphNodes()
110 for (auto &input : node_inputs) { in BuildGraphNodes()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | graph_partition.cc | 146 auto node_inputs = cnode->inputs(); in GetNextNodes() local
148 std::reverse(node_inputs.begin(), node_inputs.end()); in GetNextNodes()
149 return node_inputs; in GetNextNodes()
152 for (auto &input : node_inputs) { in GetNextNodes()
244 auto node_inputs = cnode->inputs(); in GetNodesDependencyInfo() local
246 for (auto &input : node_inputs) { in GetNodesDependencyInfo()
426 auto node_inputs = cnode->inputs(); in AddSegmentDependency() local
432 for (auto &input : node_inputs) { in AddSegmentDependency()
|
| D | transform.cc | 180 std::vector<AnfNodePtr> node_inputs = node->inputs(); in InterpretNode() local
181 if (node_inputs.empty()) { in InterpretNode()
184 AnfNodePtr fn = node_inputs[0]; in InterpretNode()
188 for (size_t i = node_inputs.size() - 1; i > 0; i--) { in InterpretNode()
|
| D | backend.cc | 615 const std::vector<AnfNodePtr> &node_inputs = cnode->inputs(); in RunControlOperator() local
616 if (node_inputs.empty()) { in RunControlOperator()
620 const AnfNodePtr &fn = node_inputs.at(0); in RunControlOperator()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | update_state_formatter.cc | 93 …AnfNodePtrList node_inputs = {cnode->input(kAnfPrimitiveIndex), cnode->input(kUpdateStateStateInpu… in Run() local
94 node_inputs.insert(node_inputs.end(), inputs.begin(), inputs.end()); in Run()
96 auto new_node = func_graph->NewCNode(node_inputs); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 189 ::mindspore::Tensor *node_inputs = proto->add_inputs(); in SetNodeInputs()
190 MS_EXCEPTION_IF_NULL(node_inputs); in SetNodeInputs()
208 mindspore::TensorShape *tensorShape = node_inputs->mutable_tensor_shape(); in SetNodeInputs()
214 node_inputs->set_tensor_type(input_data_type); in SetNodeInputs()
215 node_inputs->set_mem_device("HBM"); in SetNodeInputs()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | inline.h | 278 std::vector<AnfNodePtr> node_inputs; in SimplifyAfterParameter() local
279 node_inputs.push_back(NewValueNode(new_fg)); in SimplifyAfterParameter()
280 … std::transform(used_param_index.begin(), used_param_index.end(), std::back_inserter(node_inputs), in SimplifyAfterParameter()
282 return node->func_graph()->NewCNode(node_inputs); in SimplifyAfterParameter()
|
| /third_party/mindspore/mindspore/lite/tools/converter/ |
| D | export_model.cc | 154 std::vector<AnfNodePtr> node_inputs; in CloneFuncGraph() local
179 node_inputs.push_back(mirror_input); in CloneFuncGraph()
181 auto mirror_cnode = mirror_graph->NewCNode(mirrro_prim, node_inputs); in CloneFuncGraph()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 1647 auto node_inputs = node->cast<CNodePtr>()->inputs(); in GetRealOpNode() local
1648 if (node_inputs.size() != 3) { // "tuple_getitem" primitive must have 3 inputs in GetRealOpNode()
1653 MS_EXCEPTION_IF_NULL(node_inputs[TUPLE_GET_ITEM_INDEX]); in GetRealOpNode()
1654 if (!node_inputs[TUPLE_GET_ITEM_INDEX]->isa<ValueNode>()) { in GetRealOpNode()
1658 auto value_ptr = GetValueNode(node_inputs[TUPLE_GET_ITEM_INDEX])->cast<Int32ImmPtr>(); in GetRealOpNode()
1667 if (IsPrimitiveCNode(node_inputs[1], prim::kPrimMakeTuple)) { in GetRealOpNode()
1677 return GetRealOpNode(node_inputs[1]); in GetRealOpNode()
1697 auto node_inputs = node->cast<CNodePtr>()->inputs(); in ConvertDependNode() local
1698 for (size_t index = 1; index < node_inputs.size(); index++) { in ConvertDependNode()
1699 auto op = Convert(GetRealOpNode(node_inputs[index])); in ConvertDependNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 511 auto node_inputs = node->inputs(); in RemoveOriginParamFromSet() local
512 for (auto &input : node_inputs) { in RemoveOriginParamFromSet()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 2621 std::vector<AnfNodePtr> node_inputs{node->inputs()}; in NodeParameterName() local
2623 for (int64_t i = 0; i < UlongToLong(node_inputs.size()); ++i) { in NodeParameterName()
2625 auto input = node_inputs[LongToSize(i)]; in NodeParameterName()
3191 std::vector<AnfNodePtr> node_inputs{all_inputs.begin() + 1, all_inputs.end()}; in ExtractInputsTensorName() local
3195 for (auto &input : node_inputs) { in ExtractInputsTensorName()
|