Home
last modified time | relevance | path

Searched refs:kPrimConv2DFusion (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.h83 {{prim::kPrimConv2DFusion, false}, "Conv2D"},
85 {{prim::kPrimConv2DFusion, true}, "DepthwiseConv2D"}};
93 {prim::kPrimConv2DFusion, {schema::PrimitiveType_Conv2DFusion, kNumberTypeFloat32}}};
Dconv2d_info.cc212 …PrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2DFusion)) { in CheckConv2DPrimitiveType()
Ddepthwise_conv2d_info.cc404 …PrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2DFusion)) { in CheckDepthWiseConv2DPrimitiveType()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dunused_transpose_node_remove_pass.cc77 if (!CheckPrimitiveType(transpose_cnode->input(kTransposeInput), prim::kPrimConv2DFusion)) { in Run()
88 } else if (CheckPrimitiveType(node, prim::kPrimConv2DFusion)) { in Run()
Dupdate_conv2d_param_pass.cc94 if (CheckPrimitiveType(cnode, prim::kPrimConv2DFusion)) { in UpdateConv2DAttr()
112 if (CheckPrimitiveType(node, prim::kPrimConv2DFusion) || in Run()
Dconv1d_weight_expanding_pass.cc72 …heckPrimitiveType(node, prim::kPrimConv2D) && !CheckPrimitiveType(node, prim::kPrimConv2DFusion)) { in Run()
Dgroup_depthwise_op_convert_pass.cc41 if (!CheckPrimitiveType(node, prim::kPrimConv2DFusion)) { in Run()
Dnode_infershape.cc391 …if (CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || CheckPrimitiveType(cnode, prim::kPrimAda… in GetCNodeOutputTensors()
407 if (CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || in GetCNodeOutputTensors()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunify_format.cc64 if (opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion)) { in DecideTFConvWeightSrcFormat()
87 if (opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion)) { in DecideTFLITEConvWeightSrcFormat()
117 if (opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || in DecideONNXConvWeightSrcFormat()
Dconv1d_inout_adjust.cc108 !opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion)) { in Run()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_activation_fusion.cc63 if (CheckPrimitiveType(conv_node, prim::kPrimConv2DFusion) || in Process()
Dconv_tuple_activation_fusion.cc75 if (CheckPrimitiveType(conv_node, prim::kPrimConv2DFusion)) { in Process()
Dconv_transform_fusion.cc36 if (CheckPrimitiveType(conv_node, prim::kPrimConv2DFusion)) { in GetOutChannels()
295 if (CheckPrimitiveType(conv_node, prim::kPrimConv2DFusion)) { in CalNewWeightTensor()
Dconv_biasadd_fusion.cc30 return CheckPrimitiveType(anf_node, prim::kPrimConv2DFusion) || in IsConvExtendNode()
Dconv_pad_fusion.cc158 auto is_conv = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimConv2DFusion>); in DefinePadConvPattern()
Dconv_conv_fusion.cc39 if (!CheckPrimitiveType(anf_node, prim::kPrimConv2DFusion)) { in IsCommonConvNode()
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Dto_format_base.h46 return opt::CheckPrimitiveType(node, prim::kPrimConv2DFusion) || in IsConvFamilyNode()
/third_party/mindspore/mindspore/lite/tools/common/
Dnode_util.cc338 …(opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || opt::CheckPrimitiveType(cnode, prim::k… in GetCNodeOutputsSize()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc180 …(CheckPrimitiveType(node, prim::kPrimConv2D) || CheckPrimitiveType(node, prim::kPrimConv2DFusion)); in IsConv2D()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc699 …return CheckPrimitiveType(anf_node, prim::kPrimConv2DFusion) || CheckPrimitiveType(anf_node, prim:… in IsParallelSplitConvNode()
725 return CheckPrimitiveType(anf_node, prim::kPrimConv2DFusion) || in IsConvNode()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc111 … prim::kPrimConv2DFusion, prim::kPrimConv2dTransposeFusion, in CanOpFullQuantized()
Dfull_quant_quantizer.cc52 static const std::set<PrimitivePtr> has_bias_operator = {prim::kPrimConv2DFusion, prim::kPrimConv2d…
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc842 if (opt::CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || in SetOpOutputNode()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h644 inline const PrimitivePtr kPrimConv2DFusion = std::make_shared<Primitive>("Conv2DFusion"); variable