Home
last modified time | relevance | path

Searched refs:ROIPoolingLayerInfo (Results 1 – 25 of 44) sorted by relevance

12

/external/ComputeLibrary/tests/datasets/
DROIDataset.h41 using type = std::tuple<TensorShape, ROIPoolingLayerInfo, TensorShape>;
46 std::vector<ROIPoolingLayerInfo>::const_iterator infos_it, in iterator()
79 std::vector<ROIPoolingLayerInfo>::const_iterator _infos_it;
93 void add_config(TensorShape tensor_shape, ROIPoolingLayerInfo info, TensorShape rois_shape) in add_config()
106 std::vector<ROIPoolingLayerInfo> _infos{};
115 …add_config(TensorShape(50U, 47U, 1U, 1U), ROIPoolingLayerInfo(7U, 7U, 1.f / 4.f), TensorShape(5U, … in SmallROIDataset()
116 …add_config(TensorShape(50U, 47U, 3U, 4U), ROIPoolingLayerInfo(7U, 7U, 1.f / 4.f), TensorShape(5U, … in SmallROIDataset()
117 …add_config(TensorShape(50U, 47U, 3U, 1U), ROIPoolingLayerInfo(7U, 7U, 1.f / 4.f), TensorShape(5U, … in SmallROIDataset()
118 …add_config(TensorShape(50U, 47U, 10U, 1U), ROIPoolingLayerInfo(7U, 7U, 1.f / 4.f), TensorShape(5U,… in SmallROIDataset()
121 …add_config(TensorShape(50U, 47U, 80U, 4U), ROIPoolingLayerInfo(7U, 7U, 1.f / 4.f), TensorShape(5U,… in SmallROIDataset()
[all …]
/external/ComputeLibrary/tests/validation/NEON/
DROIPoolingLayer.cpp85 framework::dataset::make("PoolInfo", { ROIPoolingLayerInfo(7U, 7U, 1./8),
86 ROIPoolingLayerInfo(7U, 7U, 1./8),
87 ROIPoolingLayerInfo(7U, 7U, 1./8),
88 ROIPoolingLayerInfo(7U, 7U, 1./8),
89 ROIPoolingLayerInfo(7U, 7U, 1./8),
90 ROIPoolingLayerInfo(7U, 7U, 1./8),
91 ROIPoolingLayerInfo(7U, 7U, 1./8),
92 ROIPoolingLayerInfo(7U, 7U, 1./8),
DROIAlignLayer.cpp90 framework::dataset::make("PoolInfo", { ROIPoolingLayerInfo(7U, 7U, 1./8),
91 ROIPoolingLayerInfo(7U, 7U, 1./8),
92 ROIPoolingLayerInfo(7U, 7U, 1./8),
93 ROIPoolingLayerInfo(7U, 7U, 1./8),
94 ROIPoolingLayerInfo(7U, 7U, 1./8),
95 ROIPoolingLayerInfo(7U, 7U, 1./8),
96 ROIPoolingLayerInfo(7U, 7U, 1./8),
/external/ComputeLibrary/tests/validation/CL/
DROIPoolingLayer.cpp83 framework::dataset::make("PoolInfo", { ROIPoolingLayerInfo(7U, 7U, 1./8),
84 ROIPoolingLayerInfo(7U, 7U, 1./8),
85 ROIPoolingLayerInfo(7U, 7U, 1./8),
86 ROIPoolingLayerInfo(7U, 7U, 1./8),
87 ROIPoolingLayerInfo(7U, 7U, 1./8),
88 ROIPoolingLayerInfo(7U, 7U, 1./8),
89 ROIPoolingLayerInfo(7U, 7U, 1./8),
90 ROIPoolingLayerInfo(7U, 7U, 1./8),
DROIAlignLayer.cpp90 framework::dataset::make("PoolInfo", { ROIPoolingLayerInfo(7U, 7U, 1./8),
91 ROIPoolingLayerInfo(7U, 7U, 1./8),
92 ROIPoolingLayerInfo(7U, 7U, 1./8),
93 ROIPoolingLayerInfo(7U, 7U, 1./8),
94 ROIPoolingLayerInfo(7U, 7U, 1./8),
95 ROIPoolingLayerInfo(7U, 7U, 1./8),
96 ROIPoolingLayerInfo(7U, 7U, 1./8),
97 ROIPoolingLayerInfo(7U, 7U, 1./8),
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLROIAlignLayer.h34 class ROIPoolingLayerInfo; variable
71 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
87 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
103 …orInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
DCLROIPoolingLayer.h34 class ROIPoolingLayerInfo; variable
69 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
84 …nsor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
100 …orInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
/external/ComputeLibrary/src/core/CL/kernels/
DCLROIAlignLayerKernel.h64 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
80 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
96 …orInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
105 ROIPoolingLayerInfo _pool_info;
DCLROIPoolingLayerKernel.h62 … ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
77 …nsor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info);
95 … *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
101 ROIPoolingLayerInfo _pool_info;
DCLROIAlignLayerKernel.cpp45 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate_arguments()
85 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
90 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
147 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
DCLROIPoolingLayerKernel.cpp50 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
74 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
79 …ensor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
/external/ComputeLibrary/arm_compute/graph/nodes/
DROIAlignLayerNode.h43 ROIAlignLayerNode(ROIPoolingLayerInfo &pool_info);
53 const ROIPoolingLayerInfo &pooling_info() const;
62 ROIPoolingLayerInfo _pool_info;
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEROIPoolingLayer.h36 class ROIPoolingLayerInfo; variable
76 … ITensor *input, const ITensor *rois, const ITensor *output, const ROIPoolingLayerInfo &pool_info);
94 … *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
DNEROIAlignLayer.h64 …(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info);
80 …orInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
/external/ComputeLibrary/src/core/NEON/kernels/
DNEROIPoolingLayerKernel.h66 … ITensor *input, const ITensor *rois, const ITensor *output, const ROIPoolingLayerInfo &pool_info);
85 … *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
91 ROIPoolingLayerInfo _pool_info;
DNEROIAlignLayerKernel.h70 …(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info);
86 …orInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info);
95 ROIPoolingLayerInfo _pool_info;
DNEROIAlignLayerKernel.cpp52 …r<void(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, …
107 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate_arguments()
146 …e(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
170 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLROIAlignLayer.cpp34 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
41 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
46 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DCLROIPoolingLayer.cpp32 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
37 …t ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
42 …ensor *input, const ICLTensor *rois, const ICLTensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
/external/ComputeLibrary/tests/validation/fixtures/
DROIPoolingLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
63 …void generate_rois(U &&rois, const TensorShape &shape, const ROIPoolingLayerInfo &pool_info, Tenso…
107 const ROIPoolingLayerInfo &pool_info, in compute_target()
151 const ROIPoolingLayerInfo &pool_info, in compute_reference()
178 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
191 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
DROIAlignLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
64 …void generate_rois(U &&rois, const TensorShape &shape, const ROIPoolingLayerInfo &pool_info, Tenso…
118 const ROIPoolingLayerInfo &pool_info, in compute_target()
166 const ROIPoolingLayerInfo &pool_info, in compute_reference()
193 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
205 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEROIAlignLayer.cpp32 …sorInfo *input, const ITensorInfo *rois, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
39 …e(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
DNEROIPoolingLayer.cpp39 …o *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) in validate()
44 …t ITensor *input, const ITensor *rois, const ITensor *output, const ROIPoolingLayerInfo &pool_info) in configure()
/external/ComputeLibrary/tests/validation/reference/
DROIAlignLayer.cpp140 …t SimpleTensor<float> &src, const SimpleTensor<float> &rois, const ROIPoolingLayerInfo &pool_info,… in roi_align_layer()
213 …nst SimpleTensor<half> &src, const SimpleTensor<half> &rois, const ROIPoolingLayerInfo &pool_info,… in roi_align_layer()
223 …pleTensor<uint8_t> &src, const SimpleTensor<uint16_t> &rois, const ROIPoolingLayerInfo &pool_info,… in roi_align_layer()
232 …mpleTensor<int8_t> &src, const SimpleTensor<uint16_t> &rois, const ROIPoolingLayerInfo &pool_info,… in roi_align_layer()
/external/ComputeLibrary/src/graph/nodes/
DROIAlignLayerNode.cpp36 ROIAlignLayerNode::ROIAlignLayerNode(ROIPoolingLayerInfo &pool_info) in ROIAlignLayerNode()
43 const ROIPoolingLayerInfo &ROIAlignLayerNode::pooling_info() const in pooling_info()

12