Searched refs:m_rowInputStride (Results 1 – 5 of 5) sorted by relevance
208 m_rowInputStride = m_inputDepth * m_inputPlanes; in CustomTensorEvaluator()324 const Index inputIndex = depth + origInputRow * m_rowInputStride + in coeff()423 const Index inputIndex = depth + inputRows[0] * m_rowInputStride + in packet()542 inputCoords[2] * m_rowInputStride + in coeff()547 inputCoords[2] * m_rowInputStride + in coeff()614 Index m_rowInputStride; member
143 m_rowInputStride = patch_depth; in TensorContractionInputMapper()178 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()264 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_rowInputStride; } in patchDepth()308 const Index inputIndex = depth + origInputRow * m_rowInputStride + in loadCoeff()332 const Index inputIndex = depth + inputRow * m_rowInputStride + in loadCoeffStandard()397 const Index inputIndex = depth + inputRows[0] * m_rowInputStride + in loadPacketStandard()431 const Index inputIndex = depth + inputRow * m_rowInputStride + in loadPacketFast()486 Index m_rowInputStride; // row stride in the input tensor variable654 return m_base_mapper.m_rowInputStride; in patchDepth()720 return r * m_base_mapper.m_rowInputStride + in baseIndex()
166 m_rowInputStride = m_inputDepth * m_inputPlanes; in TensorContractionInputMapper()223 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()391 origInputRow * m_rowInputStride + in loadCoeff()426 inputRow * m_rowInputStride + in loadCoeffStandard()512 inputRows[0] * m_rowInputStride + in loadPacketStandard()558 inputRow * m_rowInputStride + in loadPacketFast()624 Index m_rowInputStride; // Row stride in the input tensor variable907 r * m_base_mapper.m_rowInputStride + in baseIndex()
292 m_rowInputStride = m_inputDepth;358 …const Index inputIndex = depth + origInputRow * m_rowInputStride + origInputCol * m_colInputStride…413 …const Index inputIndex = depth + inputRows[0] * m_rowInputStride + inputCols[0] * m_colInputStride…484 Index m_rowInputStride;
317 m_rowInputStride = m_inputDepth * m_inputPlanes;400 origInputRow * m_rowInputStride +487 inputRows[0] * m_rowInputStride +584 Index m_rowInputStride;