Home
last modified time | relevance | path

Searched refs:kAnfPrimitiveIndex (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Daffine_activation_fusion.cc53 …_prim = GetValueNode<std::shared_ptr<ops::Activation>>(activation_node->input(kAnfPrimitiveIndex)); in Process()
70 …o affine_prim = GetValueNode<std::shared_ptr<ops::Affine>>(affine_node->input(kAnfPrimitiveIndex)); in Process()
Daffine_fusion.cc61 …o matmul_prim = GetValueNode<std::shared_ptr<ops::MatMul>>(matmul_node->input(kAnfPrimitiveIndex)); in Process()
79 …o splice_prim = GetValueNode<std::shared_ptr<ops::Splice>>(splice_node->input(kAnfPrimitiveIndex)); in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dparallel_pass.cc28 constexpr auto kAnfPrimitiveIndex = 0; variable
34 auto prim = GetValueNode<PrimitivePtr>(c_node->input(kAnfPrimitiveIndex)); in IsParallelCareNode()
Dconv2d_info.cc94 …to conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex)); in CheckIfSplit()
157 …to conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex)); in CreateOutputsOfSplit()
215 auto prim = GetValueNode<PrimitivePtr>(cnode_->input(kAnfPrimitiveIndex)); in CheckConv2DPrimitiveType()
262 …to conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex)); in InferParallelCNodes()
Dmulti_conv_info.cc68 …onv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(conv_cnode->input(kAnfPrimitiveIndex)); in CheckSplitValid()
181 …attr = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(ori_conv_cnode->input(kAnfPrimitiveIndex)); in SplitSingleConv()
Dspliter.cc73 auto prim = GetValueNode<PrimitivePtr>(cnode->input(kAnfPrimitiveIndex)); in RecordGraphInfo()
Ddepthwise_conv2d_info.cc335 …se_conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex)); in CreateOutputsOfSplit()
397 auto prim = GetValueNode<PrimitivePtr>(cnode_->input(kAnfPrimitiveIndex)); in CheckDepthWiseConv2DPrimitiveType()
492 …se_conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex)); in InferParallelCNodes()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Deliminate_concat_split.cc75 auto pre_prim = GetValueNode<std::shared_ptr<ops::Concat>>(pre_cnode->input(kAnfPrimitiveIndex)); in ConcatSplitEliminate()
77 …auto prim = GetValueNode<std::shared_ptr<ops::SplitWithOverlap>>(cnode->input(kAnfPrimitiveIndex)); in ConcatSplitEliminate()
Dfisson_util.cc262 …onv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(conv_cnode->input(kAnfPrimitiveIndex)); in UpdateSplitInfo()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dspatial_node_adapter.cc37 constexpr auto kAnfPrimitiveIndex = 0; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_label_assign.cc64 node->set_inputs({node->input(kAnfPrimitiveIndex), node->input(kFirstDataInputIndex)}); in UpdateLabelSwitch()
Dkernel_select_ascend.cc500 auto func_graph = GetValueNode<FuncGraphPtr>(kernel_node->input(kAnfPrimitiveIndex)); in SelectKernelInfo()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc38 constexpr auto kAnfPrimitiveIndex = 0; variable
402 return IsPrimitive(cnode->input(kAnfPrimitiveIndex), primitive_type); in CheckPrimitiveType()
512 auto input = cnode->input(kAnfPrimitiveIndex); in IsGraphKernel()
686 prim = GetValueNode<PrimitivePtr>(anf_node->cast<CNodePtr>()->input(kAnfPrimitiveIndex)); in IsParallelSplitConvNode()
709 prim = GetValueNode<PrimitivePtr>(anf_node->cast<CNodePtr>()->input(kAnfPrimitiveIndex)); in IsConvNode()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc781 auto attr_input = cnode->input(kAnfPrimitiveIndex); in CreateCallSwitchInputs()
790 std::vector<AnfNodePtr> switch_inputs = {switch_cnode->input(kAnfPrimitiveIndex), in CreateCallSwitchInputs()
878 auto attr_input = cnode->input(kAnfPrimitiveIndex); in CreateCallSwitchLayerInputs()
883 std::vector<AnfNodePtr> switch_layer_inputs = {switch_layer_cnode->input(kAnfPrimitiveIndex), in CreateCallSwitchLayerInputs()
946 auto attr_input = cnode->input(kAnfPrimitiveIndex); in CreateSwitchOrPartialNode()
978 auto attr_input = cnode->input(kAnfPrimitiveIndex); in CreateValueNode()
1031 auto attr_input = cnode->input(kAnfPrimitiveIndex); in CreateNewCNode()
1510 if (cnode->input(kAnfPrimitiveIndex)->isa<CNode>()) { in CreateCNodeOfKernelGraph()
1511 fullname = cnode->input(kAnfPrimitiveIndex)->fullname_with_scope(); in CreateCNodeOfKernelGraph()
1901 auto prim = cnode->input(kAnfPrimitiveIndex); in CNodeFirstInputIsPrimitive()
Danf_runtime_algorithm.cc425 return node->input(kAnfPrimitiveIndex); in GetCNodePrimitiveNode()
449 return IsPrimitive(cnode->input(kAnfPrimitiveIndex), primitive_type); in CheckPrimitiveType()
456 auto attr_input = cnode->input(kAnfPrimitiveIndex); in GetCNodeFuncGraphPtr()
1439 auto input = cnode->input(kAnfPrimitiveIndex); in IsGraphKernel()
1924 auto input = node->input(kAnfPrimitiveIndex); in IsCondControlKernel()
2321 return IsOneOfPrimitive(cnode->inputs().at(kAnfPrimitiveIndex), prim_set); in IsOneOfPrimitiveCNode()
Dkernel_graph.cc732 auto attr_input = front_node->input(kAnfPrimitiveIndex); in FrontBackendlMapAdd()
Dascend_session.cc910 node->set_input(kAnfPrimitiveIndex, NewValueNode(new_primitive)); in StoreCNodePrimitive()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc93 …AnfNodePtrList node_inputs = {cnode->input(kAnfPrimitiveIndex), cnode->input(kUpdateStateStateInpu… in Run()
Dgraph_kernel_helper.cc482 auto graph_kernel_fg = GetValueNode<FuncGraphPtr>(fuse_cnode->input(kAnfPrimitiveIndex)); in FuseNodesToSubGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc684 if (IsValueNode<Primitive>(cnode->input(kAnfPrimitiveIndex))) { in GetValidKernelNodes()
713 auto input0 = cnode->input(kAnfPrimitiveIndex); in GetFuncGraphOutputNodes()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h508 constexpr auto kAnfPrimitiveIndex = 0; variable