Searched defs:Y_data (Results 1 – 7 of 7) sorted by relevance
76 const T* __restrict__ Y_data = Y.const_data_ptr<T>(); in _typed_quantized_matmul() local
42 T* Y_data = Y->data_ptr<T>(); in LayerNormKernelImplInternal() local102 T* Y_data = Y->data_ptr<T>(); in layer_norm_kernel_mixed_type() local
49 T* Y_data = Y.data_ptr<T>(); in GroupNormKernelImplInternal() local304 T* Y_data = Y.data_ptr<T>(); in GroupNormKernelImplChannelsLastInternal() local
108 T* Y_data = static_cast<T*>(it->data_ptr(0)); in LogitMKLKernel() local
740 T* Y_data, in launch_vectorized_layer_norm_kernel()774 T* Y_data = Y->data_ptr<T>(); in LayerNormKernelImplInternal() local
2787 scalar_t* Y_data = Y->data_ptr<scalar_t>(); local2958 scalar_t* Y_data = result.data_ptr<scalar_t>(); local3012 scalar_t* Y_data = result.data_ptr<scalar_t>(); local3075 scalar_t* Y_data = Y->data_ptr<scalar_t>(); local
1010 unsigned char *Y_data, *UV_data, *tmp; in va_TraceSurface() local