Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_spatial_convolutions-inl.h423 const Index patchIdSplit = in loadPacketStandardFromTwoColumns() local
425 const Index patchOffsetSplit = patchIdSplit / m_fastDimZero; in loadPacketStandardFromTwoColumns()
431 const Index patchIds[2] = {patchId, patchIdSplit + 1}; in loadPacketStandardFromTwoColumns()
432 const Index spans[2][2] = {{0, patchIdSplit - patchId}, in loadPacketStandardFromTwoColumns()
433 {patchIdSplit - patchId + 1, packetSize - 1}}; in loadPacketStandardFromTwoColumns()
H A Deigen_cuboid_convolution.h524 const Index patchIdSplit = in loadPacketStandardFromSingleColumnTwoRows() local
528 const Index patchOffsetSplit = patchIdSplit / m_fastDimZero; in loadPacketStandardFromSingleColumnTwoRows()
534 const Index patchIds[2] = {patchId, patchIdSplit + 1}; in loadPacketStandardFromSingleColumnTwoRows()
535 const Index spans[2][2] = {{0, patchIdSplit - patchId}, in loadPacketStandardFromSingleColumnTwoRows()
536 {patchIdSplit - patchId + 1, packetSize - 1}}; in loadPacketStandardFromSingleColumnTwoRows()