Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
iou3d_cpu.h 367 Bytes
Newer Older
Shaoshuai Shi's avatar
add iou3d_nms cuda ops and roiaware_pool3d cuda ops in the unified normative coordinate  
Shaoshuai Shi committed
1
2
3
4
5
6
7
8
9 
#ifndef IOU3D_CPU_H
#define IOU3D_CPU_H

#include <torch/serialize/tensor.h>
#include <vector>
#include <cuda.h>
#include <cuda_runtime_api.h>

int boxes_iou_bev_cpu(at::Tensor boxes_a_tensor, at::Tensor boxes_b_tensor, at::Tensor ans_iou_tensor);
yukang's avatar
Add support for VoxelNeXt (#1309)  
yukang committed
10 
int boxes_aligned_iou_bev_cpu(at::Tensor boxes_a_tensor, at::Tensor boxes_b_tensor, at::Tensor ans_iou_tensor);
Shaoshuai Shi's avatar
add iou3d_nms cuda ops and roiaware_pool3d cuda ops in the unified normative coordinate  
Shaoshuai Shi committed
11 
#endif