Searched refs:m_outputPlanes (Results 1 – 3 of 3) sorted by relevance
99 m_outputPlanes = Eigen::divup( in CustomTensorEvaluator()121 m_outputPlanes = Eigen::divup( in CustomTensorEvaluator()132 m_outputPlanes = Eigen::divup(m_input_planes_eff, m_plane_strides); in CustomTensorEvaluator()136 0, (m_outputPlanes - 1) * m_plane_strides + m_patch_planes_eff - in CustomTensorEvaluator()155 eigen_assert(m_outputPlanes > 0); in CustomTensorEvaluator()170 m_dimensions[4] = m_outputPlanes * m_outputRows * m_outputCols; in CustomTensorEvaluator()186 m_dimensions[NumDims - 5] = m_outputPlanes * m_outputRows * m_outputCols; in CustomTensorEvaluator()213 m_outputPlanesRows = m_outputPlanes * m_outputRows; in CustomTensorEvaluator()227 m_fastOutputPlanes = internal::TensorIntDivisor<Index>(m_outputPlanes); in CustomTensorEvaluator()302 (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex)); in coeff()[all …]
133 m_outputPlanes = tensor.impl().outputPlanes(); in TensorContractionInputMapper()137 m_outputPlanesRows = m_outputPlanes * m_outputRows; in TensorContractionInputMapper()196 m_fastOutputPlanes = internal::TensorIntDivisor<Index>(m_outputPlanes); in TensorContractionInputMapper()233 m_outputPlanes = base_mapper.m_outputPlanes; in TensorContractionInputMapper()599 patch3DIndex - (colIndex * m_outputRows + rowIndex) * m_outputPlanes; in computeBaseIndices()634 Index m_outputPlanes; // Number of output planes variable
230 …m_outputPlanes = numext::ceil((m_input_planes_eff + op.padding_top_z() + op.padding_bottom_z() - m…240 …m_outputPlanes = numext::ceil((m_input_planes_eff - m_patch_planes_eff + 1.f) / static_cast<float>…248 m_outputPlanes = numext::ceil(m_input_planes_eff / static_cast<float>(m_plane_strides));251 … const Index dz = m_outputPlanes * m_plane_strides + m_patch_planes_eff - 1 - m_input_planes_eff;265 eigen_assert(m_outputPlanes > 0);280 m_dimensions[4] = m_outputPlanes * m_outputRows * m_outputCols;296 m_dimensions[NumDims-5] = m_outputPlanes * m_outputRows * m_outputCols;321 m_outputPlanesRows = m_outputPlanes * m_outputRows;332 m_fastOutputPlanes = internal::TensorIntDivisor<Index>(m_outputPlanes);387 const Index planeIndex = (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex));[all …]