Searched defs:LayerNormGrad (Results 1 – 5 of 5) sorted by relevance
23 class LayerNormGrad : public OpDesc { class25 LayerNormGrad() { in LayerNormGrad() function in mindspore::graphkernel::expanders::LayerNormGrad
21 int LayerNormGrad(const float *x, const float *dy, const float *var, const float *mean, const float… in LayerNormGrad() function
356 cudaError_t LayerNormGrad(const int row_dim, const int col_dim, const int param_dim, const float ep… in LayerNormGrad() function
1325 LayerNormGrad() : BaseOperator(kNameLayerNormGrad) {} in LayerNormGrad() function
322 "LayerNormGrad": [ array