Searched defs:param_dim (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | layer_norm_impl.cu | 103 inline __device__ void LayerNorm(const int &row, const int &col_dim, const int ¶m_dim, const T … in LayerNorm()
113 inline __device__ void LayerNorm(const int &row, const int &col_dim, const int ¶m_dim, const ha… in LayerNorm()
124 __global__ void LayerNormKernel(const int row_dim, const int col_dim, const int param_dim, const T … in LayerNormKernel()
143 void LayerNorm(const int &row_dim, const int &col_dim, const int ¶m_dim, const T &epsilon, cons… in LayerNorm()
|
| D | layer_norm_grad_impl.cu | 110 inline __device__ void InputThreadReduce(const int &row, const int &col_dim, const int ¶m_dim, … in InputThreadReduce()
134 inline __device__ void InputThreadReduce(const int &row, const int &col_dim, const int ¶m_dim, … in InputThreadReduce()
190 inline __device__ void InputProp(const int &row, const int &col_dim, const int ¶m_dim, const T … in InputProp()
205 inline __device__ void InputProp(const int &row, const int &col_dim, const int ¶m_dim, const ha… in InputProp()
221 __global__ void InputPropKernel(const int row_dim, const int col_dim, const int param_dim, const T … in InputPropKernel()
236 void LayerNormGrad(const int &row_dim, const int &col_dim, const int ¶m_dim, const T &epsilon, … in LayerNormGrad()
|
| D | layer_norm_grad_grad_impl.cu | 125 …_device__ void InputThreadReduceInnerMean(const int &row, const int &col_dim, const int ¶m_dim, in InputThreadReduceInnerMean()
192 …_device__ void InputThreadReduceOuterMean(const int &row, const int &col_dim, const int ¶m_dim, in InputThreadReduceOuterMean()
230 …_device__ void InputThreadReduceOuterMean(const int &row, const int &col_dim, const int ¶m_dim, in InputThreadReduceOuterMean()
303 inline __device__ void InputProp(const int &row, const int &col_dim, const int ¶m_dim, const T … in InputProp()
331 inline __device__ void InputProp(const int &row, const int &col_dim, const int ¶m_dim, const ha… in InputProp()
360 __global__ void InputPropKernel(const int row_dim, const int col_dim, const int param_dim, const T … in InputPropKernel()
390 void LayerNormGradGrad(const int &row_dim, const int &col_dim, const int ¶m_dim, T *global_sum1… in LayerNormGradGrad()
|