| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_p_s_r_o_i_pooling_grad.py | 28 def __init__(self, input_size, spatial_scale, group_size, output_dim): argument
37 def test_net(x_shape, rois_shape, input_size, spatial_scale, group_size, output_dim): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | roi_align_grad.cc | 103 void ROIAlignGrad::set_spatial_scale(const float spatial_scale) { in set_spatial_scale()
115 …AlignGrad::Init(const int64_t pooled_height, const int64_t pooled_width, const float spatial_scale, in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | psroi_pooling_impl.cu | 33 __global__ void PSROIPoolForward(const int nthreads, const T *input, const T spatial_scale, const i… in PSROIPoolForward()
89 cudaError_t PSROIPoolForwardLauncher(const T *input, const T spatial_scale, const int rois_number, in PSROIPoolForwardLauncher()
116 … const T spatial_scale, const int feature_height, const int feature_width, in PSROIPoolBackward()
167 … const int rois_number, const T spatial_scale, const int feature_channels, in PSROIPoolBackwardLauncher()
|
| D | ps_roi_pooling_v2_impl.cu | 25 __global__ void PSROIPoolForwardV2(const int nthreads, const T *input, const T spatial_scale, const… in PSROIPoolForwardV2()
82 …obal__ void PSROIPoolForwardV2(const int nthreads, const double *input, const double spatial_scale, in PSROIPoolForwardV2()
143 …global__ void PSROIPoolForwardV2(const int nthreads, const float *input, const float spatial_scale, in PSROIPoolForwardV2()
202 __global__ void PSROIPoolForwardV2(const int nthreads, const half *input, const half spatial_scale, in PSROIPoolForwardV2()
260 cudaError_t PSROIPoolForwardV2Launcher(const T *input, const T spatial_scale, const int output_n, in PSROIPoolForwardV2Launcher()
|
| D | roi_align_impl.cu | 70 __device__ void bin_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, co… in bin_box()
120 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignKernel()
175 …ign(const T *x, const T *roi_boxes, int roi_rows, int roi_cols, T *out_data, const T spatial_scale, in ROIAlign()
208 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignGradKernel()
270 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignGrad()
|
| D | psroi_pooling_v2_impl.cu | 33 __global__ void PSROIPoolBackwardV2(const int nthreads, T *input_diff, const T spatial_scale, const… in PSROIPoolBackwardV2()
96 …lBackwardV2Launcher(T *input_diff, const int batch_size, const int output_n, const T spatial_scale, in PSROIPoolBackwardV2Launcher()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | roi_align.cc | 131 void ROIAlign::set_spatial_scale(const float spatial_scale) { in set_spatial_scale()
156 …ROIAlign::Init(const int64_t pooled_height, const int64_t pooled_weight, const float spatial_scale, in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/caffe/ |
| D | caffe_psroi_pooling_parser.cc | 40 float spatial_scale = psroi_pooling_param.spatial_scale(); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/onnx/ |
| D | onnx_roi_align_parser.cc | 43 float spatial_scale = 1.0; in Parse() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_roi_align.py | 23 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, roi_end_mode): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_ps_roi_pooling_grad_v2_op.py | 30 def __init__(self, input_size, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
|
| D | test_ps_roi_pooling_v2_op.py | 36 def __init__(self, spatial_scale, group_size, output_dim): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_roi_align_grad_op.py | 26 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, is_dyn_rank=False): argument
|
| D | test_roi_align_op.py | 26 …def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, roi_end_mode, is_dyn_ra… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | roi_align_gpu_kernel.cc | 95 T spatial_scale = static_cast<T>(spatial_scale_); in LaunchKernel() local
|
| D | roi_align_grad_gpu_kernel.cc | 121 T spatial_scale = static_cast<T>(spatial_scale_); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_roi_align.py | 46 def __init__(self, pooled_h, pooled_w, spatial_scale, strategy=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | roialign_tensorrt.h | 41 …it ROIAlignPlugin(const std::string name, int pooled_height, int pooled_width, float spatial_scale, in ROIAlignPlugin()
|
| D | roialign_tensorrt.cc | 49 float spatial_scale = op->get_spatial_scale(); in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_ps_roi_pooling_v2_op.py | 38 def __init__(self, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
|
| D | test_ps_roi_pooling_grad_op.py | 35 def __init__(self, input_size, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | roi_align_cpu_kernel.cc | 75 void bin_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sam… in bin_box()
214 const T spatial_scale = static_cast<T>(spatial_scale_); in LaunchKernel() local
|
| D | roi_align_grad_cpu_kernel.cc | 79 void bin_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sam… in bin_box()
250 const T spatial_scale = static_cast<T>(spatial_scale_); in LaunchKernel() local
|
| D | ps_roi_pooling_v2_cpu_kernel.cc | 50 auto spatial_scale = static_cast<T>(spatial_scale_); in PSROIPoolForward() local
|
| D | ps_roi_pooling_grad_cpu_kernel.cc | 49 auto spatial_scale = static_cast<T>(spatial_scale_); in PSROIPoolBackward() local
|