Home
last modified time | relevance | path

Searched defs:node_inputs (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbce_with_logits_loss_fusion.cc34 std::vector<AnfNodePtr> node_inputs = { in AddReduceNode() local
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/parser/
Dmodel_parser_test.cc93 auto node_inputs = model_layers_info_[node_name]; in BuildGraphNodes() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc65 std::vector<AnfNodePtr> node_inputs{node->inputs()}; in ExtractInputParameterByNode() local
155 std::vector<AnfNodePtr> node_inputs{node->inputs()}; in ExtractInputTypeLengthByNode() local
/third_party/mindspore/mindspore/lite/tools/converter/
Dexport_model.cc154 std::vector<AnfNodePtr> node_inputs; in CloneFuncGraph() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc146 auto node_inputs = cnode->inputs(); in GetNextNodes() local
244 auto node_inputs = cnode->inputs(); in GetNodesDependencyInfo() local
426 auto node_inputs = cnode->inputs(); in AddSegmentDependency() local
Dtransform.cc180 std::vector<AnfNodePtr> node_inputs = node->inputs(); in InterpretNode() local
Dbackend.cc615 const std::vector<AnfNodePtr> &node_inputs = cnode->inputs(); in RunControlOperator() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc93 …AnfNodePtrList node_inputs = {cnode->input(kAnfPrimitiveIndex), cnode->input(kUpdateStateStateInpu… in Run() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dinline.h278 std::vector<AnfNodePtr> node_inputs; in SimplifyAfterParameter() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc1647 auto node_inputs = node->cast<CNodePtr>()->inputs(); in GetRealOpNode() local
1697 auto node_inputs = node->cast<CNodePtr>()->inputs(); in ConvertDependNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc511 auto node_inputs = node->inputs(); in RemoveOriginParamFromSet() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2621 std::vector<AnfNodePtr> node_inputs{node->inputs()}; in NodeParameterName() local
3191 std::vector<AnfNodePtr> node_inputs{all_inputs.begin() + 1, all_inputs.end()}; in ExtractInputsTensorName() local