Searched refs:rois_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ROIAlignLayerFixture.h | 132 … TensorType rois_tensor = create_tensor<TensorType>(rois_shape, _rois_data_type, 1, rois_qinfo); in compute_target() local
134 …ape = misc::shape_calculator::compute_roi_align_shape(*(src.info()), *(rois_tensor.info()), pool_i… in compute_target()
139 roi_align_layer.configure(&src, &rois_tensor, &dst, pool_info); in compute_target()
142 ARM_COMPUTE_ASSERT(rois_tensor.info()->is_resizable()); in compute_target()
147 rois_tensor.allocator()->allocate(); in compute_target()
151 ARM_COMPUTE_ASSERT(!rois_tensor.info()->is_resizable()); in compute_target()
156 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
174 SimpleTensor<TRois> rois_tensor{ rois_shape, _rois_data_type, 1, rois_qinfo }; in compute_reference()
178 generate_rois(rois_tensor, input_shape, pool_info, rois_shape); in compute_reference()
180 return reference::roi_align_layer(src, rois_tensor, pool_info, output_qinfo); in compute_reference()
|
| D | ROIPoolingLayerFixture.h | 116 … TensorType rois_tensor = create_tensor<TensorType>(rois_shape, _rois_data_type, 1, rois_qinfo); in compute_target() local
124 roi_pool_layer.configure(&src, &rois_tensor, &dst, pool_info); in compute_target()
127 ARM_COMPUTE_ASSERT(rois_tensor.info()->is_resizable()); in compute_target()
132 rois_tensor.allocator()->allocate(); in compute_target()
136 ARM_COMPUTE_ASSERT(!rois_tensor.info()->is_resizable()); in compute_target()
141 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
159 SimpleTensor<uint16_t> rois_tensor{ rois_shape, _rois_data_type, 1, rois_qinfo }; in compute_reference()
163 generate_rois(rois_tensor, input_shape, pool_info, rois_shape); in compute_reference()
165 return reference::roi_pool_layer(src, rois_tensor, pool_info, output_qinfo); in compute_reference()
|