Searched refs:m_num_patches (Results 1 – 2 of 2) sorted by relevance
111 m_num_patches = tensor.impl().dimensions()[4]; in TensorContractionInputMapper()118 m_num_patches = tensor.impl().dimensions()[NumDims - 5]; in TensorContractionInputMapper()133 m_otherStride = m_patchStride * m_num_patches; in TensorContractionInputMapper()177 m_fastNumPatches = internal::TensorIntDivisor<Index>(m_num_patches); in TensorContractionInputMapper()213 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()784 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()803 Index m_num_patches; // number of patches to extract variable
113 m_num_patches = tensor.impl().dimensions()[3]; in TensorContractionInputMapper()119 m_num_patches = tensor.impl().dimensions()[NumDims - 4]; in TensorContractionInputMapper()163 m_fastNumPatches = internal::TensorIntDivisor<Index>(m_num_patches); in TensorContractionInputMapper()173 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()620 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()629 Index m_num_patches; // number of patches to extract. variable