| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | combined_non_max_suppression_impl.cu | 57 __global__ void permute(int q, int num_boxes, int batch_size, T *boxes, float *new_boxes) { in permute()
70 __global__ void boxsort(float *boxes_result, float *new_boxes, int batch_size, int q, int num_boxes… in boxsort()
81 __global__ void presort(int num_classes, int num_boxes, int batch_size, T *scores, float *new_score… in presort()
96 __global__ void Init(int num_classes, int num_boxes, int batch_size, bool *mask) { in Init()
105 int q, int num_boxes, bool *mask) { in nms()
128 __global__ void nmsReducePass(int batch_size, int num_classes, bool *sel, int *index, int num_boxes… in nmsReducePass()
144 __global__ void sizeperclass(int batch_size, int num_classes, bool *sel, int num_boxes, int *index, in sizeperclass()
164 int num_classes, int num_boxes, int q) { in output()
202 … float *new_scores, int batch_size, int num_boxes, float *boxes_result, int q, bool *sel, in CalSort()
221 int num_boxes, int *max_output_size_per_class, float *new_scores, bool *mask, in Calnms()
[all …]
|
| D | crop_and_resize_grad_image_impl.cu | 25 … const int *box_ind, int32_t num_boxes, int32_t batch, in CropAndResizeGradImageForwardKernel()
93 … const int *box_ind, int32_t num_boxes, int32_t batch, in CropAndResizeGradImageForwardKernel()
161 … const int *box_ind, int32_t num_boxes, int32_t batch, in CropAndResizeGradImageForwardKernel()
238 … int32_t num_boxes, int32_t batch, int32_t image_height, int32_t image_width, in CalCropAndResizeGradImage()
|
| D | crop_and_resize_grad_boxes_impl.cu | 24 … const int *box_ind, int32_t num_boxes, int32_t batch, in CropAndResizeGradBoxesForwardKernel()
108 … const float *boxes, const int *box_ind, int32_t num_boxes, in CropAndResizeGradBoxesForwardKernel()
202 … const int *box_ind, int32_t num_boxes, int32_t batch, int32_t image_height, in CalCropAndResizeGradBoxes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | detection_post_process_fp32.c | 39 int DecodeBoxes(int num_boxes, const float *input_boxes, const float *anchors, in DecodeBoxes()
66 int NmsSingleClass(const int num_boxes, const float *decoded_boxes, const int max_detections, const… in NmsSingleClass()
108 int NmsMultiClassesFastCore(const int num_boxes, const int num_classes_with_bg, const float *input_… in NmsMultiClassesFastCore()
132 int DetectionPostProcessFast(const int num_boxes, const int num_classes_with_bg, const float *input… in DetectionPostProcessFast()
170 int DetectionPostProcessRegular(const int num_boxes, const int num_classes_with_bg, const float *in… in DetectionPostProcessRegular()
|
| D | resize_fp32.c | 591 int CheckCropAndResizeBoxIdx(int32_t *box_idx, int32_t num_boxes, int32_t batch) { in CheckCropAndResizeBoxIdx()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | combined_non_max_suppression.h | 67 int num_boxes; variable
|
| D | crop_and_resize.cc | 199 auto num_boxes = out_shape_[0]; in DoCompute() local
|
| /third_party/mesa3d/src/gallium/frontends/xa/ |
| D | xa_yuv.c | 118 unsigned int num_boxes, in xa_yuv_planar_blit()
|
| D | xa_context.c | 102 int to_surface, struct xa_box *boxes, unsigned int num_boxes) in xa_surface_dma()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_vmap/ |
| D | vmap_image_ops.py | 153 def get_box_indices_offsets(axis_size, batch_size, num_boxes): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | crop_and_resize.cc | 86 auto num_boxes = ParseNumBoxes(box_shape, box_index_shape, prim_name); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | wider_face_op.cc | 143 int32_t num_boxes = 0, box_counter = 0; in GetTraValAnno() local
|
| /third_party/mesa3d/src/gallium/drivers/zink/ |
| D | zink_resource.c | 2768 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[level], struct pipe_box); in zink_resource_copy_box_intersects() local
2806 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[level], struct pipe_box); in zink_resource_copy_box_add() local
2978 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[0], struct pipe_box); in zink_resource_copies_reset() local
|
| /third_party/mesa3d/src/intel/compiler/elk/ |
| D | elk_fs_combine_constants.cpp | 897 unsigned num_boxes; member
|
| /third_party/mesa3d/src/intel/compiler/ |
| D | brw_opt_combine_constants.cpp | 851 unsigned num_boxes; member
|