| /external/tensorflow/tensorflow/core/kernels/ |
| 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,
|
| eigen_backward_spatial_convolutions_test.cc | 36 const int patch_rows = 2; local
38 const int output_rows = input_rows - patch_rows + 1;
42 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
61 for (int r = 0; r < patch_rows; ++r) {
85 const int patch_rows = 2; local
87 const int output_rows = input_rows - patch_rows + 1;
92 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
113 for (int r = 0; r < patch_rows; ++r) {
137 const int patch_rows = 2; local
140 const int output_rows = input_rows - patch_rows + 1
200 const int patch_rows = 2; local
263 const int patch_rows = 4; local
311 const int patch_rows = 4; local
362 const int patch_rows = 3; local
433 const int patch_rows = 2; local
505 const int patch_rows = 5; local
560 const int patch_rows = 5; local
616 const int patch_rows = 2; local
683 const int patch_rows = 2; local
746 const int patch_rows = 3; local
822 const int patch_rows = 3; local
901 const int patch_rows = 2; local
951 const int patch_rows = 2; local
1000 const int patch_rows = 3; local
1047 const int patch_rows = 3; local
1097 const int patch_rows = 3; local
1145 const int patch_rows = 3; local
1192 const int patch_rows = 3; local
1244 const int patch_rows = 2; local
1307 const int patch_rows = 2; local
1371 const int patch_rows = 5; local
1426 const int patch_rows = 4; local
1482 const int patch_rows = 4; local
1546 const int patch_rows = 5; local
1613 const int patch_rows = 5; local
1679 const int patch_rows = 5; local
1739 const int patch_rows = 4; local
1802 const int patch_rows = 3; local
1881 const int patch_rows = 3; local
1958 const int patch_rows = 3; local
2035 const int patch_rows = 3; local
[all...] |
| eigen_pooling_test.cc | 33 const int patch_rows = 4; local
46 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride,
59 for (int r = 0; r < patch_rows; ++r) {
81 const int patch_rows = 4; local
95 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride,
108 for (int r = 0; r < patch_rows; ++r) {
131 const int patch_rows = 4; local
148 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride,
164 for (int r = 0; r < patch_rows; ++r) {
191 const int patch_rows = 4 local
251 const int patch_rows = 4; local
313 const int patch_rows = 4; local
375 const int patch_rows = 4; local
453 const int patch_rows = 4; local
530 const int patch_rows = 3; local
578 const int patch_rows = 3; local
629 const int patch_rows = 3; local
689 const int patch_rows = 3; local
[all...] |
| eigen_spatial_convolutions_test.cc | 32 const int patch_rows = 3; local
38 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
56 for (int r = 0; r < patch_rows; ++r) {
77 const int patch_rows = 3; local
83 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
101 for (int r = 0; r < patch_rows; ++r) {
197 const int patch_rows = 4; local
199 const int output_rows = input_rows - patch_rows + 1;
203 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
225 for (int r = 0; r < patch_rows; ++r)
249 const int patch_rows = 4; local
307 const int patch_rows = 4; local
363 const int patch_rows = 3; local
411 const int patch_rows = 1; local
459 const int patch_rows = 3; local
510 const int patch_rows = 3; local
561 const int patch_rows = 3; local
615 const int patch_rows = 3; local
[all...] |
| eigen_spatial_convolutions.h | 72 Index patch_rows; local
76 patch_rows = tensor.impl().dimensions()[1];
82 patch_rows = tensor.impl().dimensions()[NumDims - 2];
89 m_colStride = patch_rows;
681 const Index patch_rows = rhs.patchRows();
685 ceil_div(peeled_k, patch_rows * patch_depth) + startCol,
692 ceil_div(peeled_k - c * patch_rows * patch_depth, patch_depth) +
694 patch_rows);
715 std::min<Index>(peeled_k - c * patch_rows * patch_depth -
[all...] |
| extract_image_patches_op.cc | 142 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \
|
| eigen_volume_patch.h | 93 op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
161 // 2: patch_rows
167 m_dimensions[2] = op.patch_rows();
177 // NumDims-3: patch_rows
183 m_dimensions[NumDims - 3] = op.patch_rows();
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:
|
| conv_ops.cc | 568 const int64 patch_rows = filter.dim_size(0); local
581 (patch_rows - 1) * row_dilation + 1 - in_rows);
696 {{patch_rows, // filter_rows
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorImagePatch.h | 25 * Calling the image patch code with patch_rows and patch_cols is equivalent
26 * to calling the regular patch extraction code with parameters d, patch_rows,
68 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorImagePatchOp(const XprType& expr, DenseIndex patch_rows, DenseIndex patch_cols,
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,
87 : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
96 DenseIndex patch_rows() const { return m_patch_rows; } function in class:Eigen::TensorImagePatchOp
217 m_patch_rows_eff = op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
253 // 1: patch_rows
[all...] |
| 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,
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,
83 : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
94 DenseIndex patch_rows() const { return m_patch_rows; } function in class:Eigen::TensorVolumePatchOp
226 m_patch_rows_eff = op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
272 // 2: patch_rows
[all...] |
| 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,
[all...] |
