Searched defs:X_data (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | group_norm_kernel.cpp | 46 const T* X_data = X.const_data_ptr<T>(); in GroupNormKernelImplInternal() local
91 const T* X_data, in ColumnwiseMoments()
124 const T* X_data, in ColumnwiseMoments()
301 const T* X_data = X.const_data_ptr<T>(); in GroupNormKernelImplChannelsLastInternal() local
901 const T* X_data = X.const_data_ptr<T>(); in GroupNormBackwardKernelImplInternal() local
1055 const T* X_data, in ApplyInputGradientsChannelsLastColMov()
1103 const T* X_data, in ApplyInputGradientsChannelsLastColMov()
1160 const T* X_data, in ApplyInputGradientsChannelsLastRowMov()
1196 const T* X_data, in ApplyInputGradientsChannelsLastRowMov()
1242 const T* X_data, in CalcInternalGradientsChannelsLast()
[all …]
|
| H A D | layer_norm_kernel.cpp | 39 const T* X_data = X.const_data_ptr<T>(); in LayerNormKernelImplInternal() local
99 const T* X_data = X.const_data_ptr<T>(); in layer_norm_kernel_mixed_type() local
187 const T* X_data, in layer_norm_backward_frame()
302 const T* X_data, in layer_norm_backward_frame()
509 const T* X_data = X.template const_data_ptr<T>(); in LayerNormBackwardKernelImplInternal() local
|
| H A D | UnaryOpsKernel.cpp | 107 const T* X_data = static_cast<T*>(it->data_ptr(1)); in LogitMKLKernel() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | group_norm_kernel.cu | 575 const T* X_data = X.const_data_ptr<T>(); in GroupNormKernelImplInternal() local
686 const T* X_data = X.const_data_ptr<T>(); in GroupNorm1dBackward() local
830 const T* X_data = X.const_data_ptr<T>(); in GroupNormBackwardKernelImplInternal() local
|
| H A D | layer_norm_kernel.cu | 737 const T* X_data, in launch_vectorized_layer_norm_kernel()
771 const T* X_data = X.const_data_ptr<T>(); in LayerNormKernelImplInternal() local
1153 const T* X_data = X.template const_data_ptr<T>(); in LayerNormBackwardKernelImplInternal() local
|
| /aosp_15_r20/external/executorch/backends/cadence/reference/operators/ |
| H A D | quantized_matmul_out.cpp | 75 const T* __restrict__ X_data = X.const_data_ptr<T>(); in _typed_quantized_matmul() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2784 scalar_t* X_data = X.data_ptr<scalar_t>(); local
2957 scalar_t* X_data = self.data_ptr<scalar_t>(); local
3011 scalar_t* X_data = self.data_ptr<scalar_t>(); local
3072 scalar_t* X_data = X.data_ptr<scalar_t>(); local
|