Searched refs:computeBaseIndices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions-inl.h | 233 computeBaseIndices(0, rowIndex, colIndex, otherIndex); in operator()
244 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in operator()
251 computeBaseIndices(0, rowIndex, colIndex, otherIndex); in loadPacket()
260 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in loadPacket()
612 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void computeBaseIndices( in computeBaseIndices() function
720 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
728 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
|
| D | eigen_cuboid_convolution.h | 296 computeBaseIndices(0, planeIndex, rowIndex, colIndex, otherIndex); in operator()
306 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in operator()
313 computeBaseIndices(0, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
322 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
772 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void computeBaseIndices( in computeBaseIndices() function
914 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
922 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
|