Lines Matching refs:m_dimensions
276 m_dimensions[0] = input_dims[0];
277 m_dimensions[1] = op.patch_planes();
278 m_dimensions[2] = op.patch_rows();
279 m_dimensions[3] = op.patch_cols();
280 m_dimensions[4] = m_outputPlanes * m_outputRows * m_outputCols;
282 m_dimensions[i] = input_dims[i-1];
292 m_dimensions[NumDims-1] = input_dims[NumInputDims-1];
293 m_dimensions[NumDims-2] = op.patch_planes();
294 m_dimensions[NumDims-3] = op.patch_rows();
295 m_dimensions[NumDims-4] = op.patch_cols();
296 m_dimensions[NumDims-5] = m_outputPlanes * m_outputRows * m_outputCols;
298 m_dimensions[i] = input_dims[i];
304 m_rowStride = m_dimensions[1];
305 m_colStride = m_dimensions[2] * m_rowStride;
306 m_patchStride = m_colStride * m_dimensions[3] * m_dimensions[0];
307 m_otherStride = m_patchStride * m_dimensions[4];
309 m_rowStride = m_dimensions[NumDims-2];
310 m_colStride = m_dimensions[NumDims-3] * m_rowStride;
311 m_patchStride = m_colStride * m_dimensions[NumDims-4] * m_dimensions[NumDims-1];
312 m_otherStride = m_patchStride * m_dimensions[NumDims-5];
336 m_fastOutputDepth = internal::TensorIntDivisor<Index>(m_dimensions[0]);
338 m_fastOutputDepth = internal::TensorIntDivisor<Index>(m_dimensions[NumDims-1]);
342 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
397 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
485 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
536 Dimensions m_dimensions;
