| /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,
|
| extract_volume_patches_op.h | 30 int patch_planes, int patch_rows, int patch_cols,
40 .extract_volume_patches(patch_cols, patch_rows, patch_planes,
47 .extract_volume_patches(patch_cols, patch_rows, patch_planes,
|
| eigen_backward_spatial_convolutions_test.cc | 35 const int patch_rows = 2; local
37 const int output_rows = input_rows - patch_rows + 1;
41 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
60 for (int r = 0; r < patch_rows; ++r) {
84 const int patch_rows = 2; local
86 const int output_rows = input_rows - patch_rows + 1;
91 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
112 for (int r = 0; r < patch_rows; ++r) {
136 const int patch_rows = 2; local
139 const int output_rows = input_rows - patch_rows + 1
199 const int patch_rows = 2; local
262 const int patch_rows = 4; local
310 const int patch_rows = 4; local
361 const int patch_rows = 3; local
432 const int patch_rows = 2; local
504 const int patch_rows = 5; local
559 const int patch_rows = 5; local
615 const int patch_rows = 2; local
682 const int patch_rows = 2; local
745 const int patch_rows = 3; local
821 const int patch_rows = 3; local
900 const int patch_rows = 2; local
950 const int patch_rows = 2; local
999 const int patch_rows = 3; local
1046 const int patch_rows = 3; local
1096 const int patch_rows = 3; local
1144 const int patch_rows = 3; local
1191 const int patch_rows = 3; local
1243 const int patch_rows = 2; local
1309 const int patch_rows = 2; local
1375 const int patch_rows = 5; local
1430 const int patch_rows = 4; local
1486 const int patch_rows = 4; local
1550 const int patch_rows = 5; local
1617 const int patch_rows = 5; local
1683 const int patch_rows = 5; local
1743 const int patch_rows = 4; local
1806 const int patch_rows = 3; local
1885 const int patch_rows = 3; local
1962 const int patch_rows = 3; local
2039 const int patch_rows = 3; local
[all...] |
| eigen_pooling_test.cc | 32 const int patch_rows = 4; local
45 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride,
58 for (int r = 0; r < patch_rows; ++r) {
80 const int patch_rows = 4; local
94 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride,
107 for (int r = 0; r < patch_rows; ++r) {
130 const int patch_rows = 4; local
147 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride,
163 for (int r = 0; r < patch_rows; ++r) {
190 const int patch_rows = 4 local
250 const int patch_rows = 4; local
312 const int patch_rows = 4; local
374 const int patch_rows = 4; local
452 const int patch_rows = 4; local
529 const int patch_rows = 3; local
577 const int patch_rows = 3; local
628 const int patch_rows = 3; local
688 const int patch_rows = 3; local
[all...] |
| eigen_spatial_convolutions_test.cc | 34 const int patch_rows = 3; local
40 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
58 for (int r = 0; r < patch_rows; ++r) {
79 const int patch_rows = 3; local
85 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
103 for (int r = 0; r < patch_rows; ++r) {
199 const int patch_rows = 4; local
201 const int output_rows = input_rows - patch_rows + 1;
205 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
227 for (int r = 0; r < patch_rows; ++r)
251 const int patch_rows = 4; local
309 const int patch_rows = 4; local
365 const int patch_rows = 3; local
413 const int patch_rows = 1; local
461 const int patch_rows = 3; local
512 const int patch_rows = 3; local
563 const int patch_rows = 3; local
617 const int patch_rows = 3; local
1506 const Index patch_rows = evaluators[0].impl().dimensions()[1]; local
[all...] |
| extract_image_patches_op.cc | 142 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \
|
| extract_volume_patches_op.cc | 171 int patch_planes, int patch_rows, int patch_cols, \
|
| conv_ops.cc | 602 const int64 patch_rows = filter.dim_size(0); local
610 if (patch_rows == 1 && patch_cols == 1 && !is_grouped_convolution &&
637 } else if (patch_rows == in_rows && patch_cols == in_cols &&
644 const uint64 k = patch_rows * patch_cols * patch_depths;
681 in_rows, patch_rows, row_dilation, row_stride, padding, &out_rows_check,
[all...] |
| conv_ops_fused_impl.h | 586 const int64 patch_rows = filter.dim_size(0); local
617 (patch_rows - 1) * dimensions.dilation_rows + 1 - in_rows);
685 filter_desc.set_input_filter_height(patch_rows)
759 {{patch_rows, // filter_rows
[all...] |
| eigen_volume_patch.h | 94 op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
162 // 2: patch_rows
168 m_dimensions[2] = op.patch_rows();
178 // NumDims-3: patch_rows
184 m_dimensions[NumDims - 3] = op.patch_rows();
464 // 0: depth, 1: patch_planes, 2: patch_rows, 3: patch_cols, 4: number of
467 // 0: batches, 1: number of patches, 2: patch_cols , 3: patch_rows, 4:
|
| eigen_spatial_convolutions-inl.h | 103 Index patch_rows; local
107 patch_rows = tensor.impl().dimensions()[1];
113 patch_rows = tensor.impl().dimensions()[NumDims - 2];
120 m_patch_col_stride = patch_rows * m_patch_row_stride;
125 m_colStride = patch_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...] |
