Searched refs:roi_pool_layer (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ROIPoolingLayerFixture.h | 123 FunctionType roi_pool_layer; in compute_target() local
124 roi_pool_layer.configure(&src, &rois_tensor, &dst, pool_info); in compute_target()
144 roi_pool_layer.run(); in compute_target()
165 return reference::roi_pool_layer(src, rois_tensor, pool_info, output_qinfo); in compute_reference()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | ROIPoolingLayer.cpp | 40 SimpleTensor<float> roi_pool_layer(const SimpleTensor<float> &src, const SimpleTensor<uint16_t> &ro… in roi_pool_layer() function
136 SimpleTensor<uint8_t> roi_pool_layer(const SimpleTensor<uint8_t> &src, const SimpleTensor<uint16_t>… in roi_pool_layer() function
139 … SimpleTensor<float> dst_tmp = roi_pool_layer<float>(src_tmp, rois, pool_info, output_qinfo); in roi_pool_layer()
|
| D | ROIPoolingLayer.h | 40 SimpleTensor<T> roi_pool_layer(const SimpleTensor<T> &src, const SimpleTensor<uint16_t> &rois, cons…
|