Searched defs:pooled_width (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | roi_align_impl.cu | 73 … const int pooled_height, const int pooled_width, int *offset, int *n, int *c, int *ph, int *pw, in bin_box()
122 … const int height, const int width, const int pooled_height, const int pooled_width) { in ROIAlignKernel()
178 const int pooled_height, const int pooled_width, cudaStream_t cuda_stream) { in ROIAlign()
206 … const int height, const int width, const int pooled_height, const int pooled_width) { in ROIAlignGradKernel()
268 … const int width, const int pooled_height, const int pooled_width, cudaStream_t cuda_stream) { 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/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | roi_pooling_fp32.c | 42 int pooled_width = param->pooledW_; in ROIPooling() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | roi_align_utils.h | 83 … const int pooled_width, int *offset, int *n, int *c, int *ph, int *pw, int *roi_bin_grid_h, in bin_box()
|
| /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
|