Searched refs:patch_dims (Results 1 – 4 of 4) sorted by relevance
21 array<ptrdiff_t, 4> patch_dims; in test_simple_patch() local23 patch_dims[0] = 1; in test_simple_patch()24 patch_dims[1] = 1; in test_simple_patch()25 patch_dims[2] = 1; in test_simple_patch()26 patch_dims[3] = 1; in test_simple_patch()29 no_patch = tensor.extract_patches(patch_dims); in test_simple_patch()49 patch_dims[0] = 2; in test_simple_patch()50 patch_dims[1] = 3; in test_simple_patch()51 patch_dims[2] = 5; in test_simple_patch()52 patch_dims[3] = 7; in test_simple_patch()[all …]
63 …IGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorPatchOp(const XprType& expr, const PatchDim& patch_dims)64 : m_xpr(expr), m_patch_dims(patch_dims) {}67 const PatchDim& patch_dims() const { return m_patch_dims; }106 const PatchDim& patch_dims = op.patch_dims();109 m_dimensions[i] = patch_dims[i];110 num_patches *= (input_dims[i] - patch_dims[i] + 1);118 m_patchStrides[i] = m_patchStrides[i-1] * (input_dims[i-1] - patch_dims[i-1] + 1);126 m_dimensions[i+1] = patch_dims[i];127 num_patches *= (input_dims[i] - patch_dims[i] + 1);135 m_patchStrides[i] = m_patchStrides[i+1] * (input_dims[i+1] - patch_dims[i+1] + 1);
1560 ### <Operation> extract_patches(const PatchDims& patch_dims)1563 each patch is of dimension specified by 'patch_dims'. The returned tensor has1586 Eigen::array<ptrdiff_t, 2> patch_dims;1587 patch_dims[0] = 2;1588 patch_dims[1] = 2;1589 patch = tensor.extract_patches(patch_dims);
682 extract_patches(const PatchDims& patch_dims) const { in extract_patches() argument683 return TensorPatchOp<const PatchDims, const Derived>(derived(), patch_dims); in extract_patches()