Searched refs:compute_roi_align_shape (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEROIAlignLayerKernel.cpp | 121 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(compute_roi_align_shape(*input, *rois, pool_inf… in validate_arguments()
151 …const TensorShape output_shape = compute_roi_align_shape(*input->info(), *rois->info(), pool_info); in configure()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLROIAlignLayerKernel.cpp | 59 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(compute_roi_align_shape(*input, *rois, pool_inf… in validate_arguments()
96 …const TensorShape output_shape = compute_roi_align_shape(*input->info(), *rois->info(), pool_info); in configure()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ROIAlignLayerFixture.h | 134 …const TensorShape dst_shape = misc::shape_calculator::compute_roi_align_shape(*(src.info()), *(roi… in compute_target()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 867 inline TensorShape compute_roi_align_shape(const ITensorInfo &input, const ITensorInfo &rois, ROIPo… in compute_roi_align_shape() function
|