Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dless_batch_normalization.cc41 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
44 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
57 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
71 …{prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTup…
72 …prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTupleGetItem, prim::kPrimBatc…
73 prim::kPrimConv2D},
76 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
87 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
90 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
101 …{prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTup…
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.h82 {{prim::kPrimConv2D, false}, "Conv2D"},
84 {{prim::kPrimConv2D, true}, "DepthwiseConv2D"},
92 {prim::kPrimConv2D, {schema::PrimitiveType_Conv2DFusion, kNumberTypeFloat32}},
Dconv2d_info.cc212 …if (!CheckPrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2… in CheckConv2DPrimitiveType()
Ddepthwise_conv2d_info.cc404 …if (!CheckPrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2… in CheckDepthWiseConv2DPrimitiveType()
/third_party/mindspore/tests/ut/cpp/transform/
Dop_adapter_test.cc82 auto prim = prim::kPrimConv2D;
87 ASSERT_EQ(prim->name(), prim::kPrimConv2D->name());
Dgraph_runner_test.cc58 PrimitivePtr conv2d = prim::kPrimConv2D; in MakeGeGraph()
Dconvert_test.cc91 PrimitivePtr conv2d = prim::kPrimConv2D; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dconv_bnreduce_fusion_pass.cc33 if (conv->isa<CNode>() && AnfAlgo::GetCNodeName(conv) == prim::kPrimConv2D->name() && in MatchConvBnreduce()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dconv1d_weight_expanding_pass.cc72 …if (!CheckPrimitiveType(node, prim::kPrimConv2D) && !CheckPrimitiveType(node, prim::kPrimConv2DFus… in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.h42 {prim::kPrimConv2D->name(), {{0, 1}, {0}}},
Dkernel_info_setter.cc384 if (kernel_name == prim::kPrimConv2D->name()) { in CheckSupportFormatTransform()
/third_party/mindspore/tests/ut/cpp/operator/
Dops_test.cc367 ASSERT_EQ(prim->name(), kPrimConv2D->name()); in TEST_F()
376 ASSERT_EQ(prim.name(), kPrimConv2D->name()); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dconv1d_inout_adjust.cc107 if (!opt::CheckPrimitiveType(cnode, prim::kPrimConv2D) && in Run()
/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc159 {prim::kPrimConv2D, {InferImplConv2D, nullptr, true}}, in GetPrimitiveToEvalImplMap()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dnn_calculation_ops_declare.cc37 REG_ADPT_DESC(Conv2D, prim::kPrimConv2D->name(), ADPT_DESC(Conv2D))
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dconv2d_unify_mindir.cc262 VectorRef pattern({prim::kPrimConv2D, X, W}); in DefinePattern()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc180 …return (CheckPrimitiveType(node, prim::kPrimConv2D) || CheckPrimitiveType(node, prim::kPrimConv2DF… in IsConv2D()
/third_party/mindspore/mindspore/core/ops/
Dconv2d.cc370 REGISTER_PRIMITIVE_EVAL_IMPL(Conv2D, prim::kPrimConv2D, Conv2dInfer, nullptr, true);
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc56 {prim::kPrimConv2D->name(), {kOpFormat_NC1HWC0, kOpFormat_FRAC_Z}}};
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc302 if (opt::CheckPrimitiveType(cnode, prim::kPrimConv2D)) { in MoveAttrMapConv2D()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h296 inline const PrimitivePtr kPrimConv2D = std::make_shared<Primitive>("Conv2D"); variable
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc699 …rimitiveType(anf_node, prim::kPrimConv2DFusion) || CheckPrimitiveType(anf_node, prim::kPrimConv2D); in IsParallelSplitConvNode()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc598 … return !(IsPrimitiveCNode(node, prim::kPrimConv2D) || IsPrimitiveCNode(node, prim::kPrimMatMul) || in FetchInfoForQuantExport()
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc628 IsPrimitiveCNode(cnode->input(1), prim::kPrimConv2D)) { in MatchAndMark()