Home
last modified time | relevance | path

Searched refs:m_fastDimZero (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions-inl.h166 m_fastDimZero = internal::TensorIntDivisor<Index>(patch_depth); in TensorContractionInputMapper()
208 m_fastDimZero = base_mapper.m_fastDimZero; in TensorContractionInputMapper()
292 const Index patchOffset = patchId / m_fastDimZero; in loadCoeff()
328 const Index patchOffset = patchId / m_fastDimZero; in loadCoeffStandard()
424 const Index patchOffsetSplit = patchIdSplit / m_fastDimZero; in loadPacketStandardFromTwoColumns()
499 const Index patchOffsets[2] = {patchId / m_fastDimZero, in loadPacketStandard()
500 (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
547 const Index patchOffsets[2] = {patchId / m_fastDimZero, in loadPacketStandard()
548 (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
581 const Index patchOffset = patchId / m_fastDimZero; in loadPacketFast()
[all …]
Deigen_cuboid_convolution.h196 m_fastDimZero = internal::TensorIntDivisor<Index>(m_patch_depth); in TensorContractionInputMapper()
269 m_fastDimZero = base_mapper.m_fastDimZero; in TensorContractionInputMapper()
357 const Index patchOffset = patchId / m_fastDimZero; in loadCoeff()
408 const Index patchOffset = patchId / m_fastDimZero; in loadCoeffStandard()
528 const Index patchOffsetSplit = patchIdSplit / m_fastDimZero; in loadPacketStandardFromSingleColumnTwoRows()
609 patchId / m_fastDimZero, (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
676 patchId / m_fastDimZero, (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
730 const Index patchOffset = patchId / m_fastDimZero; in loadPacketFast()
731 eigen_assert((patchId + packetSize - 1) / m_fastDimZero == patchOffset); in loadPacketFast()
864 internal::TensorIntDivisor<Index> m_fastDimZero; // aka output depth variable
[all …]