Searched refs:m_gpuInputStrides (Results 1 – 1 of 1) sorted by relevance
92 m_gpuInputStrides[i] = in IndexMapper()93 m_gpuInputStrides[i - 1] * gpuInputDimensions[i - 1]; in IndexMapper()97 m_gpuInputStrides[i] = 1; in IndexMapper()104 m_gpuInputStrides[i] = in IndexMapper()105 m_gpuInputStrides[i + 1] * gpuInputDimensions[i + 1]; in IndexMapper()109 m_gpuInputStrides[i] = 1; in IndexMapper()120 const Index idx = p / m_gpuInputStrides[d]; in mapGpuInputPlaneToTensorInputOffset()122 p -= idx * m_gpuInputStrides[d]; in mapGpuInputPlaneToTensorInputOffset()131 const Index idx = p / m_gpuInputStrides[d]; in mapGpuInputPlaneToTensorInputOffset()133 p -= idx * m_gpuInputStrides[d]; in mapGpuInputPlaneToTensorInputOffset()[all …]