Searched defs:Y_ptr (Results 1 – 3 of 3) sorted by relevance
64 T* Y_ptr = Y_data + i * inner_size; in GroupNormKernelImplInternal() local75 T* Y_ptr = Y_data + (i * D + j) * HxW; in GroupNormKernelImplInternal() local232 T* Y_ptr, in ApplyScaleBias()251 T* Y_ptr, in ApplyScaleBias()372 T* Y_ptr = Y_data + n * HxW * C + m * C + g * D; in GroupNormKernelImplChannelsLastInternal() local474 T* Y_ptr = Y_data + i * C; in GroupNormKernelImplChannelsLastInternal() local
53 T* Y_ptr = Y_data + i * N; in LayerNormKernelImplInternal() local113 T* Y_ptr = Y_data + i * N; in layer_norm_kernel_mixed_type() local
2406 typename T::underlying* Y_ptr,2468 auto* Y_ptr = reinterpret_cast<typename scalar_t::underlying*>(Y + i * C); local2815 scalar_t* Y_ptr = Y_data + i * N; local2963 scalar_t* Y_ptr = Y_data + i; local3017 scalar_t* Y_ptr = Y_data + i; local3157 scalar_t* Y_ptr = Y_data + n * N * G + g * channels_per_group + hwIdx * C; local3279 scalar_t* Y_ptr = Y_data + nhwIdx * C; local