Searched refs:kPrimBiasAddGrad (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | bias_add_grad.h | 32 BiasAddGrad() : PrimitiveC(prim::kPrimBiasAddGrad->name()) { InitIOName({"x"}, {"output"}); } in BiasAddGrad()
|
| D | bias_add_grad.cc | 77 REGISTER_PRIMITIVE_EVAL_IMPL(BiasAddGrad, prim::kPrimBiasAddGrad, BiasAddGradInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 56 {prim::kPrimBiasAddGrad->name(), {{0}, {}}},
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 69 {kGPUDevice, OpLevel_1, prim::kPrimBiasAddGrad}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_calculation_ops_declare.cc | 25 REG_ADPT_DESC(BiasAddGrad, prim::kPrimBiasAddGrad->name(), ADPT_DESC(BiasAddGrad))
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 227 {prim::kPrimBiasAddGrad, {InferImplBiasAddGrad, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 275 auto prim = prim::kPrimBiasAddGrad; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 327 inline const PrimitivePtr kPrimBiasAddGrad = std::make_shared<Primitive>(kBiasAddGrad); variable
|