| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | roi_align_grad.cc | 97 void ROIAlignGrad::set_pooled_width(const int64_t pooled_width) { in set_pooled_width()
115 void ROIAlignGrad::Init(const int64_t pooled_height, const int64_t pooled_width, const float spatia… in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | psroi_pooling_impl.cu | 35 … const int pooled_width, const int group_size, const int output_channels, in PSROIPoolForward()
91 … const int pooled_height, const int pooled_width, const T *roi_boxes, in PSROIPoolForwardLauncher()
117 … const int feature_channels, const int pooled_height, const int pooled_width, in PSROIPoolBackward()
168 … const int feature_height, const int feature_width, const int pooled_width, in PSROIPoolBackwardLauncher()
|
| D | ps_roi_pooling_v2_impl.cu | 27 … const int pooled_width, const int group_size, const int output_channels, in PSROIPoolForwardV2()
84 … const int pooled_height, const int pooled_width, const int group_size, in PSROIPoolForwardV2()
145 … const int pooled_height, const int pooled_width, const int group_size, in PSROIPoolForwardV2()
204 … const int pooled_height, const int pooled_width, const int group_size, in PSROIPoolForwardV2()
262 … const int pooled_height, const int pooled_width, const T *roi_boxes, in PSROIPoolForwardV2Launcher()
|
| D | roi_align_impl.cu | 72 … const int pooled_height, const int pooled_width, int *offset, int *n, int *c, int *ph, int *pw, in bin_box()
121 … const int height, const int width, const int pooled_height, const int pooled_width) { in ROIAlignKernel()
177 const int pooled_height, const int pooled_width, const uint32_t &device_id, in ROIAlign()
209 … const int height, const int width, const int pooled_height, const int pooled_width) { in ROIAlignGradKernel()
271 … const int height, const int width, const int pooled_height, const int pooled_width, in ROIAlignGrad()
|
| D | psroi_pooling_v2_impl.cu | 35 … const int pooled_width, const int output_channels, T *output_diff, T *roi_boxes, in PSROIPoolBackwardV2()
98 … const int pooled_width, const int pooled_height, const int output_channels, in PSROIPoolBackwardV2Launcher()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | roialign_tensorrt.cc | 48 int pooled_width = op->get_pooled_width(); in AddInnerOp() local
105 auto pooled_width = exprBuilder.constant(pooled_width_); in getOutputDimensions() local
|
| D | roialign_tensorrt.h | 41 …explicit ROIAlignPlugin(const std::string name, int pooled_height, int pooled_width, float spatial… in ROIAlignPlugin()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | roi_align.cc | 87 auto pooled_width = GetValue<int64_t>(pooled_width_ptr); in InferShape() local
122 void ROIAlign::set_pooled_width(const int64_t pooled_width) { in set_pooled_width()
|
| /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/ |
| 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/cpu/kernel/nnacl/fp32/ |
| D | roi_pooling_fp32.c | 42 int pooled_width = param->pooledW_; in ROIPooling() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | roi_align_info.cc | 228 auto pooled_width = GetValue<int64_t>(attrs_[POOLED_WIDTH]); in InitForCostModel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ps_roi_pooling_v2_cpu_kernel.cc | 47 auto pooled_width = pooled_width_; in PSROIPoolForward() local
|
| D | roi_align_cpu_kernel.cc | 77 … const int pooled_width, int *offset, int *n, int *c, int *ph, int *pw, int *roi_bin_grid_h, in bin_box()
|
| D | roi_align_grad_cpu_kernel.cc | 81 … const int pooled_width, int *offset, int *n, int *c, int *ph, int *pw, int *roi_bin_grid_h, in bin_box()
|
| D | ps_roi_pooling_grad_cpu_kernel.cc | 44 auto pooled_width = pooled_width_; in PSROIPoolBackward() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _grad_ops.py | 1584 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num=2): argument
1604 pooled_height, pooled_width, spatial_scale, out_dim): argument
|
| D | _inner_ops.py | 1415 def __init__(self, pooled_height, pooled_width, num_rois, spatial_scale, out_dim, group_size): argument
|
| D | nn_ops.py | 3585 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num=2, roi_end_mode=1): argument
|