Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc36 const int patch_cols = 2; in TEST() local
38 const int output_cols = input_cols - patch_cols + 1; in TEST()
41 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
59 for (int c = 0; c < patch_cols; ++c) { in TEST()
85 const int patch_cols = 2; in TEST() local
87 const int output_cols = input_cols - patch_cols + 1; in TEST()
91 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
111 for (int c = 0; c < patch_cols; ++c) { in TEST()
137 const int patch_cols = 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.cc36 const int patch_cols = 2; in TEST() local
39 const int output_cols = input_cols - patch_cols + 1; in TEST()
46 patch_cols); in TEST()
67 for (int c = 0; c < patch_cols; ++c) { in TEST()
99 const int patch_cols = 2; in TEST() local
102 const int output_cols = input_cols - patch_cols + 1; in TEST()
108 Tensor<float, 5, RowMajor> kernel(patch_cols, patch_rows, patch_planes, in TEST()
130 for (int c = 0; c < patch_cols; ++c) { in TEST()
162 const int patch_cols = 2; in TEST() local
172 patch_cols); in TEST()
[all …]
Deigen_pooling_test.cc33 const int patch_cols = 4; in TEST() local
45 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, in TEST()
59 for (int c = 0; c < patch_cols; ++c) { in TEST()
81 const int patch_cols = 4; in TEST() local
94 result = SpatialMaxPooling(input, patch_rows, patch_cols, stride, stride, in TEST()
108 for (int c = 0; c < patch_cols; ++c) { in TEST()
131 const int patch_cols = 3; in TEST() local
147 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride, in TEST()
164 for (int c = 0; c < patch_cols; ++c) { in TEST()
191 const int patch_cols = 3; in TEST() local
[all …]
Deigen_spatial_convolutions_test.cc35 const int patch_cols = 4; in TEST() local
40 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
57 for (int c = 0; c < patch_cols; ++c) { in TEST()
80 const int patch_cols = 4; in TEST() local
85 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
102 for (int c = 0; c < patch_cols; ++c) { in TEST()
200 const int patch_cols = 4; in TEST() local
202 const int output_cols = input_cols - patch_cols + 1; in TEST()
205 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
226 for (int c = 0; c < patch_cols; ++c) { in TEST()
[all …]
Deigen_volume_patch.h96 op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1); in CustomTensorEvaluator()
169 m_dimensions[3] = op.patch_cols(); in CustomTensorEvaluator()
185 m_dimensions[NumDims - 4] = op.patch_cols(); in CustomTensorEvaluator()
Dconv_ops.cc671 const int64 patch_cols = filter.dim_size(1); in operator ()() local
678 if (patch_rows == 1 && patch_cols == 1 && !is_grouped_convolution && in operator ()()
705 } else if (patch_rows == in_rows && patch_cols == in_cols && in operator ()()
712 const uint64 k = patch_rows * patch_cols * patch_depths; in operator ()()
778 status = GetWindowedOutputSizeVerboseV2(in_cols, patch_cols, col_dilation, in operator ()()
893 .set_input_filter_width(patch_cols) in operator ()()
974 patch_cols, // filter_cols in operator ()()
Dconv_ops_fused_impl.h466 const int64 patch_cols = filter.dim_size(1);
580 .set_input_filter_width(patch_cols)
653 patch_cols, // filter_cols
/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.h30 int patch_planes, int patch_rows, int patch_cols, in operator()
40 .extract_volume_patches(patch_cols, patch_rows, patch_planes, in operator()
47 .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.h68 …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 …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),
98 DenseIndex patch_cols() const { return m_patch_cols; }
218 m_patch_cols_eff = op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1);
259 m_dimensions[2] = op.patch_cols();
273 m_dimensions[NumDims-3] = op.patch_cols();
DTensorVolumePatch.h63 …PatchOp(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 …PatchOp(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),
96 DenseIndex patch_cols() const { return m_patch_cols; }
227 m_patch_cols_eff = op.patch_cols() + (op.patch_cols() - 1) * (m_in_col_strides - 1);
279 m_dimensions[3] = op.patch_cols();
295 m_dimensions[NumDims-4] = op.patch_cols();
DTensorBase.h688 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_…
698 extract_image_patches(const Index patch_rows, const Index patch_cols, in extract_image_patches() argument
705 …return TensorImagePatchOp<Dynamic, Dynamic, const Derived>(derived(), patch_rows, patch_cols, row_… in extract_image_patches()
712 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
715 … Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, plane_stride, r…
721 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
727 … Dynamic, Dynamic, const Derived>(derived(), patch_planes, patch_rows, patch_cols, plane_stride, r…
DREADME.md1647 ### <Operation> extract_image_patches(const Index patch_rows, const Index patch_cols,