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.h131 m_colStride = m_patch_rows * m_rowStride; in TensorContractionInputMapper()
132 m_patchStride = m_colStride * m_patch_cols * m_patch_depth; in TensorContractionInputMapper()
194 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride); in TensorContractionInputMapper()
220 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()
367 (patchOffset - colOffset * m_colStride) / m_fastRowStride; in loadCoeff()
375 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeff()
412 (patchOffset - colOffset * m_colStride) / m_fastRowStride; in loadCoeffStandard()
414 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeffStandard()
471 patchOffsets[0] - colOffset * m_colStride - rowOffset * m_rowStride, in loadPartialPacketStandard()
472 patchOffsets[1] - colOffset * m_colStride - rowOffset * m_rowStride}; in loadPartialPacketStandard()
[all …]
Deigen_spatial_convolutions-inl.h129 m_colStride = patch_rows; in TensorContractionInputMapper()
164 m_fastColStride = internal::TensorIntDivisor<Index>(m_colStride); in TensorContractionInputMapper()
181 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()
272 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_colStride; } in patchRows()
301 const Index rowOffset = patchOffset - colOffset * m_colStride; in loadCoeff()
330 const Index rowOffset = patchOffset - colOffset * m_colStride; in loadCoeffStandard()
373 const Index rowOffsets[2] = {patchOffsets[0] - colOffset * m_colStride, in loadPartialPacketStandard()
374 patchOffsets[1] - colOffset * m_colStride}; in loadPartialPacketStandard()
423 ((colOffsets[1] * m_colStride) * m_rowInputStride) - 1; in loadPacketStandardFromTwoColumns()
455 const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0] * m_colStride, in loadPacketStandardFromSingleColumn()
[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;