Searched defs:x_bvec (Results 1 – 4 of 4) sorted by relevance
120 bVec x_bvec = bVec::loadu(X_ptr + d); in layer_norm_kernel_mixed_type() local370 bVec x_bvec = bVec::loadu(X_ptr, N); in layer_norm_backward_frame() local392 bVec x_bvec = bVec::loadu(X_ptr); in layer_norm_backward_frame() local465 bVec x_bvec = bVec::loadu(X_ptr + d); in layer_norm_backward_frame() local476 bVec x_bvec = bVec::loadu(X_ptr + d, N - d); in layer_norm_backward_frame() local
92 const Vec x_bvec = Vec::loadu(X_ptr + j * Vec::size()); in UpdateMomentsVec() local
1053 bVec x_bvec = bVec::loadu(x_ptr + d); in batch_norm_cpu_backward_contiguous_internal() local1188 bVec x_bvec = bVec::loadu(x_ptr + d); in batch_norm_cpu_backward_channels_last_internal() local1237 bVec x_bvec = bVec::loadu(x_ptr + d); in batch_norm_cpu_backward_channels_last_internal() local
138 Vec x_bvec = Vec::loadu(X_ptr + d); in ColumnwiseMoments() local144 Vec x_bvec = Vec::loadu(X_ptr + d, D - d); in ColumnwiseMoments() local584 const Vec x_bvec = Vec::loadu(X_ptr + j); in ComputeInternalGradients() local