Home
last modified time | relevance | path

Searched refs:patch_planes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions_test.cc37 const int patch_planes = 2; in TEST() local
40 const int output_planes = input_planes - patch_planes + 1; in TEST()
45 Tensor<float, 5> kernel(output_depth, input_depth, patch_planes, patch_rows, in TEST()
69 for (int p = 0; p < patch_planes; ++p) { in TEST()
100 const int patch_planes = 2; in TEST() local
103 const int output_planes = input_planes - patch_planes + 1; in TEST()
108 Tensor<float, 5, RowMajor> kernel(patch_cols, patch_rows, patch_planes, in TEST()
132 for (int p = 0; p < patch_planes; ++p) { in TEST()
163 const int patch_planes = 4; in TEST() local
171 Tensor<float, 5> kernel(output_depth, input_depth, patch_planes, patch_rows, in TEST()
[all …]
Deigen_pooling_test.cc132 const int patch_planes = 2; in TEST() local
147 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride, in TEST()
162 for (int p = 0; p < patch_planes; ++p) { in TEST()
192 const int patch_planes = 2; in TEST() local
207 result = CuboidMaxPooling(input, patch_planes, patch_rows, patch_cols, stride, in TEST()
222 for (int p = 0; p < patch_planes; ++p) { in TEST()
252 const int patch_planes = 2; in TEST() local
267 result = CuboidAvgPooling(input, patch_planes, patch_rows, patch_cols, stride, in TEST()
283 for (int p = 0; p < patch_planes; ++p) { in TEST()
314 const int patch_planes = 2; in TEST() local
[all …]
Deigen_volume_patch.h92 op.patch_planes() + (op.patch_planes() - 1) * (m_in_plane_strides - 1); in CustomTensorEvaluator()
167 m_dimensions[1] = op.patch_planes(); in CustomTensorEvaluator()
183 m_dimensions[NumDims - 2] = op.patch_planes(); in CustomTensorEvaluator()
/external/tensorflow/tensorflow/core/kernels/image/
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, \
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h63 …GEN_STRONG_INLINE TensorVolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex pat…
68 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
75 …GEN_STRONG_INLINE TensorVolumePatchOp(const XprType& expr, DenseIndex patch_planes, DenseIndex pat…
83 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
92 DenseIndex patch_planes() const { return m_patch_planes; }
225 m_patch_planes_eff = op.patch_planes() + (op.patch_planes() - 1) * (m_in_plane_strides - 1);
277 m_dimensions[1] = op.patch_planes();
293 m_dimensions[NumDims-2] = op.patch_planes();
DTensorBase.h712 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
715 …return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patc…
721 extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols,
727 …return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(derived(), patch_planes, patc…