Searched defs:X_ptr (Results 1 – 7 of 7) sorted by relevance
60 const T* X_ptr = X_data + i * inner_size; in GroupNormKernelImplInternal() local100 const T* X_ptr = X_data + m * C; in ColumnwiseMoments() local135 const T* X_ptr = X_data + m * C; in ColumnwiseMoments() local165 const T* X_ptr, in CalcMeanVar()188 const T* X_ptr, in CalcMeanVar()233 const T* X_ptr, in ApplyScaleBias()252 const T* X_ptr, in ApplyScaleBias()371 const T* X_ptr = X_data + n * HxW * C + m * C + g * D; in GroupNormKernelImplChannelsLastInternal() local412 const T* X_ptr = X_data + i * C; in GroupNormKernelImplChannelsLastInternal() local473 const T* X_ptr = X_data + i * C; in GroupNormKernelImplChannelsLastInternal() local[all …]
59 const T* X_ptr, in UpdateMomentsVec()82 const T* X_ptr, in UpdateMomentsVec()126 const T* X_ptr = X + i * kChunkSize * kVecSize; variable
52 const T* X_ptr = X_data + i * N; in LayerNormKernelImplInternal() local112 const T* X_ptr = X_data + i * N; in layer_norm_kernel_mixed_type() local203 const T* X_ptr = X_data + i * N; in layer_norm_backward_frame() local319 const T* X_ptr = X_data + i * N; in layer_norm_backward_frame() local
69 const T* X_ptr, in UpdateMomentsVec()113 const T* X_ptr = X + i * kChunkSize * kVecSize; variable
2405 typename T::underlying* X_ptr,2467 auto* X_ptr = reinterpret_cast<typename scalar_t::underlying*>(X + i * C); local2814 scalar_t* X_ptr = X_data + i * N; local2962 scalar_t* X_ptr = X_data + i * N; local3016 scalar_t* X_ptr = X_data + i * N; local3127 scalar_t* X_ptr = X_data + n * N * G + g * channels_per_group + hw * C; local3156 const scalar_t* X_ptr = X_data + n * N * G + g * channels_per_group + hwIdx * C; local3209 scalar_t* X_ptr = X_data + nhwIdx * C; local3278 const scalar_t* X_ptr = X_data + nhwIdx * C; local
50 X *X_ptr; variable
2061 X_ptr, argument