Home
last modified time | relevance | path

Searched refs:rowOffsets (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_cuboid_convolution.h516 const Index rowOffsets[]) const { in loadPacketStandardFromSingleColumnTwoRows() argument
518 rowOffsets[1] == rowOffsets[0] + 1); in loadPacketStandardFromSingleColumnTwoRows()
525 (((rowOffsets[1] * m_rowStride) + (colOffsets[0] * m_colStride)) * in loadPacketStandardFromSingleColumnTwoRows()
545 colOffsets[0], rowOffsets[0]), in loadPacketStandardFromSingleColumnTwoRows()
548 colOffsets[1], rowOffsets[1])); in loadPacketStandardFromSingleColumnTwoRows()
557 const Index rowOffsets[], const Index inputCols[], in loadPacketStandardFromSingleColumnSingleRow() argument
560 rowOffsets[1] == rowOffsets[0]); in loadPacketStandardFromSingleColumnSingleRow()
563 rowOffsets[0] * m_rowStride, in loadPacketStandardFromSingleColumnSingleRow()
565 rowOffsets[1] * m_rowStride}; in loadPacketStandardFromSingleColumnSingleRow()
622 const Index rowOffsets[2] = { in loadPacketStandard() local
[all …]
Deigen_volume_patch.h383 const Index rowOffsets[2] = { in packet() local
386 eigen_assert(rowOffsets[0] <= rowOffsets[1]); in packet()
389 rowIndex * m_row_strides + rowOffsets[0] - m_rowPaddingTop, in packet()
390 rowIndex * m_row_strides + rowOffsets[1] - m_rowPaddingTop}; in packet()
404 rowOffsets[0] * m_rowStride, in packet()
406 rowOffsets[1] * m_rowStride}; in packet()
Deigen_spatial_convolutions-inl.h373 const Index rowOffsets[2] = {patchOffsets[0] - colOffset * m_colStride, in loadPartialPacketStandard() local
375 const Index inputRows[2] = {rowIndex + rowOffsets[0], in loadPartialPacketStandard()
376 rowIndex + rowOffsets[1]}; in loadPartialPacketStandard()
455 const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0] * m_colStride, in loadPacketStandardFromSingleColumn() local
457 eigen_assert(rowOffsets[0] <= rowOffsets[1]); in loadPacketStandardFromSingleColumn()
458 const Index inputRows[2] = {rowIndex + rowOffsets[0], in loadPacketStandardFromSingleColumn()
459 rowIndex + rowOffsets[1]}; in loadPacketStandardFromSingleColumn()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h452 const Index rowOffsets[2] = {
455 eigen_assert(rowOffsets[0] <= rowOffsets[1]);
458 rowIndex * m_row_strides + rowOffsets[0] - m_rowPaddingTop,
459 rowIndex * m_row_strides + rowOffsets[1] - m_rowPaddingTop};
471 patchOffsets[0] - colOffsets[0] * m_colStride - rowOffsets[0] * m_rowStride,
472 patchOffsets[1] - colOffsets[1] * m_colStride - rowOffsets[1] * m_rowStride};
DTensorImagePatch.h399 …const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0]*m_colStride, patchOffsets[1] - colOff…
400 eigen_assert(rowOffsets[0] <= rowOffsets[1]);
402 const Index inputRows[2] = {rowIndex * m_row_strides + rowOffsets[0] -
403 m_rowPaddingTop, rowIndex * m_row_strides + rowOffsets[1] - m_rowPaddingTop};