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