Searched refs:kPrimReluGrad (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | relu_grad.h | 32 ReLUGrad() : PrimitiveC(prim::kPrimReluGrad->name()) { InitIOName({"x"}, {"output"}); } in ReLUGrad()
|
| D | relu_grad.cc | 72 REGISTER_PRIMITIVE_EVAL_IMPL(ReLUGrad, prim::kPrimReluGrad, ReLUGradInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 59 {prim::kPrimReluGrad->name(), {{0, 1}, {0}}},
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_grad_fusion.cc | 32 VectorRef relu_grad = VectorRef({prim::kPrimReluGrad, dy_, y_}); in DefinePattern()
|
| D | relu_v2_pass.cc | 126 VectorRef relu_grad({prim::kPrimReluGrad, dy_, VectorRef({prim::kPrimRelu, x_})}); in DefinePattern()
|
| D | batch_norm_add_relu_grad_fusion.cc | 152 VectorRef relu_grad = VectorRef({prim::kPrimReluGrad, dy_, y_}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | derelu_fusion.cc | 102 VectorRef relu_grad({prim::kPrimReluGrad, i0, relu}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nonlinear_fuc_ops_declare.cc | 108 REG_ADPT_DESC(ReluGrad, prim::kPrimReluGrad->name(), ADPT_DESC(ReluGrad))
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 83 {kGPUDevice, OpLevel_0, prim::kPrimReluGrad}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | eltwise_grad_cpu_kernel.cc | 236 elt_map{{prim::kPrimReluGrad->name(), &EltWiseGradCPUKernel<T>::ReluGrad}, in Launch()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 230 auto prim = prim::kPrimReluGrad; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 311 inline const PrimitivePtr kPrimReluGrad = std::make_shared<Primitive>(kReLUGrad); variable
|