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
148 m_rowInputStride = patch_depth; in TensorContractionInputMapper()183 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()270 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_rowInputStride; } in patchDepth()314 const Index inputIndex = depth + origInputRow * m_rowInputStride + in loadCoeff()338 const Index inputIndex = depth + inputRow * m_rowInputStride + in loadCoeffStandard()386 const Index inputIndex = depth + inputRows[0] * m_rowInputStride + in loadPartialPacketStandard()423 ((colOffsets[1] * m_colStride) * m_rowInputStride) - 1; in loadPacketStandardFromTwoColumns()469 const Index inputIndex = depth + inputRows[0] * m_rowInputStride + in loadPacketStandardFromSingleColumn()595 const Index inputIndex = depth + inputRow * m_rowInputStride + in loadPacketFast()650 Index m_rowInputStride; // row stride in the input tensor variable[all …]
168 m_rowInputStride = m_inputDepth * m_inputPlanes; in TensorContractionInputMapper()225 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()393 origInputRow * m_rowInputStride + in loadCoeff()428 inputRow * m_rowInputStride + in loadCoeffStandard()485 inputRow * m_rowInputStride + in loadPartialPacketStandard()576 inputRows[0] * m_rowInputStride + in loadPacketStandardFromSingleColumnSingleRow()751 inputRow * m_rowInputStride + in loadPacketFast()817 Index m_rowInputStride; // Row stride in the input tensor variable1101 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;