Searched defs:LayerNormGrad (Results 1 – 5 of 5) sorted by relevance
32 LayerNormGrad() : PrimitiveC(kNameLayerNormGrad) {} in LayerNormGrad() function33 explicit LayerNormGrad(const std::string k_name) : PrimitiveC(k_name) {} in LayerNormGrad() function
22 class LayerNormGrad(Expander): class
21 int LayerNormGrad(const float *x, const float *dy, const float *var, const float *mean, const float… in LayerNormGrad() function
236 void LayerNormGrad(const int &row_dim, const int &col_dim, const int ¶m_dim, const T &epsilon, … in LayerNormGrad() function
1160 class LayerNormGrad(Primitive): class