Searched refs:computeBaseIndices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions-inl.h | 227 computeBaseIndices(0, rowIndex, colIndex, otherIndex); in operator()
238 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in operator()
245 computeBaseIndices(0, rowIndex, colIndex, otherIndex); in loadPacket()
254 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in loadPacket()
448 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void computeBaseIndices( in computeBaseIndices() function
553 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
561 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
|
| D | eigen_cuboid_convolution.h | 294 computeBaseIndices(0, planeIndex, rowIndex, colIndex, otherIndex); in operator()
304 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in operator()
311 computeBaseIndices(0, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
320 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
579 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void computeBaseIndices( in computeBaseIndices() function
721 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
729 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
|