Searched refs:kPrimLayerNormBetaGammaBackprop (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | LayerNormBetaGammaBackprop.h | 32 LayerNormBetaGammaBackprop() : PrimitiveC(prim::kPrimLayerNormBetaGammaBackprop->name()) {} in LayerNormBetaGammaBackprop()
|
| D | LayerNormBetaGammaBackprop.cc | 65 REGISTER_PRIMITIVE_EVAL_IMPL(LayerNormBetaGammaBackprop, prim::kPrimLayerNormBetaGammaBackprop,
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 352 inline const PrimitivePtr kPrimLayerNormBetaGammaBackprop = std::make_shared<Primitive>("LayerNormB… variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 160 …{prim::kPrimLayerNormBetaGammaBackprop->name(), {{{0, 1}, {1, 0}, {2, 2}, {3, 3}}, {{0, 1}, {1, 0}…
|