Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_cuboid_convolution.h109 m_num_patches = tensor.impl().dimensions()[4]; in TensorContractionInputMapper()
116 m_num_patches = tensor.impl().dimensions()[NumDims - 5]; in TensorContractionInputMapper()
131 m_otherStride = m_patchStride * m_num_patches; in TensorContractionInputMapper()
175 m_fastNumPatches = internal::TensorIntDivisor<Index>(m_num_patches); in TensorContractionInputMapper()
211 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()
591 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()
610 Index m_num_patches; // number of patches to extract variable
Deigen_spatial_convolutions-inl.h109 m_num_patches = tensor.impl().dimensions()[3]; in TensorContractionInputMapper()
115 m_num_patches = tensor.impl().dimensions()[NumDims - 4]; in TensorContractionInputMapper()
158 m_fastNumPatches = internal::TensorIntDivisor<Index>(m_num_patches); in TensorContractionInputMapper()
168 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()
456 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()
465 Index m_num_patches; // number of patches to extract. variable