Search

Home

Sort by

Searched refs:patch_cols (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A D	extract_image_patches_op.h	`29 int patch_rows, int patch_cols, int stride_rows, 40 .extract_image_patches(patch_cols, patch_rows, stride_cols, 46 .extract_image_patches(patch_cols, patch_rows, stride_cols, 28 operator ()(const Device& d, typename TTypes<T, 4>::ConstTensor input, int patch_rows, int patch_cols, int stride_rows, int stride_cols, int rate_rows, int rate_cols, const Eigen::PaddingType& padding, typename TTypes<T, 4>::Tensor output) argument`
H A D	eigen_backward_spatial_convolutions_test.cc	37 const int patch_cols = 2; local 39 const int output_cols = input_cols - patch_cols + 1; 42 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); 60 for (int c = 0; c < patch_cols; ++c) { 86 const int patch_cols = 2; local 88 const int output_cols = input_cols - patch_cols + 1; 92 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, 112 for (int c = 0; c < patch_cols; ++c) { 138 const int patch_cols = 2; local 141 const int output_cols = input_cols - patch_cols 201 const int patch_cols = 2; local 264 const int patch_cols = 4; local 312 const int patch_cols = 4; local 363 const int patch_cols = 2; local 434 const int patch_cols = 3; local 506 const int patch_cols = 5; local 561 const int patch_cols = 5; local 617 const int patch_cols = 2; local 684 const int patch_cols = 2; local 747 const int patch_cols = 3; local 823 const int patch_cols = 3; local 902 const int patch_cols = 2; local 952 const int patch_cols = 2; local 1001 const int patch_cols = 3; local 1048 const int patch_cols = 3; local 1098 const int patch_cols = 3; local 1146 const int patch_cols = 3; local 1193 const int patch_cols = 3; local 1245 const int patch_cols = 2; local 1308 const int patch_cols = 2; local 1372 const int patch_cols = 5; local 1427 const int patch_cols = 4; local 1483 const int patch_cols = 4; local 1547 const int patch_cols = 5; local 1614 const int patch_cols = 5; local 1680 const int patch_cols = 5; local 1740 const int patch_cols = 4; local 1803 const int patch_cols = 2; local 1882 const int patch_cols = 2; local 1959 const int patch_cols = 2; local 2036 const int patch_cols = 2; local [all...]
H A D	eigen_pooling_test.cc	34 const int patch_cols = 4; local 46 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, 60 for (int c = 0; c < patch_cols; ++c) { 82 const int patch_cols = 4; local 95 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, 109 for (int c = 0; c < patch_cols; ++c) { 132 const int patch_cols = 3; local 148 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride, 165 for (int c = 0; c < patch_cols; ++c) { 192 const int patch_cols local 252 const int patch_cols = 3; local 314 const int patch_cols = 3; local 376 const int patch_cols = 3; local 454 const int patch_cols = 3; local 531 const int patch_cols = 3; local 579 const int patch_cols = 3; local 630 const int patch_cols = 3; local 690 const int patch_cols = 3; local [all...]
H A D	eigen_spatial_convolutions_test.cc	33 const int patch_cols = 4; local 38 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); 55 for (int c = 0; c < patch_cols; ++c) { 78 const int patch_cols = 4; local 83 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, 100 for (int c = 0; c < patch_cols; ++c) { 198 const int patch_cols = 4; local 200 const int output_cols = input_cols - patch_cols + 1; 203 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); 224 for (int c = 0; c < patch_cols; 250 const int patch_cols = 4; local 308 const int patch_cols = 4; local 364 const int patch_cols = 3; local 412 const int patch_cols = 1; local 460 const int patch_cols = 3; local 511 const int patch_cols = 3; local 562 const int patch_cols = 3; local 616 const int patch_cols = 3; local [all...]
H A D	extract_image_patches_op.cc	`142 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \`
H A D	eigen_volume_patch.h	`95 op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1); 162 // 3: patch_cols 168 m_dimensions[3] = op.patch_cols(); 178 // NumDims-4: patch_cols 184 m_dimensions[NumDims - 4] = op.patch_cols(); 463 // 0: depth, 1: patch_planes, 2: patch_rows, 3: patch_cols, 4: number of 466 // 0: batches, 1: number of patches, 2: patch_cols , 3: patch_rows, 4:`
H A D	conv_ops.cc	`569 const int64 patch_cols = filter.dim_size(1); local 584 (patch_cols - 1) * col_dilation + 1 - in_cols); 697 patch_cols}}, // filter_cols`
H A D	eigen_spatial_convolutions.h	`680 const Index patch_cols = rhs.patchCols(); 686 patch_cols);`
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A D	TensorImagePatch.h	25 * Calling the image patch code with patch_rows and patch_cols is equivalent 27 * patch_cols, and 1 for all the additional dimensions. 68 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorImagePatchOp(const XprType& expr, DenseIndex patch_rows, DenseIndex patch_cols, argument 73 : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols), 80 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorImagePatchOp(const XprType& expr, DenseIndex patch_rows, DenseIndex patch_cols, argument 87 : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols), 98 DenseIndex patch_cols() const { return m_patch_cols; } function in class:Eigen::TensorImagePatchOp 218 m_patch_cols_eff = op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1); 254 // 2: patch_cols [all...]
H A D	TensorVolumePatch.h	19 * Calling the volume patch code with patch_planes, patch_rows, and patch_cols 21 * d, patch_planes, patch_rows, patch_cols, and 1 for all the additional 63 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorVolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex patch_rows, DenseIndex patch_cols, argument 68 : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols), 75 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorVolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex patch_rows, DenseIndex patch_cols, argument 83 : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols), 96 DenseIndex patch_cols() const { return m_patch_cols; } function in class:Eigen::TensorVolumePatchOp 227 m_patch_cols_eff = op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1); 273 // 3: patch_cols [all...]
H A D	TensorBase.h	688 extract_image_patches(const Index patch_rows = 1, const Index patch_cols = 1, 692 return TensorImagePatchOp<Dynamic, Dynamic, const Derived>(derived(), patch_rows, patch_cols, row_stride, col_stride, 698 extract_image_patches(const Index patch_rows, const Index patch_cols, argument 705 return TensorImagePatchOp<Dynamic, Dynamic, const Derived>(derived(), patch_rows, patch_cols, row_stride, col_stride, 712 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols, argument 715 return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, plane_stride, row_stride, col_stride, 1, 1, 1, 1, 1, 1, padding_type, padding_value); 721 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols, argument 727 return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, plane_stride, row_stride, col_stride, 1, 1, 1, plane_inflate_stride, row_inflate_stride, col_inflate_stride, padding_top_z, padding_bottom_z, padding_top, padding_bottom, padding_left, padding_right, padding_value);

Completed in 1744 milliseconds