Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h195 m_colStride = m_dimensions[2] * m_rowStride; in CustomTensorEvaluator()
196 m_patchStride = m_colStride * m_dimensions[3] * m_dimensions[0]; in CustomTensorEvaluator()
200 m_colStride = m_dimensions[NumDims - 3] * m_rowStride; in CustomTensorEvaluator()
202 m_colStride * m_dimensions[NumDims - 4] * m_dimensions[NumDims - 1]; in CustomTensorEvaluator()
218 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride); in CustomTensorEvaluator()
287 (patchOffset - colOffset * m_colStride) / m_fastRowStride; in coeff()
304 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in coeff()
384 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride, in packet()
385 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride}; in packet()
403 patchOffsets[0] - colOffsets[0] * m_colStride - in packet()
[all …]
Deigen_cuboid_convolution.h129 m_colStride = m_patch_rows * m_rowStride; in TensorContractionInputMapper()
130 m_patchStride = m_colStride * m_patch_cols * m_patch_depth; in TensorContractionInputMapper()
192 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride); in TensorContractionInputMapper()
218 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()
365 (patchOffset - colOffset * m_colStride) / m_fastRowStride; in loadCoeff()
373 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeff()
410 (patchOffset - colOffset * m_colStride) / m_fastRowStride; in loadCoeffStandard()
412 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeffStandard()
484 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride, in loadPacketStandard()
485 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride}; in loadPacketStandard()
[all …]
Deigen_spatial_convolutions-inl.h125 m_colStride = patch_rows; in TensorContractionInputMapper()
159 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride); in TensorContractionInputMapper()
176 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()
266 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_colStride; } in patchRows()
295 const Index rowOffset = patchOffset - colOffset * m_colStride; in loadCoeff()
324 const Index rowOffset = patchOffset - colOffset * m_colStride; in loadCoeffStandard()
383 patchOffsets[0] - colOffsets[0] * m_colStride, in loadPacketStandard()
384 patchOffsets[1] - colOffsets[1] * m_colStride}; in loadPacketStandard()
421 const Index rowOffset = patchOffset - colOffset * m_colStride; in loadPacketFast()
482 Index m_colStride; variable
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h305 m_colStride = m_dimensions[2] * m_rowStride;
306 m_patchStride = m_colStride * m_dimensions[3] * m_dimensions[0];
310 m_colStride = m_dimensions[NumDims-3] * m_rowStride;
311 m_patchStride = m_colStride * m_dimensions[NumDims-4] * m_dimensions[NumDims-1];
326 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride);
378 const Index rowOffset = (patchOffset - colOffset * m_colStride) / m_fastRowStride;
388 const Index planeOffset = patchOffset - colOffset * m_colStride - rowOffset * m_rowStride;
453 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride,
454 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride};
471 patchOffsets[0] - colOffsets[0] * m_colStride - rowOffsets[0] * m_rowStride,
[all …]
DTensorImagePatch.h282 m_colStride = m_dimensions[1];
283 m_patchStride = m_colStride * m_dimensions[2] * m_dimensions[0];
286 m_colStride = m_dimensions[NumDims-2];
287 m_patchStride = m_colStride * m_dimensions[NumDims-3] * m_dimensions[NumDims-1];
299 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride);
347 const Index rowOffset = patchOffset - colOffset * m_colStride;
399 …owOffsets[2] = {patchOffsets[0] - colOffsets[0]*m_colStride, patchOffsets[1] - colOffsets[1]*m_col…
463 Index m_colStride;