Home
last modified time | relevance | path

Searched refs:mapGpuInputKernelToTensorInputOffset (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorConvolution.h164 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapGpuInputKernelToTensorInputOffset(Index i) const { in mapGpuInputKernelToTensorInputOffset() function
178 …EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapGpuInputKernelToTensorInputOffset(Index i, Index j)… in mapGpuInputKernelToTensorInputOffset() function
192 …EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapGpuInputKernelToTensorInputOffset(Index i, Index j,… in mapGpuInputKernelToTensorInputOffset() function
607 …const int tensor_index = plane_input_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(i+f…
670 …const int tensor_index = plane_input_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(i+f…
739 …const int tensor_index = plane_input_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(i+f…
H A DTensorConvolutionSycl.h73 plane_tensor_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(i + input_offset);
148 … const size_t tensor_index = plane_input_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(
239 plane_input_offset + indexMapper.mapGpuInputKernelToTensorInputOffset(