Searched defs:rstd_data (Results 1 – 8 of 8) sorted by relevance
44 T* rstd_data = rstd ? rstd->data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local104 param_t* rstd_data = rstd ? rstd->data_ptr<param_t>() : nullptr; in layer_norm_kernel_mixed_type() local189 const T2* rstd_data, in layer_norm_backward_frame()304 const float* rstd_data, in layer_norm_backward_frame()511 const T2* rstd_data = rstd.template const_data_ptr<T2>(); in LayerNormBackwardKernelImplInternal() local
51 PT* rstd_data = rstd.data_ptr<PT>(); in GroupNormKernelImplInternal() local306 PT* rstd_data = rstd.data_ptr<PT>(); in GroupNormKernelImplChannelsLastInternal() local903 const PT* rstd_data = rstd.const_data_ptr<PT>(); in GroupNormBackwardKernelImplInternal() local1383 const PT* rstd_data = rstd.const_data_ptr<PT>(); in GroupNormBackwardKernelImplChannelsLastInternal() local
577 T* rstd_data = rstd.mutable_data_ptr<T>(); in GroupNormKernelImplInternal() local688 const T* rstd_data = rstd.const_data_ptr<T>(); in GroupNorm1dBackward() local832 const T* rstd_data = rstd.const_data_ptr<T>(); in GroupNormBackwardKernelImplInternal() local
742 T_ACC* rstd_data in launch_vectorized_layer_norm_kernel()776 T_ACC* rstd_data = rstd->data_ptr<T_ACC>(); in LayerNormKernelImplInternal() local1155 const T_ACC* rstd_data = rstd.template const_data_ptr<T_ACC>(); in LayerNormBackwardKernelImplInternal() local
45 CTYPE* rstd_data = rstd.mutable_data_ptr<CTYPE>(); in layer_norm() local
51 CTYPE* rstd_data = rstd.mutable_data_ptr<CTYPE>(); in group_norm() local
50 CTYPE* rstd_data = rstd.mutable_data_ptr<CTYPE>(); in layer_norm() local
3197 float* rstd_data = rstd.mutable_data_ptr<float>(); local