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.h161 m_fastDimZero = internal::TensorIntDivisor<Index>(patch_depth); in TensorContractionInputMapper()
202 m_fastDimZero = base_mapper.m_fastDimZero; in TensorContractionInputMapper()
286 const Index patchOffset = patchId / m_fastDimZero; in loadCoeff()
322 const Index patchOffset = patchId / m_fastDimZero; in loadCoeffStandard()
370 patchId / m_fastDimZero, (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
417 const Index patchOffset = patchId / m_fastDimZero; in loadPacketFast()
418 eigen_assert((patchId + packetSize - 1) / m_fastDimZero == patchOffset); in loadPacketFast()
505 internal::TensorIntDivisor<Index> m_fastDimZero; variable
680 return m_base_mapper.m_fastDimZero; // patch_depth in fastPatchRowStride()
735 const Index patchOffset = m_depth_offset / m_base_mapper.m_fastDimZero; in rowOffset()
[all …]
Deigen_cuboid_convolution.h194 m_fastDimZero = internal::TensorIntDivisor<Index>(m_patch_depth); in TensorContractionInputMapper()
267 m_fastDimZero = base_mapper.m_fastDimZero; in TensorContractionInputMapper()
355 const Index patchOffset = patchId / m_fastDimZero; in loadCoeff()
406 const Index patchOffset = patchId / m_fastDimZero; in loadCoeffStandard()
470 patchId / m_fastDimZero, (patchId + packetSize - 1) / m_fastDimZero}; in loadPacketStandard()
537 const Index patchOffset = patchId / m_fastDimZero; in loadPacketFast()
538 eigen_assert((patchId + packetSize - 1) / m_fastDimZero == patchOffset); in loadPacketFast()
671 internal::TensorIntDivisor<Index> m_fastDimZero; // aka output depth variable
861 return m_base_mapper.m_fastDimZero; // patch_depth in fastPatchPlaneStride()
913 const Index patchOffset = m_depth_offset / m_base_mapper.m_fastDimZero; in planeOffset()
[all …]