Searched defs:spatial_scale (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | roi_align_impl.cu | 71 __device__ void bin_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, co… in bin_box()
121 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignKernel()
176 …ign(const T *x, const T *roi_boxes, int roi_rows, int roi_cols, T *out_data, const T spatial_scale, in ROIAlign()
205 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignGradKernel()
267 …const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, const int heigh… in ROIAlignGrad()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_roi_align_grad_op.py | 28 def __init__(self, xdiff_shape, pooled_height, pooled_width, spatial_scale, sample_num): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_roi_align_grad_op.py | 28 def __init__(self, xdiff_shape, pooled_height, pooled_width, spatial_scale, sample_num): argument
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
| D | roi_align.py | 38 spatial_scale, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | roi_align_utils.h | 81 void bin_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sam… in bin_box()
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 1110 optional float spatial_scale = 3 [default = 1]; field
1605 required float spatial_scale = 1; field
1643 optional float spatial_scale = 3 [default = 1]; field
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 1709 def __init__(self, xdiff_shape, pooled_height, pooled_width, spatial_scale, sample_num=2): argument
|
| D | nn_ops.py | 4474 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num=2, roi_end_mode=1): argument
|