Home
last modified time | relevance | path

Searched refs:patch_rows (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc35 const int patch_rows = 2; in TEST() local
37 const int output_rows = input_rows - patch_rows + 1; in TEST()
41 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
60 for (int r = 0; r < patch_rows; ++r) { in TEST()
84 const int patch_rows = 2; in TEST() local
86 const int output_rows = input_rows - patch_rows + 1; in TEST()
91 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
112 for (int r = 0; r < patch_rows; ++r) { in TEST()
136 const int patch_rows = 4; in TEST() local
142 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
[all …]
Deigen_backward_cuboid_convolutions_test.cc35 const int patch_rows = 2; in TEST() local
38 const int output_rows = input_rows - patch_rows + 1; in TEST()
45 Tensor<float, 5> kernel(output_depth, input_depth, patch_planes, patch_rows, in TEST()
68 for (int r = 0; r < patch_rows; ++r) { in TEST()
98 const int patch_rows = 2; in TEST() local
101 const int output_rows = input_rows - patch_rows + 1; in TEST()
108 Tensor<float, 5, RowMajor> kernel(patch_cols, patch_rows, patch_planes, in TEST()
131 for (int r = 0; r < patch_rows; ++r) { in TEST()
161 const int patch_rows = 3; in TEST() local
171 Tensor<float, 5> kernel(output_depth, input_depth, patch_planes, patch_rows, in TEST()
[all …]
Deigen_pooling_test.cc32 const int patch_rows = 4; in TEST() local
45 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, in TEST()
58 for (int r = 0; r < patch_rows; ++r) { in TEST()
80 const int patch_rows = 4; in TEST() local
94 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, in TEST()
107 for (int r = 0; r < patch_rows; ++r) { in TEST()
130 const int patch_rows = 4; in TEST() local
147 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride, in TEST()
163 for (int r = 0; r < patch_rows; ++r) { in TEST()
190 const int patch_rows = 4; in TEST() local
[all …]
Deigen_spatial_convolutions_test.cc34 const int patch_rows = 3; in TEST() local
40 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
58 for (int r = 0; r < patch_rows; ++r) { in TEST()
79 const int patch_rows = 3; in TEST() local
85 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
103 for (int r = 0; r < patch_rows; ++r) { in TEST()
199 const int patch_rows = 4; in TEST() local
201 const int output_rows = input_rows - patch_rows + 1; in TEST()
205 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
227 for (int r = 0; r < patch_rows; ++r) { in TEST()
[all …]
Dconv_ops.cc800 const int64_t patch_rows = filter.dim_size(0); in operator ()() local
813 if (patch_rows == 1 && patch_cols == 1 && !is_grouped_convolution && in operator ()()
833 } else if (patch_rows == in_rows && patch_cols == in_cols && in operator ()()
840 const uint64 k = patch_rows * patch_cols * patch_depths; in operator ()()
894 in_rows, patch_rows, row_dilation, row_stride, padding, &out_rows_check, in operator ()()
1014 filter_desc.set_input_filter_height(patch_rows) in operator ()()
1095 {{patch_rows, // filter_rows in operator ()()
Dconv_ops_fused_impl.h472 const int64_t patch_rows = filter.dim_size(0);
586 filter_desc.set_input_filter_height(patch_rows)
659 {{patch_rows, // filter_rows
Deigen_spatial_convolutions-inl.h108 Index patch_rows; in TensorContractionInputMapper() local
112 patch_rows = tensor.impl().dimensions()[1]; in TensorContractionInputMapper()
118 patch_rows = tensor.impl().dimensions()[NumDims - 2]; in TensorContractionInputMapper()
125 m_patch_col_stride = patch_rows * m_patch_row_stride; in TensorContractionInputMapper()
130 m_colStride = patch_rows; in TensorContractionInputMapper()
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_image_patches_op.h29 int patch_rows, int patch_cols, int stride_rows, in operator()
40 .extract_image_patches(patch_cols, patch_rows, stride_cols, in operator()
46 .extract_image_patches(patch_cols, patch_rows, stride_cols, in operator()
Dextract_volume_patches_op.h29 int patch_planes, int patch_rows, int patch_cols, in operator()
39 .extract_volume_patches(patch_cols, patch_rows, patch_planes, in operator()
46 .extract_volume_patches(patch_cols, patch_rows, patch_planes, in operator()
Dextract_volume_patches_op.cc174 int patch_planes, int patch_rows, int patch_cols, \
Dextract_image_patches_op.cc146 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h130 …EIGEN_STRONG_INLINE TensorImagePatchOp(const XprType& expr, DenseIndex patch_rows, DenseIndex patc…
135 … : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
142 …EIGEN_STRONG_INLINE TensorImagePatchOp(const XprType& expr, DenseIndex patch_rows, DenseIndex patc…
149 … : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
159 DenseIndex patch_rows() const { return m_patch_rows; }
288 m_patch_rows_eff = op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
335 m_dimensions[1] = op.patch_rows();
349 m_dimensions[NumDims-2] = op.patch_rows();
DTensorVolumePatch.h66 …VolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex patch_rows, DenseIndex patc…
71 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
78 …VolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex patch_rows, DenseIndex patc…
86 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
97 DenseIndex patch_rows() const { return m_patch_rows; }
236 m_patch_rows_eff = op.patch_rows() + (op.patch_rows() - 1) * (m_in_row_strides - 1);
288 m_dimensions[2] = op.patch_rows();
304 m_dimensions[NumDims-3] = op.patch_rows();
DTensorBase.h833 extract_image_patches(const Index patch_rows = 1, const Index patch_cols = 1,
837 …return TensorImagePatchOp<Dynamic, Dynamic, const Derived>(derived(), patch_rows, patch_cols, row_…
843 extract_image_patches(const Index patch_rows, const Index patch_cols, in extract_image_patches() argument
850 …return TensorImagePatchOp<Dynamic, Dynamic, const Derived>(derived(), patch_rows, patch_cols, row_… in extract_image_patches()
857 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
860 …hOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, pla…
866 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
872 …hOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, pla…
DREADME.md1707 ### <Operation> extract_image_patches(const Index patch_rows, const Index patch_cols, const Index …