Searched refs:kPrimReluV2 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reluv2.h | 35 ReLUV2() : PrimitiveC(prim::kPrimReluV2->name()) { InitIOName({"x"}, {"output", "mask"}); } in ReLUV2()
|
| D | reluv2.cc | 109 REGISTER_PRIMITIVE_EVAL_IMPL(ReLUV2, prim::kPrimReluV2, ReLUV2Infer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | conv_double_in_fusion_pass.cc | 66 !AnfAlgo::CheckPrimitiveType(node, prim::kPrimReluV2)) { in MatchSingleFusionPattern()
|
| D | depthwiseconv_eltwise_fusion_pass.cc | 44 …ode>() && (IsPrimitiveCNode(relu, prim::kPrimRelu) || IsPrimitiveCNode(relu, prim::kPrimReluV2))) { in MatchDepthwiseConvRelu()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | add_relu_v2_fusion.cc | 56 VectorRef relu = VectorRef({prim::kPrimReluV2, VectorRef({prim::kPrimAdd, x1_, x2_})}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 370 inline const PrimitivePtr kPrimReluV2 = std::make_shared<Primitive>(kReLUV2); variable
|