Home
last modified time | relevance | path

Searched refs:spatial_scale (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DROIPoolingLayer.cpp54 const float spatial_scale = pool_info.spatial_scale(); in roi_pool_layer() local
82 const int roi_anchor_x = support::cpp11::round(x1 * spatial_scale); in roi_pool_layer()
83 const int roi_anchor_y = support::cpp11::round(y1 * spatial_scale); in roi_pool_layer()
84 … const int roi_width = std::max(support::cpp11::round((x2 - x1) * spatial_scale), 1.f); in roi_pool_layer()
85 … const int roi_height = std::max(support::cpp11::round((y2 - y1) * spatial_scale), 1.f); in roi_pool_layer()
H A DROIAlignLayer.cpp167 const float roi_anchor_x = x1 * pool_info.spatial_scale(); in roi_align_layer()
168 const float roi_anchor_y = y1 * pool_info.spatial_scale(); in roi_align_layer()
169 const float roi_dims_x = std::max((x2 - x1) * pool_info.spatial_scale(), 1.0f); in roi_align_layer()
170 const float roi_dims_y = std::max((y2 - y1) * pool_info.spatial_scale(), 1.0f); in roi_align_layer()
H A DComputeAllAnchors.cpp45 const float stride = 1. / info.spatial_scale(); in compute_all_anchors()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEROIPoolingLayerKernel.cpp165 const float spatial_scale = _pool_info.spatial_scale(); in run() local
179 const int roi_anchor_x = support::cpp11::round(x1 * spatial_scale); in run()
180 const int roi_anchor_y = support::cpp11::round(y1 * spatial_scale); in run()
181 const int roi_width = std::max(support::cpp11::round((x2 - x1) * spatial_scale), 1.f); in run()
182 const int roi_height = std::max(support::cpp11::round((y2 - y1) * spatial_scale), 1.f); in run()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/
H A DTypes.h1410 …ROIPoolingLayerInfo(unsigned int pooled_width, unsigned int pooled_height, float spatial_scale, un…
1411 …: _pooled_width(pooled_width), _pooled_height(pooled_height), _spatial_scale(spatial_scale), _samp… in _pooled_width()
1425 float spatial_scale() const in spatial_scale() function
1458 …GenerateProposalsInfo(float im_width, float im_height, float im_scale, float spatial_scale = 1.0, …
1460 …(im_height), _im_width(im_width), _im_scale(im_scale), _spatial_scale(spatial_scale), _pre_nms_top… in _im_height()
1501 float spatial_scale() const in spatial_scale() function
1534 …ComputeAnchorsInfo(float feat_width, float feat_height, float spatial_scale, size_t values_per_roi…
1537 _spatial_scale(spatial_scale), in _feat_height()
1555 float spatial_scale() const in spatial_scale() function
/aosp_15_r20/external/pytorch/docs/cpp/source/notes/
H A Dinference_mode.rst99 double spatial_scale,
104 ctx->saved_data["spatial_scale"] = spatial_scale;
114 input, rois, spatial_scale, pooled_height,
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/roialign/generic/neon/
H A Dimpl.cpp257 const float roi_anchor_x = x1 * pool_info.spatial_scale(); in roi_align()
258 const float roi_anchor_y = y1 * pool_info.spatial_scale(); in roi_align()
259 const float roi_dims_x = std::max((x2 - x1) * pool_info.spatial_scale(), 1.0f); in roi_align()
260 const float roi_dims_y = std::max((y2 - y1) * pool_info.spatial_scale(), 1.0f); in roi_align()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/genproposals/generic/neon/
H A Dimpl.cpp38 const T stride = 1.f / anchors_info.spatial_scale(); in compute_all_anchors()
71 const float stride = 1.f / anchors_info.spatial_scale(); in compute_all_anchors_qasymm16()
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEGenerateProposalsLayer.cpp96 …nfigure(anchors, &_all_anchors, ComputeAnchorsInfo(feat_width, feat_height, info.spatial_scale())); in configure()
240 …te(anchors, &all_anchors_info, ComputeAnchorsInfo(feat_width, feat_height, info.spatial_scale()))); in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLGenerateProposalsLayer.cpp105 …ontext, anchors, &_all_anchors, ComputeAnchorsInfo(feat_width, feat_height, info.spatial_scale())); in configure()
238 …te(anchors, &all_anchors_info, ComputeAnchorsInfo(feat_width, feat_height, info.spatial_scale()))); in validate()
/aosp_15_r20/external/pytorch/test/onnx/expect/
H A DTestOperators.test_c2_op.expect15 name: "spatial_scale"
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLGenerateProposalsLayerKernel.cpp106 …ild_opts.add_option("-DSTRIDE=" + float_to_string_with_full_precision(1.f / info.spatial_scale())); in configure()
H A DCLROIPoolingLayerKernel.cpp107 … build_opts.add_option("-DSPATIAL_SCALE=" + support::cpp11::to_string(pool_info.spatial_scale())); in configure()
H A DCLROIAlignLayerKernel.cpp119 …ts.add_option("-DSPATIAL_SCALE=" + float_to_string_with_full_precision(pool_info.spatial_scale())); in configure()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DROIPoolingLayerFixture.h73 const float roi_scale = pool_info.spatial_scale();
H A DROIAlignLayerFixture.h74 const float roi_scale = pool_info.spatial_scale();
/aosp_15_r20/external/ComputeLibrary/utils/
H A DTypePrinter.h360 …< pool_info.pooled_width() << "x" << pool_info.pooled_height() << "~" << pool_info.spatial_scale();
513 …ors_info.feat_width() << "x" << anchors_info.feat_height() << ")~" << anchors_info.spatial_scale();