Searched refs:m_gpuOutputStrides (Results 1 – 1 of 1) sorted by relevance
94 m_gpuOutputStrides[i] = in IndexMapper()95 m_gpuOutputStrides[i - 1] * gpuOutputDimensions[i - 1]; in IndexMapper()98 m_gpuOutputStrides[i] = 1; in IndexMapper()106 m_gpuOutputStrides[i] = in IndexMapper()107 m_gpuOutputStrides[i + 1] * gpuOutputDimensions[i + 1]; in IndexMapper()110 m_gpuOutputStrides[i] = 1; in IndexMapper()144 const Index idx = p / m_gpuOutputStrides[d]; in mapGpuOutputPlaneToTensorOutputOffset()146 p -= idx * m_gpuOutputStrides[d]; in mapGpuOutputPlaneToTensorOutputOffset()155 const Index idx = p / m_gpuOutputStrides[d]; in mapGpuOutputPlaneToTensorOutputOffset()157 p -= idx * m_gpuOutputStrides[d]; in mapGpuOutputPlaneToTensorOutputOffset()[all …]