Home
last modified time | relevance | path

Searched defs:LayerNormGrad (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dlayernorm_grad.cc23 class LayerNormGrad : public OpDesc { class
25 LayerNormGrad() { in LayerNormGrad() function in mindspore::graphkernel::expanders::LayerNormGrad
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dlayernorm_grad.c21 int LayerNormGrad(const float *x, const float *dy, const float *var, const float *mean, const float… in LayerNormGrad() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dlayer_norm_grad_impl.cu356 cudaError_t LayerNormGrad(const int row_dim, const int col_dim, const int param_dim, const float ep… in LayerNormGrad() function
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h1325 LayerNormGrad() : BaseOperator(kNameLayerNormGrad) {} in LayerNormGrad() function
/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json322 "LayerNormGrad": [ array