Home
last modified time | relevance | path

Searched refs:pooled_size (Results 1 – 25 of 70) sorted by relevance

123

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/x86/
H A Droi_align.py29 def roi_align_nchw_ir(data, rois, w_pc, pos_pc, pooled_size, spatial_scale, sample_ratio): argument
66 pooled_size_h = pooled_size[0]
67 pooled_size_w = pooled_size[1]
189 def roi_align_nchw_cpu(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
216 if not isinstance(pooled_size, (tuple, list)):
217 pooled_size = (pooled_size, pooled_size)
224 max_roi_bin_grid_h = math.ceil(height / pooled_size[0])
225 max_roi_bin_grid_w = math.ceil(width / pooled_size[1])
227 * pooled_size[0] * pooled_size[1], 4)
231 pooled_size = tvm.convert(pooled_size)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/
H A Droi_align.py28 def roi_align_nchw_ir(data, rois, num_rois, w_pc, pos_pc, pooled_size, spatial_scale, sample_ratio): argument
68 pooled_size_h = pooled_size[0]
69 pooled_size_w = pooled_size[1]
204 def roi_align_nchw(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
231 if not isinstance(pooled_size, (tuple, list)):
232 pooled_size = (pooled_size, pooled_size)
239 max_roi_bin_grid_h = math.ceil(height / pooled_size[0])
240 max_roi_bin_grid_w = math.ceil(width / pooled_size[1])
244 max_roi_bin_grid_h * max_roi_bin_grid_w * pooled_size[0] * pooled_size[1],
250 pooled_size = tvm.runtime.convert(pooled_size)
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/x86/
H A Droi_align.py29 def roi_align_nchw_ir(data, rois, w_pc, pos_pc, pooled_size, spatial_scale, sample_ratio): argument
66 pooled_size_h = pooled_size[0]
67 pooled_size_w = pooled_size[1]
189 def roi_align_nchw_cpu(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
216 if not isinstance(pooled_size, (tuple, list)):
217 pooled_size = (pooled_size, pooled_size)
224 max_roi_bin_grid_h = math.ceil(height / pooled_size[0])
225 max_roi_bin_grid_w = math.ceil(width / pooled_size[1])
227 * pooled_size[0] * pooled_size[1], 4)
231 pooled_size = tvm.convert(pooled_size)
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/testing/
H A Droi_align_python.py22 def roi_align_nchw_python(a_np, rois_np, pooled_size, spatial_scale, sample_ratio): argument
26 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
28 if isinstance(pooled_size, int):
29 pooled_size_h = pooled_size_w = pooled_size
31 pooled_size_h, pooled_size_w = pooled_size
64 roi_bin_grid_h = int(math.ceil(roi_h / pooled_size))
65 roi_bin_grid_w = int(math.ceil(roi_w / pooled_size))
H A Droi_pool_python.py22 def roi_pool_nchw_python(a_np, rois_np, pooled_size, spatial_scale): argument
26 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
28 if isinstance(pooled_size, int):
29 pooled_size_h = pooled_size_w = pooled_size
31 pooled_size_h, pooled_size_w = pooled_size
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/testing/
H A Droi_align_python.py23 def roi_align_nchw_python(a_np, rois_np, pooled_size, spatial_scale, sample_ratio): argument
27 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
29 if isinstance(pooled_size, int):
30 pooled_size_h = pooled_size_w = pooled_size
32 pooled_size_h, pooled_size_w = pooled_size
67 roi_bin_grid_h = int(math.ceil(roi_h / pooled_size))
68 roi_bin_grid_w = int(math.ceil(roi_w / pooled_size))
H A Droi_pool_python.py23 def roi_pool_nchw_python(a_np, rois_np, pooled_size, spatial_scale): argument
27 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
29 if isinstance(pooled_size, int):
30 pooled_size_h = pooled_size_w = pooled_size
32 pooled_size_h, pooled_size_w = pooled_size
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/testing/
H A Droi_align_python.py22 def roi_align_nchw_python(a_np, rois_np, pooled_size, spatial_scale, sample_ratio): argument
26 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
28 if isinstance(pooled_size, int):
29 pooled_size_h = pooled_size_w = pooled_size
31 pooled_size_h, pooled_size_w = pooled_size
64 roi_bin_grid_h = int(math.ceil(roi_h / pooled_size))
65 roi_bin_grid_w = int(math.ceil(roi_w / pooled_size))
H A Droi_pool_python.py22 def roi_pool_nchw_python(a_np, rois_np, pooled_size, spatial_scale): argument
26 b_np = np.zeros((num_roi, channel, pooled_size, pooled_size), dtype=a_np.dtype)
28 if isinstance(pooled_size, int):
29 pooled_size_h = pooled_size_w = pooled_size
31 pooled_size_h, pooled_size_w = pooled_size
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/vision/
H A Drcnn_op.cc51 oshape = {rshape[0], dshape[1], roi_align_attrs->pooled_size[0], in ROIAlignRel()
52 roi_align_attrs->pooled_size[1]}; in ROIAlignRel()
55 oshape = {rshape[0], roi_align_attrs->pooled_size[0], roi_align_attrs->pooled_size[1], in ROIAlignRel()
78 Expr MakeROIAlign(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIAlign() argument
81 attrs->pooled_size = pooled_size; in MakeROIAlign()
125 {rshape[0], dshape[1], roi_pool_attrs->pooled_size[0], roi_pool_attrs->pooled_size[1]}); in ROIPoolRel()
130 Expr MakeROIPool(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIPool() argument
133 attrs->pooled_size = pooled_size; in MakeROIPool()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/op/vision/
H A Drcnn_op.cc47 {rshape[0], dshape[1], roi_align_attrs->pooled_size[0], roi_align_attrs->pooled_size[1]}); in ROIAlignRel()
52 Expr MakeROIAlign(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIAlign() argument
55 attrs->pooled_size = pooled_size; in MakeROIAlign()
98 {rshape[0], dshape[1], roi_pool_attrs->pooled_size[0], roi_pool_attrs->pooled_size[1]}); in ROIPoolRel()
103 Expr MakeROIPool(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIPool() argument
106 attrs->pooled_size = pooled_size; in MakeROIPool()
/dports/misc/tvm/incubator-tvm-0.6.1/src/relay/op/vision/
H A Drcnn_op.cc47 {rshape[0], dshape[1], roi_align_attrs->pooled_size[0], roi_align_attrs->pooled_size[1]}); in ROIAlignRel()
52 Expr MakeROIAlign(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIAlign() argument
55 attrs->pooled_size = pooled_size; in MakeROIAlign()
98 {rshape[0], dshape[1], roi_pool_attrs->pooled_size[0], roi_pool_attrs->pooled_size[1]}); in ROIPoolRel()
103 Expr MakeROIPool(Expr data, Expr rois, Array<IndexExpr> pooled_size, double spatial_scale, in MakeROIPool() argument
106 attrs->pooled_size = pooled_size; in MakeROIPool()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/vision/
H A Drcnn.py21 def roi_align(data, rois, pooled_size, spatial_scale, sample_ratio=-1, layout="NCHW"): argument
48 return _make.roi_align(data, rois, pooled_size, spatial_scale, sample_ratio, layout)
51 def roi_pool(data, rois, pooled_size, spatial_scale, layout="NCHW"): argument
75 return _make.roi_pool(data, rois, pooled_size, spatial_scale, layout)
H A D_vision.py89 def _roi_align_shape_func(data_shape, rois_shape, pooled_size): argument
93 out[2] = int64(pooled_size[0])
94 out[3] = int64(pooled_size[1])
100 return [_roi_align_shape_func(inputs[0], inputs[1], convert(attrs.pooled_size))]
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/vision/
H A Drcnn.py21 def roi_align(data, rois, pooled_size, spatial_scale, sample_ratio=-1, layout='NCHW'): argument
48 return _make.roi_align(data, rois, pooled_size, spatial_scale, sample_ratio, layout)
51 def roi_pool(data, rois, pooled_size, spatial_scale, layout='NCHW'): argument
75 return _make.roi_pool(data, rois, pooled_size, spatial_scale, layout)
H A D_rcnn.py30 inputs[0], inputs[1], pooled_size=get_const_tuple(attrs.pooled_size),
46 inputs[0], inputs[1], pooled_size=get_const_tuple(attrs.pooled_size),
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/vision/
H A Drcnn.py21 def roi_align(data, rois, pooled_size, spatial_scale, sample_ratio=-1, layout='NCHW'): argument
48 return _make.roi_align(data, rois, pooled_size, spatial_scale, sample_ratio, layout)
51 def roi_pool(data, rois, pooled_size, spatial_scale, layout='NCHW'): argument
75 return _make.roi_pool(data, rois, pooled_size, spatial_scale, layout)
H A D_rcnn.py30 inputs[0], inputs[1], pooled_size=get_const_tuple(attrs.pooled_size),
46 inputs[0], inputs[1], pooled_size=get_const_tuple(attrs.pooled_size),
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/
H A Ddeformable_psroi_pooling-inl.h39 index_t pooled_size; member
50 DMLC_DECLARE_FIELD(pooled_size).describe("fix pooled size"); in DMLC_DECLARE_PARAMETER()
100 param_.spatial_scale, param_.output_dim, param_.group_size, param_.pooled_size, in Forward()
155 param_.pooled_size, param_.part_size, param_.sample_per_part, param_.trans_std); in Backward()
193 param_.part_size = param_.pooled_size; in Init()
227 Shape4(bshape[0], param_.output_dim, param_.pooled_size, param_.pooled_size)); in InferShape()
229 Shape4(bshape[0], param_.output_dim, param_.pooled_size, param_.pooled_size)); in InferShape()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/
H A Ddeformable_psroi_pooling-inl.h39 index_t pooled_size; member
50 DMLC_DECLARE_FIELD(pooled_size).describe("fix pooled size"); in DMLC_DECLARE_PARAMETER()
100 param_.spatial_scale, param_.output_dim, param_.group_size, param_.pooled_size, in Forward()
155 param_.pooled_size, param_.part_size, param_.sample_per_part, param_.trans_std); in Backward()
193 param_.part_size = param_.pooled_size; in Init()
227 Shape4(bshape[0], param_.output_dim, param_.pooled_size, param_.pooled_size)); in InferShape()
229 Shape4(bshape[0], param_.output_dim, param_.pooled_size, param_.pooled_size)); in InferShape()
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/
H A Dtest_op_level5.py353 def verify_roi_align(data_shape, rois_shape, pooled_size, spatial_scale, sample_ratio): argument
356 z = relay.vision.roi_align(data, rois, pooled_size=(pooled_size, pooled_size),
363 (num_roi, channel, pooled_size, pooled_size), "float32")
370 ref_res = topi.testing.roi_align_nchw_python(np_data, np_rois, pooled_size=pooled_size,
382 verify_roi_align((4, 4, 16, 16), (32, 5), pooled_size=7, spatial_scale=0.5, sample_ratio=2)
386 def verify_roi_pool(data_shape, rois_shape, pooled_size, spatial_scale): argument
389 z = relay.vision.roi_pool(data, rois, pooled_size=(pooled_size, pooled_size),
395 (num_roi, channel, pooled_size, pooled_size), "float32")
402 ref_res = topi.testing.roi_pool_nchw_python(np_data, np_rois, pooled_size=pooled_size,
412 verify_roi_pool((1, 4, 16, 16), (32, 5), pooled_size=7, spatial_scale=1.0)
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/
H A Dtest_op_level5.py353 def verify_roi_align(data_shape, rois_shape, pooled_size, spatial_scale, sample_ratio): argument
356 z = relay.vision.roi_align(data, rois, pooled_size=(pooled_size, pooled_size),
363 (num_roi, channel, pooled_size, pooled_size), "float32")
370 ref_res = topi.testing.roi_align_nchw_python(np_data, np_rois, pooled_size=pooled_size,
382 verify_roi_align((4, 4, 16, 16), (32, 5), pooled_size=7, spatial_scale=0.5, sample_ratio=2)
386 def verify_roi_pool(data_shape, rois_shape, pooled_size, spatial_scale): argument
389 z = relay.vision.roi_pool(data, rois, pooled_size=(pooled_size, pooled_size),
395 (num_roi, channel, pooled_size, pooled_size), "float32")
402 ref_res = topi.testing.roi_pool_nchw_python(np_data, np_rois, pooled_size=pooled_size,
412 verify_roi_pool((1, 4, 16, 16), (32, 5), pooled_size=7, spatial_scale=1.0)
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/vision/rcnn/
H A Droi_align.py25 def roi_align_nchw(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
56 if isinstance(pooled_size, int):
57 pooled_size_h = pooled_size_w = pooled_size
59 pooled_size_h, pooled_size_w = pooled_size
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/vision/rcnn/
H A Droi_align.py25 def roi_align_nchw(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
56 if isinstance(pooled_size, int):
57 pooled_size_h = pooled_size_w = pooled_size
59 pooled_size_h, pooled_size_w = pooled_size
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/vision/rcnn/
H A Droi_align.py25 def roi_align_nchw(data, rois, pooled_size, spatial_scale, sample_ratio=-1): argument
56 if isinstance(pooled_size, int):
57 pooled_size_h = pooled_size_w = pooled_size
59 pooled_size_h, pooled_size_w = pooled_size

123