/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/x86/ |
H A D | roi_align.py | 29 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 D | roi_align.py | 28 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 D | roi_align.py | 29 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 D | roi_align_python.py | 22 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 D | roi_pool_python.py | 22 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 D | roi_align_python.py | 23 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 D | roi_pool_python.py | 23 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 D | roi_align_python.py | 22 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 D | roi_pool_python.py | 22 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 D | rcnn_op.cc | 51 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 D | rcnn_op.cc | 47 {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 D | rcnn_op.cc | 47 {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 D | rcnn.py | 21 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.py | 89 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 D | rcnn.py | 21 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.py | 30 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 D | rcnn.py | 21 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.py | 30 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 D | deformable_psroi_pooling-inl.h | 39 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 D | deformable_psroi_pooling-inl.h | 39 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 D | test_op_level5.py | 353 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 D | test_op_level5.py | 353 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 D | roi_align.py | 25 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 D | roi_align.py | 25 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 D | roi_align.py | 25 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
|