Searched refs:kPrimLayerNormGrad (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | layer_norm_grad.cc | 60 REGISTER_PRIMITIVE_EVAL_IMPL(LayerNormGrad, prim::kPrimLayerNormGrad, LayerNormGradInfer, nullptr, …
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | layer_norm_grad_split.cc | 97 VectorRef pattern({prim::kPrimLayerNormGrad, Xs}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_norm_ops_declare.cc | 104 REG_ADPT_DESC(LayerNormGrad, prim::kPrimLayerNormGrad->name(), ADPT_DESC(LayerNormGrad))
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 77 {kGPUDevice, OpLevel_1, prim::kPrimLayerNormGrad}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 349 inline const PrimitivePtr kPrimLayerNormGrad = std::make_shared<Primitive>(kLayerNormGrad); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 158 {prim::kPrimLayerNormGrad->name(),
|