Home
last modified time | relevance | path

Searched refs:m_patchStride (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h283 m_patchStride = m_colStride * m_dimensions[2] * m_dimensions[0];
284 m_otherStride = m_patchStride * m_dimensions[3];
287 m_patchStride = m_colStride * m_dimensions[NumDims-3] * m_dimensions[NumDims-1];
288 m_otherStride = m_patchStride * m_dimensions[NumDims-4];
298 m_fastPatchStride = internal::TensorIntDivisor<Index>(m_patchStride);
329 const Index patchOffset = (index - patchIndex * m_patchStride) / m_fastOutputDepth;
381 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
382 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth};
462 Index m_patchStride;
DTensorVolumePatch.h306 m_patchStride = m_colStride * m_dimensions[3] * m_dimensions[0];
307 m_otherStride = m_patchStride * m_dimensions[4];
311 m_patchStride = m_colStride * m_dimensions[NumDims-4] * m_dimensions[NumDims-1];
312 m_otherStride = m_patchStride * m_dimensions[NumDims-5];
325 m_fastPatchStride = internal::TensorIntDivisor<Index>(m_patchStride);
360 const Index patchOffset = (index - patchIndex * m_patchStride) / m_fastOutputDepth;
428 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
429 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth};
578 Index m_patchStride;
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h196 m_patchStride = m_colStride * m_dimensions[3] * m_dimensions[0]; in CustomTensorEvaluator()
197 m_otherStride = m_patchStride * m_dimensions[4]; in CustomTensorEvaluator()
201 m_patchStride = in CustomTensorEvaluator()
203 m_otherStride = m_patchStride * m_dimensions[NumDims - 5]; in CustomTensorEvaluator()
217 m_fastPatchStride = internal::TensorIntDivisor<Index>(m_patchStride); in CustomTensorEvaluator()
259 (index - patchIndex * m_patchStride) / m_fastOutputDepth; in coeff()
355 (indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth, in packet()
356 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth}; in packet()
608 Index m_patchStride; member
Deigen_cuboid_convolution.h130 m_patchStride = m_colStride * m_patch_cols * m_patch_depth; in TensorContractionInputMapper()
131 m_otherStride = m_patchStride * m_num_patches; in TensorContractionInputMapper()
219 m_patchStride = base_mapper.m_patchStride; in TensorContractionInputMapper()
620 Index m_patchStride; variable