Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_spatial_convolutions-inl.h128 m_patch_col_inflate_strides = tensor.impl().colInflateStride(); in TensorContractionInputMapper()
163 internal::TensorIntDivisor<Index>(m_patch_col_inflate_strides); in TensorContractionInputMapper()
180 m_patch_col_inflate_strides = base_mapper.m_patch_col_inflate_strides; in TensorContractionInputMapper()
218 m_patch_row_inflate_strides != 1 || m_patch_col_inflate_strides != 1; in nonStandardPatches()
298 (m_patch_col_inflate_strides == 1) in loadCoeff()
310 (inputCol != origInputCol * m_patch_col_inflate_strides) || in loadCoeff()
640 Index m_patch_col_inflate_strides; // the strides for col inflation in the variable
937 *orig_col = (m_base_mapper.m_patch_col_inflate_strides == 1) in padOrSkipCol()
944 (input_col != *orig_col * m_base_mapper.m_patch_col_inflate_strides); in padOrSkipCol()
H A Deigen_cuboid_convolution.h151 m_patch_col_inflate_strides = tensor.impl().colInflateStride(); in TensorContractionInputMapper()
191 internal::TensorIntDivisor<Index>(m_patch_col_inflate_strides); in TensorContractionInputMapper()
249 m_patch_col_inflate_strides = base_mapper.m_patch_col_inflate_strides; in TensorContractionInputMapper()
280 m_patch_row_inflate_strides != 1 || m_patch_col_inflate_strides != 1; in nonStandardPatches()
362 (m_patch_col_inflate_strides == 1) in loadCoeff()
385 (inputCol != origInputCol * m_patch_col_inflate_strides) || in loadCoeff()
844 Index m_patch_col_inflate_strides; variable