SYMBOL INDEX (2558 symbols across 467 files)

FILE: .dev_scripts/check_installation.py
  function check_installation (line 8) | def check_installation():

FILE: mmcv/arraymisc/quantization.py
  function quantize (line 7) | def quantize(arr: np.ndarray,
  function dequantize (line 38) | def dequantize(arr: np.ndarray,

FILE: mmcv/cnn/alexnet.py
  class AlexNet (line 10) | class AlexNet(nn.Module):
    method __init__ (line 17) | def __init__(self, num_classes: int = -1):
    method init_weights (line 46) | def init_weights(self, pretrained: Optional[str] = None) -> None:
    method forward (line 56) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/activation.py
  class SiLU (line 21) | class SiLU(nn.Module):
    method __init__ (line 24) | def __init__(self, inplace=False):
    method forward (line 28) | def forward(self, inputs) -> torch.Tensor:
  class Clamp (line 39) | class Clamp(nn.Module):
    method __init__ (line 52) | def __init__(self, min: float = -1., max: float = 1.):
    method forward (line 57) | def forward(self, x) -> torch.Tensor:
  class GELU (line 69) | class GELU(nn.Module):
    method forward (line 91) | def forward(self, input: torch.Tensor) -> torch.Tensor:
  function build_activation_layer (line 102) | def build_activation_layer(cfg: Dict) -> nn.Module:

FILE: mmcv/cnn/bricks/context_block.py
  function last_zero_init (line 10) | def last_zero_init(m: Union[nn.Module, nn.Sequential]) -> None:
  class ContextBlock (line 18) | class ContextBlock(nn.Module):
    method __init__ (line 37) | def __init__(self,
    method reset_parameters (line 76) | def reset_parameters(self):
    method spatial_pool (line 86) | def spatial_pool(self, x: torch.Tensor) -> torch.Tensor:
    method forward (line 112) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/conv.py
  function build_conv_layer (line 14) | def build_conv_layer(cfg: Optional[Dict], *args, **kwargs) -> nn.Module:

FILE: mmcv/cnn/bricks/conv2d_adaptive_padding.py
  class Conv2dAdaptivePadding (line 12) | class Conv2dAdaptivePadding(nn.Conv2d):
    method __init__ (line 34) | def __init__(self,
    method forward (line 46) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/conv_module.py
  function efficient_conv_bn_eval_forward (line 18) | def efficient_conv_bn_eval_forward(bn: _BatchNorm,
  class ConvModule (line 69) | class ConvModule(nn.Module):
    method __init__ (line 126) | def __init__(self,
    method norm (line 229) | def norm(self):
    method init_weights (line 235) | def init_weights(self):
    method forward (line 256) | def forward(self,
    method turn_on_efficient_conv_bn_eval (line 289) | def turn_on_efficient_conv_bn_eval(self, efficient_conv_bn_eval=True):
    method create_from_conv_bn (line 300) | def create_from_conv_bn(conv: torch.nn.modules.conv._ConvNd,

FILE: mmcv/cnn/bricks/conv_ws.py
  function conv_ws_2d (line 11) | def conv_ws_2d(input: torch.Tensor,
  class ConvWS2d (line 28) | class ConvWS2d(nn.Conv2d):
    method __init__ (line 30) | def __init__(self,
    method forward (line 51) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class ConvAWS2d (line 57) | class ConvAWS2d(nn.Conv2d):
    method __init__ (line 80) | def __init__(self,
    method _get_weight (line 103) | def _get_weight(self, weight: torch.Tensor) -> torch.Tensor:
    method forward (line 111) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method _load_from_state_dict (line 116) | def _load_from_state_dict(self, state_dict: OrderedDict, prefix: str,

FILE: mmcv/cnn/bricks/depthwise_separable_conv_module.py
  class DepthwiseSeparableConvModule (line 10) | class DepthwiseSeparableConvModule(nn.Module):
    method __init__ (line 51) | def __init__(self,
    method forward (line 96) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/drop.py
  function drop_path (line 9) | def drop_path(x: torch.Tensor,
  class DropPath (line 31) | class DropPath(nn.Module):
    method __init__ (line 42) | def __init__(self, drop_prob: float = 0.1):
    method forward (line 46) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class Dropout (line 51) | class Dropout(nn.Dropout):
    method __init__ (line 62) | def __init__(self, drop_prob: float = 0.5, inplace: bool = False):
  function build_dropout (line 66) | def build_dropout(cfg: Dict, default_args: Optional[Dict] = None) -> Any:

FILE: mmcv/cnn/bricks/generalized_attention.py
  class GeneralizedAttention (line 13) | class GeneralizedAttention(nn.Module):
    method __init__ (line 46) | def __init__(self,
    method get_position_embedding (line 165) | def get_position_embedding(self,
    method forward (line 215) | def forward(self, x_input: torch.Tensor) -> torch.Tensor:
    method init_weights (line 402) | def init_weights(self):

FILE: mmcv/cnn/bricks/hsigmoid.py
  class HSigmoid (line 10) | class HSigmoid(nn.Module):
    method __init__ (line 29) | def __init__(self,
    method forward (line 47) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/hswish.py
  class HSwish (line 9) | class HSwish(nn.Module):
    method __init__ (line 25) | def __init__(self, inplace: bool = False):
    method forward (line 29) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/non_local.py
  class _NonLocalNd (line 13) | class _NonLocalNd(nn.Module, metaclass=ABCMeta):
    method __init__ (line 36) | def __init__(self,
    method init_weights (line 100) | def init_weights(self, std: float = 0.01, zeros_init: bool = True) -> ...
    method gaussian (line 117) | def gaussian(self, theta_x: torch.Tensor,
    method embedded_gaussian (line 126) | def embedded_gaussian(self, theta_x: torch.Tensor,
    method dot_product (line 138) | def dot_product(self, theta_x: torch.Tensor,
    method concatenation (line 147) | def concatenation(self, theta_x: torch.Tensor,
    method forward (line 165) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class NonLocal1d (line 219) | class NonLocal1d(_NonLocalNd):
    method __init__ (line 231) | def __init__(self,
  class NonLocal2d (line 250) | class NonLocal2d(_NonLocalNd):
    method __init__ (line 264) | def __init__(self,
  class NonLocal3d (line 282) | class NonLocal3d(_NonLocalNd):
    method __init__ (line 294) | def __init__(self,

FILE: mmcv/cnn/bricks/norm.py
  function infer_abbr (line 24) | def infer_abbr(class_type):
  function build_norm_layer (line 73) | def build_norm_layer(cfg: Dict,
  function is_norm (line 133) | def is_norm(layer: nn.Module,

FILE: mmcv/cnn/bricks/padding.py
  function build_padding_layer (line 13) | def build_padding_layer(cfg: Dict, *args, **kwargs) -> nn.Module:

FILE: mmcv/cnn/bricks/plugin.py
  function infer_abbr (line 15) | def infer_abbr(class_type: type) -> str:
  function build_plugin_layer (line 58) | def build_plugin_layer(cfg: Dict,

FILE: mmcv/cnn/bricks/scale.py
  class Scale (line 6) | class Scale(nn.Module):
    method __init__ (line 16) | def __init__(self, scale: float = 1.0):
    method forward (line 20) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class LayerScale (line 24) | class LayerScale(nn.Module):
    method __init__ (line 37) | def __init__(self,
    method forward (line 49) | def forward(self, x) -> torch.Tensor:

FILE: mmcv/cnn/bricks/swish.py
  class Swish (line 8) | class Swish(nn.Module):
    method __init__ (line 20) | def __init__(self):
    method forward (line 23) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/bricks/transformer.py
  function build_positional_encoding (line 39) | def build_positional_encoding(cfg, default_args=None):
  function build_attention (line 44) | def build_attention(cfg, default_args=None):
  function build_feedforward_network (line 49) | def build_feedforward_network(cfg, default_args=None):
  function build_transformer_layer (line 54) | def build_transformer_layer(cfg, default_args=None):
  function build_transformer_layer_sequence (line 59) | def build_transformer_layer_sequence(cfg, default_args=None):
  class AdaptivePadding (line 64) | class AdaptivePadding(nn.Module):
    method __init__ (line 99) | def __init__(self, kernel_size=1, stride=1, dilation=1, padding='corne...
    method get_pad_shape (line 112) | def get_pad_shape(self, input_shape):
    method forward (line 133) | def forward(self, x):
  class PatchEmbed (line 154) | class PatchEmbed(BaseModule):
    method __init__ (line 182) | def __init__(self,
    method forward (line 254) | def forward(self, x):
  class PatchMerging (line 278) | class PatchMerging(BaseModule):
    method __init__ (line 310) | def __init__(self,
    method forward (line 359) | def forward(self, x, input_size):
  class MultiheadAttention (line 408) | class MultiheadAttention(BaseModule):
    method __init__ (line 430) | def __init__(self,
    method forward (line 462) | def forward(self,
  class FFN (line 556) | class FFN(BaseModule):
    method __init__ (line 586) | def __init__(self,
    method forward (line 624) | def forward(self, x, identity=None):
  class BaseTransformerLayer (line 639) | class BaseTransformerLayer(BaseModule):
    method __init__ (line 675) | def __init__(self,
    method forward (line 768) | def forward(self,
  class TransformerLayerSequence (line 870) | class TransformerLayerSequence(BaseModule):
    method __init__ (line 889) | def __init__(self, transformerlayers=None, num_layers=None, init_cfg=N...
    method forward (line 905) | def forward(self,

FILE: mmcv/cnn/bricks/upsample.py
  class PixelShufflePack (line 16) | class PixelShufflePack(nn.Module):
    method __init__ (line 30) | def __init__(self, in_channels: int, out_channels: int, scale_factor: ...
    method init_weights (line 44) | def init_weights(self):
    method forward (line 47) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  function build_upsample_layer (line 53) | def build_upsample_layer(cfg: Dict, *args, **kwargs) -> nn.Module:

FILE: mmcv/cnn/bricks/wrappers.py
  function obsolete_torch_version (line 23) | def obsolete_torch_version(torch_version, version_threshold) -> bool:
  class NewEmptyTensorOp (line 27) | class NewEmptyTensorOp(torch.autograd.Function):
    method forward (line 30) | def forward(ctx, x: torch.Tensor, new_shape: tuple) -> torch.Tensor:
    method backward (line 35) | def backward(ctx, grad: torch.Tensor) -> tuple:
  class Conv2d (line 41) | class Conv2d(nn.Conv2d):
    method forward (line 43) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class Conv3d (line 62) | class Conv3d(nn.Conv3d):
    method forward (line 64) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class ConvTranspose2d (line 84) | class ConvTranspose2d(nn.ConvTranspose2d):
    method forward (line 86) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class ConvTranspose3d (line 106) | class ConvTranspose3d(nn.ConvTranspose3d):
    method forward (line 108) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class MaxPool2d (line 126) | class MaxPool2d(nn.MaxPool2d):
    method forward (line 128) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class MaxPool3d (line 144) | class MaxPool3d(nn.MaxPool3d):
    method forward (line 146) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class Linear (line 163) | class Linear(torch.nn.Linear):
    method forward (line 165) | def forward(self, x: torch.Tensor) -> torch.Tensor:

FILE: mmcv/cnn/resnet.py
  function conv3x3 (line 12) | def conv3x3(in_planes: int,
  class BasicBlock (line 27) | class BasicBlock(nn.Module):
    method __init__ (line 30) | def __init__(self,
    method forward (line 50) | def forward(self, x: Tensor) -> Tensor:
  class Bottleneck (line 69) | class Bottleneck(nn.Module):
    method __init__ (line 72) | def __init__(self,
    method forward (line 115) | def forward(self, x: Tensor) -> Tensor:
  function make_res_layer (line 148) | def make_res_layer(block: nn.Module,
  class ResNet (line 186) | class ResNet(nn.Module):
    method __init__ (line 215) | def __init__(self,
    method init_weights (line 271) | def init_weights(self, pretrained: Optional[str] = None) -> None:
    method forward (line 284) | def forward(self, x: Tensor) -> Union[Tensor, Tuple[Tensor]]:
    method train (line 300) | def train(self, mode: bool = True) -> None:

FILE: mmcv/cnn/rfsearch/operator.py
  class BaseConvRFSearchOp (line 14) | class BaseConvRFSearchOp(BaseModule):
    method __init__ (line 22) | def __init__(self, op_layer: nn.Module, global_config: dict):
    method normlize (line 27) | def normlize(self, weights: nn.Parameter) -> nn.Parameter:
  class Conv2dRFSearchOp (line 41) | class Conv2dRFSearchOp(BaseConvRFSearchOp):
    method __init__ (line 63) | def __init__(self,
    method forward (line 88) | def forward(self, input: Tensor) -> Tensor:
    method estimate_rates (line 119) | def estimate_rates(self) -> None:
    method expand_rates (line 147) | def expand_rates(self) -> None:
    method get_padding (line 164) | def get_padding(self, dilation) -> tuple:

FILE: mmcv/cnn/rfsearch/search.py
  class RFSearchHook (line 17) | class RFSearchHook(Hook):
    method __init__ (line 55) | def __init__(self,
    method init_model (line 73) | def init_model(self, model: nn.Module):
    method after_train_epoch (line 99) | def after_train_epoch(self, runner):
    method after_train_iter (line 104) | def after_train_iter(self, runner, batch_idx, data_batch, outputs):
    method step (line 109) | def step(self, model: nn.Module, work_dir: str) -> None:
    method estimate_and_expand (line 134) | def estimate_and_expand(self, model: nn.Module) -> None:
    method wrap_model (line 145) | def wrap_model(self,
    method set_model (line 185) | def set_model(self,

FILE: mmcv/cnn/rfsearch/utils.py
  function write_to_json (line 6) | def write_to_json(config: dict, filename: str):
  function expand_rates (line 18) | def expand_rates(dilation: tuple, config: dict) -> list:
  function get_single_padding (line 64) | def get_single_padding(kernel_size: int,

FILE: mmcv/cnn/utils/flops_counter.py
  function get_model_complexity_info (line 39) | def get_model_complexity_info(model: nn.Module,
  function flops_to_string (line 121) | def flops_to_string(flops: float,
  function params_to_string (line 166) | def params_to_string(num_params: float,
  function print_model_with_flops (line 205) | def print_model_with_flops(model: nn.Module,
  function get_model_parameters_number (line 314) | def get_model_parameters_number(model: nn.Module) -> float:
  function add_flops_counting_methods (line 327) | def add_flops_counting_methods(net_main_module: nn.Module) -> nn.Module:
  function compute_average_flops_cost (line 344) | def compute_average_flops_cost(self) -> Tuple[float, float]:
  function start_flops_count (line 362) | def start_flops_count(self) -> None:
  function stop_flops_count (line 385) | def stop_flops_count(self) -> None:
  function reset_flops_count (line 396) | def reset_flops_count(self) -> None:
  function empty_flops_counter_hook (line 407) | def empty_flops_counter_hook(module: nn.Module, input: tuple,
  function upsample_flops_counter_hook (line 412) | def upsample_flops_counter_hook(module: nn.Module, input: tuple,
  function relu_flops_counter_hook (line 422) | def relu_flops_counter_hook(module: nn.Module, input: tuple,
  function linear_flops_counter_hook (line 428) | def linear_flops_counter_hook(module: nn.Module, input: tuple,
  function pool_flops_counter_hook (line 435) | def pool_flops_counter_hook(module: nn.Module, input: tuple,
  function norm_flops_counter_hook (line 440) | def norm_flops_counter_hook(module: nn.Module, input: tuple,
  function deconv_flops_counter_hook (line 449) | def deconv_flops_counter_hook(conv_module: nn.Module, input: tuple,
  function conv_flops_counter_hook (line 475) | def conv_flops_counter_hook(conv_module: nn.Module, input: tuple,
  function batch_counter_hook (line 505) | def batch_counter_hook(module: nn.Module, input: tuple, output: Any) -> ...
  function add_batch_counter_variables_or_reset (line 516) | def add_batch_counter_variables_or_reset(module: nn.Module) -> None:
  function add_batch_counter_hook_function (line 521) | def add_batch_counter_hook_function(module: nn.Module) -> None:
  function remove_batch_counter_hook_function (line 529) | def remove_batch_counter_hook_function(module: nn.Module) -> None:
  function add_flops_counter_variable_or_reset (line 535) | def add_flops_counter_variable_or_reset(module: nn.Module) -> None:
  function is_supported_instance (line 545) | def is_supported_instance(module: nn.Module) -> bool:
  function remove_flops_counter_hook_function (line 551) | def remove_flops_counter_hook_function(module: nn.Module) -> None:
  function get_modules_mapping (line 558) | def get_modules_mapping() -> Dict:

FILE: mmcv/cnn/utils/fuse_conv_bn.py
  function _fuse_conv_bn (line 6) | def _fuse_conv_bn(conv: nn.Module, bn: nn.Module) -> nn.Module:
  function fuse_conv_bn (line 27) | def fuse_conv_bn(module: nn.Module) -> nn.Module:

FILE: mmcv/cnn/vgg.py
  function conv3x3 (line 11) | def conv3x3(in_planes: int, out_planes: int, dilation: int = 1) -> nn.Mo...
  function make_vgg_layer (line 21) | def make_vgg_layer(inplanes: int,
  class VGG (line 39) | class VGG(nn.Module):
    method __init__ (line 63) | def __init__(self,
    method init_weights (line 127) | def init_weights(self, pretrained: Optional[str] = None) -> None:
    method forward (line 142) | def forward(self, x: Tensor) -> Union[Tensor, Tuple[Tensor, ...]]:
    method train (line 160) | def train(self, mode: bool = True) -> None:

FILE: mmcv/image/colorspace.py
  function imconvert (line 8) | def imconvert(img: np.ndarray, src: str, dst: str) -> np.ndarray:
  function bgr2gray (line 24) | def bgr2gray(img: np.ndarray, keepdim: bool = False) -> np.ndarray:
  function rgb2gray (line 41) | def rgb2gray(img: np.ndarray, keepdim: bool = False) -> np.ndarray:
  function gray2bgr (line 58) | def gray2bgr(img: np.ndarray) -> np.ndarray:
  function gray2rgb (line 72) | def gray2rgb(img: np.ndarray) -> np.ndarray:
  function _convert_input_type_range (line 86) | def _convert_input_type_range(img: np.ndarray) -> np.ndarray:
  function _convert_output_type_range (line 114) | def _convert_output_type_range(
  function rgb2ycbcr (line 146) | def rgb2ycbcr(img: np.ndarray, y_only: bool = False) -> np.ndarray:
  function bgr2ycbcr (line 180) | def bgr2ycbcr(img: np.ndarray, y_only: bool = False) -> np.ndarray:
  function ycbcr2rgb (line 214) | def ycbcr2rgb(img: np.ndarray) -> np.ndarray:
  function ycbcr2bgr (line 246) | def ycbcr2bgr(img: np.ndarray) -> np.ndarray:
  function convert_color_factory (line 278) | def convert_color_factory(src: str, dst: str) -> Callable:

FILE: mmcv/image/geometric.py
  function _scale_size (line 17) | def _scale_size(
  function imresize (line 77) | def imresize(
  function imresize_to_multiple (line 127) | def imresize_to_multiple(
  function imresize_like (line 195) | def imresize_like(
  function rescale_size (line 219) | def rescale_size(old_size: tuple,
  function imrescale (line 258) | def imrescale(
  function imflip (line 291) | def imflip(img: np.ndarray, direction: str = 'horizontal') -> np.ndarray:
  function imflip_ (line 311) | def imflip_(img: np.ndarray, direction: str = 'horizontal') -> np.ndarray:
  function imrotate (line 331) | def imrotate(img: np.ndarray,
  function bbox_clip (line 385) | def bbox_clip(bboxes: np.ndarray, img_shape: Tuple[int, int]) -> np.ndar...
  function bbox_scaling (line 403) | def bbox_scaling(bboxes: np.ndarray,
  function imcrop (line 431) | def imcrop(
  function impad (line 492) | def impad(img: np.ndarray,
  function impad_to_multiple (line 576) | def impad_to_multiple(img: np.ndarray,
  function cutout (line 594) | def cutout(img: np.ndarray,
  function _get_shear_matrix (line 651) | def _get_shear_matrix(magnitude: Union[int, float],
  function imshear (line 670) | def imshear(img: np.ndarray,
  function _get_translate_matrix (line 721) | def _get_translate_matrix(offset: Union[int, float],
  function imtranslate (line 740) | def imtranslate(img: np.ndarray,

FILE: mmcv/image/io.py
  function use_backend (line 45) | def use_backend(backend: str) -> None:
  function _jpegflag (line 71) | def _jpegflag(flag: str = 'color', channel_order: str = 'bgr'):
  function _pillow2array (line 87) | def _pillow2array(img,
  function imread (line 144) | def imread(img_or_path: Union[np.ndarray, str, Path],
  function imfrombytes (line 232) | def imfrombytes(content: bytes,
  function imwrite (line 291) | def imwrite(img: np.ndarray,

FILE: mmcv/image/misc.py
  function tensor2imgs (line 14) | def tensor2imgs(tensor,

FILE: mmcv/image/photometric.py
  function imnormalize (line 14) | def imnormalize(img, mean, std, to_rgb=True):
  function imnormalize_ (line 30) | def imnormalize_(img, mean, std, to_rgb=True):
  function imdenormalize (line 53) | def imdenormalize(img, mean, std, to_bgr=True):
  function iminvert (line 64) | def iminvert(img):
  function solarize (line 76) | def solarize(img, thr=128):
  function posterize (line 90) | def posterize(img, bits):
  function adjust_color (line 105) | def adjust_color(img, alpha=1, beta=None, gamma=0, backend=None):
  function imequalize (line 155) | def imequalize(img):
  function adjust_brightness (line 200) | def adjust_brightness(img, factor=1., backend=None):
  function adjust_contrast (line 250) | def adjust_contrast(img, factor=1., backend=None):
  function auto_contrast (line 298) | def auto_contrast(img, cutoff=0):
  function adjust_sharpness (line 354) | def adjust_sharpness(img, factor=1., kernel=None):
  function adjust_lighting (line 398) | def adjust_lighting(img, eigval, eigvec, alphastd=0.1, to_rgb=True):
  function lut_transform (line 441) | def lut_transform(img, lut_table):
  function clahe (line 465) | def clahe(img, clip_limit=40.0, tile_grid_size=(8, 8)):
  function adjust_hue (line 491) | def adjust_hue(img: np.ndarray,

FILE: mmcv/ops/active_rotated_filter.py
  class ActiveRotatedFilterFunction (line 15) | class ActiveRotatedFilterFunction(Function):
    method forward (line 24) | def forward(ctx, input: torch.Tensor,
    method backward (line 47) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, None]:

FILE: mmcv/ops/assign_score_withk.py
  class AssignScoreWithK (line 12) | class AssignScoreWithK(Function):
    method forward (line 32) | def forward(ctx,
    method backward (line 84) | def backward(

FILE: mmcv/ops/ball_query.py
  class BallQuery (line 13) | class BallQuery(Function):
    method forward (line 17) | def forward(
    method backward (line 83) | def backward(ctx, a=None) -> Tuple[None, None, None, None]:

FILE: mmcv/ops/bbox.py
  function _bbox_overlaps_cpu (line 9) | def _bbox_overlaps_cpu(bboxes1: torch.Tensor,
  function bbox_overlaps (line 50) | def bbox_overlaps(bboxes1: torch.Tensor,

FILE: mmcv/ops/bezier_align.py
  class BezierAlignFunction (line 16) | class BezierAlignFunction(Function):
    method forward (line 19) | def forward(ctx,
    method backward (line 51) | def backward(ctx, grad_output: torch.Tensor):
  class BezierAlign (line 70) | class BezierAlign(nn.Module):
    method __init__ (line 105) | def __init__(
    method forward (line 119) | def forward(self, input: torch.Tensor,
    method __repr__ (line 131) | def __repr__(self):

FILE: mmcv/ops/bias_act.py
  class EasyDict (line 25) | class EasyDict(dict):
    method __getattr__ (line 29) | def __getattr__(self, name: str) -> Any:
    method __setattr__ (line 35) | def __setattr__(self, name: str, value: Any) -> None:
    method __delattr__ (line 38) | def __delattr__(self, name: str) -> None:
  function bias_act (line 195) | def bias_act(input: torch.Tensor,
  function _bias_act_ref (line 262) | def _bias_act_ref(input: torch.Tensor,
  function _bias_act_cuda (line 340) | def _bias_act_cuda(dim: int = 1,
  function _bias_act_musa (line 463) | def _bias_act_musa(dim: int = 1,

FILE: mmcv/ops/border_align.py
  class BorderAlignFunction (line 18) | class BorderAlignFunction(Function):
    method symbolic (line 21) | def symbolic(g, input, boxes, pool_size):
    method forward (line 26) | def forward(ctx, input: torch.Tensor, boxes: torch.Tensor,
    method backward (line 51) | def backward(ctx,
  class BorderAlign (line 69) | class BorderAlign(nn.Module):
    method __init__ (line 92) | def __init__(self, pool_size: int):
    method forward (line 96) | def forward(self, input: torch.Tensor,
    method __repr__ (line 111) | def __repr__(self):

FILE: mmcv/ops/box_iou_quadri.py
  function box_iou_quadri (line 9) | def box_iou_quadri(bboxes1: torch.Tensor,

FILE: mmcv/ops/box_iou_rotated.py
  function box_iou_rotated (line 9) | def box_iou_rotated(bboxes1: torch.Tensor,

FILE: mmcv/ops/carafe.py
  class CARAFENaiveFunction (line 21) | class CARAFENaiveFunction(Function):
    method symbolic (line 24) | def symbolic(g, features: Tensor, masks: Tensor, kernel_size: int,
    method forward (line 35) | def forward(ctx, features: Tensor, masks: Tensor, kernel_size: int,
    method backward (line 65) | def backward(
  class CARAFENaive (line 93) | class CARAFENaive(Module):
    method __init__ (line 95) | def __init__(self, kernel_size: int, group_size: int, scale_factor: int):
    method forward (line 104) | def forward(self, features: Tensor, masks: Tensor) -> Tensor:
  class CARAFEFunction (line 109) | class CARAFEFunction(Function):
    method symbolic (line 112) | def symbolic(g, features: Tensor, masks: Tensor, kernel_size: int,
    method forward (line 123) | def forward(ctx, features: Tensor, masks: Tensor, kernel_size: int,
    method backward (line 159) | def backward(
  class CARAFE (line 192) | class CARAFE(Module):
    method __init__ (line 207) | def __init__(self, kernel_size: int, group_size: int, scale_factor: int):
    method forward (line 216) | def forward(self, features: Tensor, masks: Tensor) -> Tensor:
  class CARAFEPack (line 222) | class CARAFEPack(nn.Module):
    method __init__ (line 243) | def __init__(self,
    method init_weights (line 271) | def init_weights(self):
    method kernel_normalizer (line 277) | def kernel_normalizer(self, mask: Tensor) -> Tensor:
    method feature_reassemble (line 290) | def feature_reassemble(self, x: Tensor, mask: Tensor) -> Tensor:
    method forward (line 294) | def forward(self, x: Tensor) -> Tensor:

FILE: mmcv/ops/cc_attention.py
  function NEG_INF_DIAG (line 10) | def NEG_INF_DIAG(n: int, device: torch.device) -> torch.Tensor:
  class CrissCrossAttention (line 20) | class CrissCrossAttention(nn.Module):
    method __init__ (line 45) | def __init__(self, in_channels: int) -> None:
    method forward (line 53) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method __repr__ (line 82) | def __repr__(self) -> str:

FILE: mmcv/ops/chamfer_distance.py
  class ChamferDistanceFunction (line 15) | class ChamferDistanceFunction(Function):
    method forward (line 24) | def forward(ctx, xyz1: Tensor, xyz2: Tensor) -> Sequence[Tensor]:
    method backward (line 60) | def backward(ctx,

FILE: mmcv/ops/contour_expand.py
  function contour_expand (line 12) | def contour_expand(kernel_mask: Union[np.array, torch.Tensor],

FILE: mmcv/ops/conv2d_gradfix.py
  function no_weight_gradients (line 27) | def no_weight_gradients(disable=True):
  function conv2d (line 36) | def conv2d(input: torch.Tensor,
  function conv_transpose2d (line 69) | def conv_transpose2d(input: torch.Tensor,
  function _should_use_custom_op (line 97) | def _should_use_custom_op(input):
  function _to_tuple (line 108) | def _to_tuple(x, ndim):
  function _conv2d_gradfix (line 119) | def _conv2d_gradfix(

FILE: mmcv/ops/convex_iou.py
  function convex_giou (line 11) | def convex_giou(pointsets: torch.Tensor,
  function convex_iou (line 34) | def convex_iou(pointsets: torch.Tensor,

FILE: mmcv/ops/corner_pool.py
  function _corner_pool (line 9) | def _corner_pool(x: Tensor, dim: int, flip: bool) -> Tensor:
  class CornerPool (line 38) | class CornerPool(nn.Module):
    method __init__ (line 68) | def __init__(self, mode: str):
    method forward (line 73) | def forward(self, x: Tensor) -> Tensor:

FILE: mmcv/ops/correlation.py
  class CorrelationFunction (line 16) | class CorrelationFunction(Function):
    method forward (line 19) | def forward(ctx,
    method backward (line 65) | def backward(
    method _output_size (line 100) | def _output_size(ctx, input1):
  class Correlation (line 118) | class Correlation(nn.Module):
    method __init__ (line 171) | def __init__(self,
    method forward (line 186) | def forward(self, input1: Tensor, input2: Tensor) -> Tensor:
    method __repr__ (line 192) | def __repr__(self) -> str:

FILE: mmcv/ops/csrc/common/box_iou_rotated_utils.hpp
  type RotatedBox (line 21) | struct RotatedBox {
  type Point (line 26) | struct Point {
    method HOST_DEVICE_INLINE (line 29) | HOST_DEVICE_INLINE Point operator+(const Point& p) const {
    method HOST_DEVICE_INLINE (line 32) | HOST_DEVICE_INLINE Point& operator+=(const Point& p) {
    method HOST_DEVICE_INLINE (line 37) | HOST_DEVICE_INLINE Point operator-(const Point& p) const {
    method HOST_DEVICE_INLINE (line 40) | HOST_DEVICE_INLINE Point operator*(const T coeff) const {
  function HOST_DEVICE_INLINE (line 46) | HOST_DEVICE_INLINE T dot_2d(const Point<T>& A, const Point<T>& B) {
  function HOST_DEVICE_INLINE (line 51) | HOST_DEVICE_INLINE T cross_2d(const Point<T>& A, const Point<T>& B) {
  function HOST_DEVICE_INLINE (line 56) | HOST_DEVICE_INLINE void get_rotated_vertices(const RotatedBox<T>& box,
  function HOST_DEVICE_INLINE (line 77) | HOST_DEVICE_INLINE int get_intersection_points(const Point<T> (&pts1)[4],
  function HOST_DEVICE_INLINE (line 157) | HOST_DEVICE_INLINE int convex_hull_graham(const Point<T> (&p)[24],
  function HOST_DEVICE_INLINE (line 274) | HOST_DEVICE_INLINE T quadri_box_area(const Point<T> (&q)[4]) {
  function HOST_DEVICE_INLINE (line 285) | HOST_DEVICE_INLINE T polygon_area(const Point<T> (&q)[24], const int& m) {
  function HOST_DEVICE_INLINE (line 299) | HOST_DEVICE_INLINE T rotated_boxes_intersection(const RotatedBox<T>& box1,
  function HOST_DEVICE_INLINE (line 323) | HOST_DEVICE_INLINE T quadri_boxes_intersection(const Point<T> (&pts1)[4],
  function HOST_DEVICE_INLINE (line 344) | HOST_DEVICE_INLINE T single_box_iou_rotated(T const* const box1_raw,
  function HOST_DEVICE_INLINE (line 380) | HOST_DEVICE_INLINE T single_box_iou_quadri(T const* const pts1_raw,

FILE: mmcv/ops/csrc/common/cuda/common_cuda_helper.hpp
  function GET_BLOCKS (line 22) | inline int GET_BLOCKS(const int N, const int num_threads = THREADS_PER_B...
  function __device__ (line 29) | __device__ T bilinear_interpolate(const T* input, const int height,
  function __device__ (line 73) | __device__ void bilinear_interpolate_gradient(

FILE: mmcv/ops/csrc/common/mlu/common_mlu_helper.hpp
  function __mlu_func__ (line 39) | __mlu_func__ inline scalar_t min(scalar_t a, scalar_t b) {
  function __mlu_func__ (line 44) | __mlu_func__ inline scalar_t max(scalar_t a, scalar_t b) {
  function __mlu_func__ (line 64) | __mlu_func__ void convertInt2Float(float *dst, float *dst_addition, int ...
  function __mlu_func__ (line 140) | __mlu_func__ void convertFloat2Int(int *dst, float *dst_addition, float ...
  function __mlu_func__ (line 214) | __mlu_func__ inline void convertFloat2half(half *dst, float *src,
  function __mlu_func__ (line 235) | __mlu_func__ void recursiveSumPool(T *dst, int low_dim, int high_dim,

FILE: mmcv/ops/csrc/common/mps/MPSDevice.h
  type id (line 15) | typedef id<MTLDevice> MTLDevice_t;
  function namespace (line 23) | namespace at {

FILE: mmcv/ops/csrc/common/mps/MPSLibrary.h
  type id (line 12) | typedef id<MTLComputePipelineState> MTLComputePipelineState_t;
  type id (line 13) | typedef id<MTLLibrary> MTLLibrary_t;
  function class (line 21) | class MPSLibrary {
  function class (line 38) | class MPSLibraryManager {

FILE: mmcv/ops/csrc/common/mps/MPSStream.h
  type id (line 21) | typedef id<MTLCommandQueue> MTLCommandQueue_t;
  type id (line 22) | typedef id<MTLCommandBuffer> MTLCommandBuffer_t;
  type id (line 23) | typedef id<MTLSharedEvent> MTLSharedEvent_t;
  type id (line 24) | typedef id<MTLDevice> MTLDevice_t;
  function namespace (line 36) | namespace at {

FILE: mmcv/ops/csrc/common/mps/MPSUtils.h
  type id (line 9) | typedef id<MTLBuffer> MTLBuffer_t;
  type id (line 10) | typedef id<MTLComputeCommandEncoder> MTLComputeCommandEncoder_t;
  function MTLBuffer_t (line 19) | static inline MTLBuffer_t getMTLBufferStorage(const at::Tensor& tensor) {
  function setMTLArgsImpl (line 38) | inline void setMTLArgsImpl(MTLComputeCommandEncoder_t, int) {}

FILE: mmcv/ops/csrc/common/musa/common_musa_helper.hpp
  function GET_BLOCKS (line 22) | inline int GET_BLOCKS(const int N, const int num_threads = THREADS_PER_B...
  function __device__ (line 29) | __device__ T bilinear_interpolate(const T* input, const int height,
  function __device__ (line 73) | __device__ void bilinear_interpolate_gradient(

FILE: mmcv/ops/csrc/common/parrots_cuda_helper.hpp
  function __device__ (line 65) | static __inline__ __device__ double atomicAdd(double* address, double va...
  function __device__ (line 79) | static __inline__ __device__ float16 atomicAdd(float16* address, float16...

FILE: mmcv/ops/csrc/common/pytorch_device_registry.hpp
  function GetDeviceStr (line 19) | inline std::string GetDeviceStr(const at::Device& device) {
  class DeviceRegistry (line 30) | class DeviceRegistry
  class DeviceRegistry<Ret (*)(Args...), f> (line 33) | class DeviceRegistry<Ret (*)(Args...), f> {
    method Register (line 39) | void Register(at::DeviceType device, FunctionType function) {
    method FunctionType (line 43) | FunctionType Find(at::DeviceType device) const {
    method DeviceRegistry (line 47) | static DeviceRegistry& instance() {
    method DeviceRegistry (line 53) | DeviceRegistry() {
  function GetFirstTensorDevice (line 66) | at::Device GetFirstTensorDevice(T&& t, Args&&... args) {
  function GetFirstTensorDevice (line 72) | at::Device GetFirstTensorDevice(T&& t, Args&&... args) {
  function CheckDeviceConsistency (line 78) | inline std::pair<int, at::Device> CheckDeviceConsistency(
  function CheckDeviceConsistency (line 93) | std::pair<int, at::Device> CheckDeviceConsistency(const at::Device& device,
  function CheckDeviceConsistency (line 107) | std::pair<int, at::Device> CheckDeviceConsistency(const at::Device& device,
  function Dispatch (line 116) | auto Dispatch(const R& registry, const char* name, Args&&... args) {

FILE: mmcv/ops/csrc/common/pytorch_mlu_helper.hpp
  function getJobLimitCapability (line 28) | inline int32_t getJobLimitCapability() {
  function getCoreNumOfJobLimitCapability (line 39) | inline int32_t getCoreNumOfJobLimitCapability() {

FILE: mmcv/ops/csrc/common/pytorch_npu_util.hpp
  type aclOpExecutor (line 44) | struct aclOpExecutor
  type aclTensor (line 45) | struct aclTensor
  type aclScalar (line 46) | struct aclScalar
  type aclIntArray (line 47) | struct aclIntArray
  type aclFloatArray (line 48) | struct aclFloatArray
  type aclBoolArray (line 49) | struct aclBoolArray
  type aclTensorList (line 50) | struct aclTensorList
  function ConvertTensorToScalar (line 161) | inline c10::Scalar ConvertTensorToScalar(const at::Tensor &tensor) {
  function CopyTensorHostToDevice (line 204) | inline at::Tensor CopyTensorHostToDevice(const at::Tensor &cpu_tensor) {
  function CopyScalarToDevice (line 211) | inline at::Tensor CopyScalarToDevice(const c10::Scalar &cpu_scalar,
  function aclTensor (line 217) | inline aclTensor *ConvertType(const at::Tensor &at_tensor) {
  function aclScalar (line 277) | inline aclScalar *ConvertType(const at::Scalar &at_scalar) {
  function aclIntArray (line 318) | inline aclIntArray *ConvertType(const at::IntArrayRef &at_array) {
  function aclBoolArray (line 328) | inline aclBoolArray *ConvertType(const std::array<bool, N> &value) {
  function aclBoolArray (line 338) | inline aclBoolArray *ConvertType(const at::ArrayRef<bool> &value) {
  function aclTensorList (line 348) | inline aclTensorList *ConvertType(const at::TensorList &at_tensor_list) {
  function aclTensor (line 363) | inline aclTensor *ConvertType(const c10::optional<at::Tensor> &opt_tenso...
  function aclIntArray (line 370) | inline aclIntArray *ConvertType(
  function aclScalar (line 378) | inline aclScalar *ConvertType(const c10::optional<at::Scalar> &opt_scala...
  function aclDataType (line 385) | inline aclDataType ConvertType(const at::ScalarType scalarType) {
  function T (line 390) | T ConvertType(T value) {
  function ConvertToOpApiFunc (line 395) | auto ConvertToOpApiFunc(const Tuple &params, void *opApiAddr,
  function ConvertToOpApiFunc (line 404) | auto ConvertToOpApiFunc(const Tuple &params, void *opApiAddr) {
  function Release (line 410) | inline void Release(aclTensor *p) {
  function Release (line 418) | inline void Release(aclScalar *p) {
  function Release (line 426) | inline void Release(aclIntArray *p) {
  function Release (line 435) | inline void Release(aclBoolArray *p) {
  function Release (line 444) | inline void Release(aclTensorList *p) {
  function Release (line 455) | void Release(T value) {
  function CallRelease (line 460) | void CallRelease(Tuple t, std::index_sequence<I...>) {
  function ReleaseConvertTypes (line 465) | void ReleaseConvertTypes(Tuple &t) {
  function ConvertTypes (line 471) | constexpr auto ConvertTypes(Ts &...args) {
  function call (line 476) | auto call(Function f, Tuple t, std::index_sequence<I...>) {
  function call (line 481) | auto call(Function f, Tuple t) {
  function AddParamToBuf (line 487) | void AddParamToBuf(const std::array<bool, N> &value) {
  function AddParamToBuf (line 492) | void AddParamToBuf(const T &value) {
  function AddParamToBuf (line 509) | void AddParamToBuf(const T &arg, Args &...args) {

FILE: mmcv/ops/csrc/common/utils/spconv/paramsgrid.h
  function namespace (line 20) | namespace detail {

FILE: mmcv/ops/csrc/common/utils/spconv/prettyprint.h
  function namespace (line 25) | namespace pretty_print {
  type custom_delims_base (line 385) | struct custom_delims_base {
  type T (line 429) | typedef const T *const_iterator;
  type T (line 430) | typedef T value_type;
  type typename (line 447) | typedef typename T::const_local_iterator const_iterator;
  type typename (line 448) | typedef typename T::size_type size_type;
  function namespace (line 480) | namespace std {

FILE: mmcv/ops/csrc/common/utils/spconv/spconv/geometry.h
  function Index (line 24) | Index getValidOutPos(const Index *input_pos,
  function Index (line 87) | Index getValidOutPosTranspose(

FILE: mmcv/ops/csrc/common/utils/spconv/spconv/indice.h
  function namespace (line 19) | namespace functor {

FILE: mmcv/ops/csrc/common/utils/spconv/spconv/maxpool.h
  function namespace (line 19) | namespace functor {

FILE: mmcv/ops/csrc/common/utils/spconv/spconv/mp_helper.h
  function namespace (line 12) | namespace detail {
  function namespace (line 26) | namespace detail {

FILE: mmcv/ops/csrc/common/utils/spconv/spconv/reordering.h
  function namespace (line 19) | namespace functor {

FILE: mmcv/ops/csrc/common/utils/spconv/tensorview/helper_launch.h
  function namespace (line 4) | namespace tv {

FILE: mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h
  function namespace (line 28) | namespace tv {
  type CPU (line 115) | struct CPU {}
  type size_type (line 176) | typedef size_t size_type;
  function class (line 178) | class iterator {
  function rowArrayIdx (line 390) | unsigned rowArrayIdx(std::vector<int> &shape,
  function rowArrayIdx (line 406) | TV_HOST_DEVICE_INLINE unsigned rowArrayIdx(std::vector<int> &shape,
  function rowArrayIdx (line 418) | unsigned rowArrayIdx(const Shape &shape,
  function rowArrayIdx (line 431) | TV_HOST_DEVICE_INLINE unsigned rowArrayIdx(const Shape &shape,
  function rowArrayIdx (line 443) | unsigned rowArrayIdx(const Index *indexes,
  function Index (line 456) | Index rowArrayIdxInv(Index index, Index *output,
  function TV_HOST_DEVICE_INLINE (line 469) | TV_HOST_DEVICE_INLINE static unsigned run(const Shape &shape,
  type ArrayIndexRowMajor (line 477) | struct ArrayIndexRowMajor
  function TV_HOST_DEVICE_INLINE (line 478) | TV_HOST_DEVICE_INLINE static unsigned run(const Shape &shape,
  function namespace (line 484) | namespace detail {
  function TV_HOST_DEVICE_INLINE (line 515) | TV_HOST_DEVICE_INLINE TensorView() {}
  function explicit (line 516) | explicit TV_HOST_DEVICE_INLINE TensorView(scalar_t *ptr, Shape shape)
  function TV_HOST_DEVICE_INLINE (line 519) | TV_HOST_DEVICE_INLINE TensorView(scalar_t *ptr, Integers... shapes)
  function TV_HOST_DEVICE_INLINE (line 574) | TV_HOST_DEVICE_INLINE scalar_t &operator()() {
  function TV_HOST_DEVICE_INLINE (line 814) | TV_HOST_DEVICE_INLINE scalar_t &operator[](int idx) {
  function TV_HOST_DEVICE_INLINE (line 835) | TV_HOST_DEVICE_INLINE scalar_t *data() { return mPtr; }
  function TV_HOST_DEVICE_INLINE (line 836) | TV_HOST_DEVICE_INLINE const scalar_t *data() const { return mPtr; }
  function TV_HOST_DEVICE_INLINE (line 838) | TV_HOST_DEVICE_INLINE int dim(int idx) const { return mShape[idx]; }
  function Shape (line 842) | Shape shapes{int(newShapes)...};
  function Shape (line 855) | Shape shapes{int(newShapes)...};
  function Shape (line 887) | Shape new_shape{to_slice(slices)[0]...};
  function Slice (line 1021) | Slice to_slice(T1 s) const {
  function TV_HOST_DEVICE_INLINE (line 1025) | TV_HOST_DEVICE_INLINE Slice to_slice(Slice s) const { return Slice(s); }
  function namespace (line 1043) | namespace detail {
  function printTensorView (line 1073) | void printTensorView(const TensorView<scalar_t> tensor,
  function printTensorView (line 1120) | void printTensorView(const scalar_t *ptr, Shape shape) {
  function printTensorView (line 1126) | void printTensorView(const scalar_t *ptr, Shape shape,

FILE: mmcv/ops/csrc/parrots/active_rotated_filter.cpp
  function active_rotated_filter_forward_impl (line 8) | void active_rotated_filter_forward_impl(const Tensor input,
  function active_rotated_filter_backward_impl (line 14) | void active_rotated_filter_backward_impl(const Tensor grad_out,
  function active_rotated_filter_forward (line 20) | void active_rotated_filter_forward(const Tensor input, const Tensor indi...
  function active_rotated_filter_backward (line 25) | void active_rotated_filter_backward(const Tensor grad_out, const Tensor ...

FILE: mmcv/ops/csrc/parrots/active_rotated_filter_parrots.cpp
  function active_rotated_filter_forward_cuda_parrots (line 10) | void active_rotated_filter_forward_cuda_parrots(
  function active_rotated_filter_backward_cuda_parrots (line 19) | void active_rotated_filter_backward_cuda_parrots(
  function active_rotated_filter_forward_cpu_parrots (line 29) | void active_rotated_filter_forward_cpu_parrots(
  function active_rotated_filter_backward_cpu_parrots (line 38) | void active_rotated_filter_backward_cpu_parrots(

FILE: mmcv/ops/csrc/parrots/assign_score_withk.cpp
  function assign_score_withk_forward_impl (line 6) | void assign_score_withk_forward_impl(int B, int N0, int N1, int M, int K...
  function assign_score_withk_backward_impl (line 15) | void assign_score_withk_backward_impl(
  function assign_score_withk_forward (line 25) | void assign_score_withk_forward(const Tensor& points, const Tensor& cent...
  function assign_score_withk_backward (line 33) | void assign_score_withk_backward(const Tensor& grad_out, const Tensor& p...

FILE: mmcv/ops/csrc/parrots/assign_score_withk_parrots.cpp
  function assign_score_withk_forward_cuda_parrots (line 11) | void assign_score_withk_forward_cuda_parrots(CudaContext& ctx,
  function assign_score_withk_backward_cuda_parrots (line 36) | void assign_score_withk_backward_cuda_parrots(

FILE: mmcv/ops/csrc/parrots/ball_query._parrots.cpp
  function ball_query_parrots (line 11) | void ball_query_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/ball_query.cpp
  function ball_query_forward_impl (line 7) | void ball_query_forward_impl(int b, int n, int m, float min_radius,
  function ball_query_forward (line 15) | void ball_query_forward(Tensor new_xyz_tensor, Tensor xyz_tensor,

FILE: mmcv/ops/csrc/parrots/bbox_overlaps.cpp
  function bbox_overlaps_impl (line 5) | void bbox_overlaps_impl(const Tensor bboxes1, const Tensor bboxes2, Tens...
  function bbox_overlaps (line 11) | void bbox_overlaps(const Tensor bboxes1, const Tensor bboxes2, Tensor ious,

FILE: mmcv/ops/csrc/parrots/bbox_overlaps_parrots.cpp
  function bbox_overlaps_parrots (line 15) | void bbox_overlaps_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/border_align.cpp
  function border_align_forward_impl (line 5) | void border_align_forward_impl(const Tensor &input, const Tensor &boxes,
  function border_align_backward_impl (line 12) | void border_align_backward_impl(const Tensor &grad_output, const Tensor ...
  function border_align_forward (line 19) | void border_align_forward(const Tensor &input, const Tensor &boxes,
  function border_align_backward (line 25) | void border_align_backward(const Tensor &grad_output, const Tensor &boxes,

FILE: mmcv/ops/csrc/parrots/border_align_parrots.cpp
  function border_align_forward_cuda_parrots (line 11) | void border_align_forward_cuda_parrots(CudaContext& ctx, const SSElement...
  function border_align_backward_cuda_parrots (line 25) | void border_align_backward_cuda_parrots(CudaContext& ctx, const SSElemen...

FILE: mmcv/ops/csrc/parrots/box_iou_rotated.cpp
  function box_iou_rotated_impl (line 7) | void box_iou_rotated_impl(const Tensor boxes1, const Tensor boxes2, Tens...
  function box_iou_rotated (line 16) | void box_iou_rotated(const Tensor boxes1, const Tensor boxes2, Tensor ious,

FILE: mmcv/ops/csrc/parrots/box_iou_rotated_parrots.cpp
  function box_iou_rotated_cpu_parrots (line 14) | void box_iou_rotated_cpu_parrots(HostContext& ctx, const SSElement& attr,
  function box_iou_rotated_cuda_parrots (line 35) | void box_iou_rotated_cuda_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/carafe.cpp
  function carafe_forward_impl (line 5) | void carafe_forward_impl(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_impl (line 12) | void carafe_backward_impl(Tensor top_grad, Tensor rfeatures, Tensor masks,
  function carafe_forward (line 23) | void carafe_forward(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward (line 30) | void carafe_backward(Tensor top_grad, Tensor rfeatures, Tensor masks,

FILE: mmcv/ops/csrc/parrots/carafe_naive.cpp
  function carafe_naive_forward_impl (line 5) | void carafe_naive_forward_impl(Tensor features, Tensor masks, Tensor out...
  function carafe_naive_backward_impl (line 12) | void carafe_naive_backward_impl(Tensor top_grad, Tensor features, Tensor...
  function carafe_naive_forward (line 21) | void carafe_naive_forward(Tensor features, Tensor masks, Tensor output,
  function carafe_naive_backward (line 27) | void carafe_naive_backward(Tensor top_grad, Tensor features, Tensor masks,

FILE: mmcv/ops/csrc/parrots/carafe_naive_parrots.cpp
  function carafe_naive_forward_cuda_parrots (line 15) | void carafe_naive_forward_cuda_parrots(CudaContext& ctx, const SSElement...
  function carafe_naive_backward_cuda_parrots (line 37) | void carafe_naive_backward_cuda_parrots(CudaContext& ctx, const SSElemen...

FILE: mmcv/ops/csrc/parrots/carafe_parrots.cpp
  function carafe_forward_cuda_parrots (line 16) | void carafe_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,
  function carafe_backward_cuda_parrots (line 45) | void carafe_backward_cuda_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/chamfer_distance.cpp
  function chamfer_distance_forward_impl (line 8) | void chamfer_distance_forward_impl(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward_impl (line 15) | void chamfer_distance_backward_impl(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_forward (line 23) | void chamfer_distance_forward(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward (line 29) | void chamfer_distance_backward(const Tensor xyz1, const Tensor xyz2,

FILE: mmcv/ops/csrc/parrots/chamfer_distance_parrots.cpp
  function chamfer_distance_forward_cuda_parrots (line 10) | void chamfer_distance_forward_cuda_parrots(CudaContext& ctx,
  function chamfer_distance_backward_cuda_parrots (line 23) | void chamfer_distance_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/contour_expand.cpp
  class Point2d (line 10) | class Point2d {
    method Point2d (line 15) | Point2d() : x(0), y(0) {}
    method Point2d (line 16) | Point2d(int _x, int _y) : x(_x), y(_y) {}
  function kernel_dilate (line 19) | void kernel_dilate(const uint8_t *data, IntArrayRef data_shape,
  function contour_expand (line 89) | std::vector<std::vector<int>> contour_expand(Tensor kernel_mask,

FILE: mmcv/ops/csrc/parrots/contour_expand_parrots.cpp
  function contour_expand_parrots (line 12) | void contour_expand_parrots(T& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/convex_iou.cpp
  function convex_iou_impl (line 7) | void convex_iou_impl(const Tensor pointsets, const Tensor polygons,
  function convex_iou (line 12) | void convex_iou(const Tensor pointsets, const Tensor polygons, Tensor io...
  function convex_giou_impl (line 16) | void convex_giou_impl(const Tensor pointsets, const Tensor polygons,
  function convex_giou (line 21) | void convex_giou(const Tensor pointsets, const Tensor polygons, Tensor o...

FILE: mmcv/ops/csrc/parrots/convex_iou_parrots.cpp
  function convex_iou_forward_cuda_parrots (line 10) | void convex_iou_forward_cuda_parrots(CudaContext& ctx, const SSElement& ...
  function convex_giou_forward_cuda_parrots (line 19) | void convex_giou_forward_cuda_parrots(CudaContext& ctx, const SSElement&...

FILE: mmcv/ops/csrc/parrots/correlation.cpp
  function correlation_forward_impl (line 7) | void correlation_forward_impl(Tensor input1, Tensor input2, Tensor output,
  function correlation_backward_impl (line 17) | void correlation_backward_impl(Tensor grad_output, Tensor input1, Tensor...
  function correlation_forward (line 29) | void correlation_forward(Tensor input1, Tensor input2, Tensor output, in...
  function correlation_backward (line 38) | void correlation_backward(Tensor grad_output, Tensor input1, Tensor input2,

FILE: mmcv/ops/csrc/parrots/correlation_parrots.cpp
  function correlation_forward_cuda_parrots (line 11) | void correlation_forward_cuda_parrots(CudaContext& ctx, const SSElement&...
  function correlation_backward_cuda_parrots (line 41) | void correlation_backward_cuda_parrots(CudaContext& ctx, const SSElement...
  function correlation_forward_cpu_parrots (line 74) | void correlation_forward_cpu_parrots(HostContext& ctx, const SSElement& ...
  function correlation_backward_cpu_parrots (line 104) | void correlation_backward_cpu_parrots(HostContext& ctx, const SSElement&...

FILE: mmcv/ops/csrc/parrots/cudabind.cpp
  function assign_score_withk_forward_cuda (line 15) | void assign_score_withk_forward_cuda(int B, int N0, int N1, int M, int K...
  function assign_score_withk_backward_cuda (line 24) | void assign_score_withk_backward_cuda(
  function ball_query_forward_cuda (line 56) | void ball_query_forward_cuda(int b, int n, int m, float min_radius,
  function bbox_overlaps_cuda (line 74) | void bbox_overlaps_cuda(const Tensor bboxes1, const Tensor bboxes2, Tens...
  function border_align_forward_cuda (line 94) | void border_align_forward_cuda(const Tensor& input, const Tensor& boxes,
  function border_align_backward_cuda (line 101) | void border_align_backward_cuda(const Tensor& grad_output, const Tensor&...
  function carafe_forward_cuda (line 141) | void carafe_forward_cuda(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_cuda (line 149) | void carafe_backward_cuda(Tensor top_grad, Tensor rfeatures, Tensor masks,
  function carafe_naive_forward_cuda (line 184) | void carafe_naive_forward_cuda(Tensor features, Tensor masks, Tensor out...
  function carafe_naive_backward_cuda (line 191) | void carafe_naive_backward_cuda(Tensor top_grad, Tensor features, Tensor...
  function correlation_forward_cuda (line 228) | void correlation_forward_cuda(Tensor input1, Tensor input2, Tensor output,
  function correlation_backward_cuda (line 238) | void correlation_backward_cuda(Tensor grad_output, Tensor input1, Tensor...
  function deform_roi_pool_forward_cuda (line 333) | void deform_roi_pool_forward_cuda(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_cuda (line 342) | void deform_roi_pool_backward_cuda(Tensor grad_output, Tensor input,
  function sigmoid_focal_loss_forward_cuda (line 392) | void sigmoid_focal_loss_forward_cuda(Tensor input, Tensor target, Tensor...
  function sigmoid_focal_loss_backward_cuda (line 398) | void sigmoid_focal_loss_backward_cuda(Tensor input, Tensor target,
  function softmax_focal_loss_forward_cuda (line 405) | void softmax_focal_loss_forward_cuda(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_backward_cuda (line 411) | void softmax_focal_loss_backward_cuda(Tensor input, Tensor target,
  function furthest_point_sampling_forward_cuda (line 450) | void furthest_point_sampling_forward_cuda(Tensor points_tensor,
  function furthest_point_sampling_with_dist_forward_cuda (line 459) | void furthest_point_sampling_with_dist_forward_cuda(Tensor points_tensor,
  function gather_points_forward_cuda (line 505) | void gather_points_forward_cuda(int b, int c, int n, int npoints,
  function gather_points_backward_cuda (line 511) | void gather_points_backward_cuda(int b, int c, int n, int npoints,
  function group_points_forward_cuda (line 540) | void group_points_forward_cuda(int b, int c, int n, int npoints, int nsa...
  function group_points_backward_cuda (line 547) | void group_points_backward_cuda(int b, int c, int n, int npoints, int ns...
  function knn_forward_cuda (line 571) | void knn_forward_cuda(int b, int n, int m, int nsample, const Tensor xyz,
  function masked_im2col_forward_cuda (line 593) | void masked_im2col_forward_cuda(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_cuda (line 603) | void masked_col2im_forward_cuda(const Tensor col, const Tensor mask_h_idx,
  function Tensor (line 717) | Tensor nms_cuda(Tensor boxes, Tensor scores, float iou_threshold, int of...
  function points_in_boxes_part_forward_cuda (line 734) | void points_in_boxes_part_forward_cuda(int batch_size, int boxes_num,
  function points_in_boxes_all_forward_cuda (line 742) | void points_in_boxes_all_forward_cuda(int batch_size, int boxes_num,
  function psamask_forward_cuda (line 776) | void psamask_forward_cuda(const int psa_type, const Tensor input, Tensor...
  function psamask_backward_cuda (line 786) | void psamask_backward_cuda(const int psa_type, const Tensor grad_output,
  function roi_align_forward_cuda (line 823) | void roi_align_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_cuda (line 833) | void roi_align_backward_cuda(Tensor grad_output, Tensor rois, Tensor arg...
  function roi_align_rotated_forward_cuda (line 870) | void roi_align_rotated_forward_cuda(Tensor input, Tensor rois, Tensor ou...
  function roi_align_rotated_backward_cuda (line 891) | void roi_align_rotated_backward_cuda(Tensor top_grad, Tensor rois,
  function riroi_align_rotated_forward_cuda (line 941) | void riroi_align_rotated_forward_cuda(Tensor features, Tensor rois,
  function riroi_align_rotated_backward_cuda (line 963) | void riroi_align_rotated_backward_cuda(Tensor top_grad, Tensor rois,
  function roiaware_pool3d_forward_cuda (line 1013) | void roiaware_pool3d_forward_cuda(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_cuda (line 1025) | void roiaware_pool3d_backward_cuda(int boxes_num, int out_x, int out_y,
  function roipoint_pool3d_forward_cuda (line 1060) | void roipoint_pool3d_forward_cuda(int batch_size, int pts_num, int boxes...
  function roi_pool_forward_cuda (line 1089) | void roi_pool_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_cuda (line 1096) | void roi_pool_backward_cuda(Tensor grad_output, Tensor rois, Tensor argmax,
  function dynamic_point_to_voxel_forward_cuda (line 1124) | std::vector<torch::Tensor> dynamic_point_to_voxel_forward_cuda(
  function dynamic_point_to_voxel_backward_cuda (line 1131) | void dynamic_point_to_voxel_backward_cuda(
  function sync_bn_forward_mean_cuda (line 1179) | void sync_bn_forward_mean_cuda(const Tensor input, Tensor mean) {
  function sync_bn_forward_var_cuda (line 1183) | void sync_bn_forward_var_cuda(const Tensor input, const Tensor mean,
  function sync_bn_forward_output_cuda (line 1188) | void sync_bn_forward_output_cuda(const Tensor input, const Tensor mean,
  function sync_bn_backward_param_cuda (line 1199) | void sync_bn_backward_param_cuda(const Tensor grad_output, const Tensor ...
  function sync_bn_backward_data_cuda (line 1205) | void sync_bn_backward_data_cuda(const Tensor grad_output, const Tensor w...
  function three_interpolate_forward_cuda (line 1254) | void three_interpolate_forward_cuda(int b, int c, int m, int n,
  function three_interpolate_backward_cuda (line 1261) | void three_interpolate_backward_cuda(int b, int c, int n, int m,
  function three_nn_forward_cuda (line 1284) | void three_nn_forward_cuda(int b, int n, int m, const Tensor unknown,
  function tin_shift_forward_cuda (line 1299) | void tin_shift_forward_cuda(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_cuda (line 1303) | void tin_shift_backward_cuda(Tensor grad_output, Tensor shift,
  function hard_voxelize_forward_cuda (line 1342) | int hard_voxelize_forward_cuda(const at::Tensor& points, at::Tensor& vox...
  function nondeterministic_hard_voxelize_forward_cuda (line 1354) | int nondeterministic_hard_voxelize_forward_cuda(
  function dynamic_voxelize_forward_cuda (line 1364) | void dynamic_voxelize_forward_cuda(const at::Tensor& points, at::Tensor&...
  function rotated_feature_align_forward_cuda (line 1410) | void rotated_feature_align_forward_cuda(const Tensor features,
  function rotated_feature_align_backward_cuda (line 1418) | void rotated_feature_align_backward_cuda(const Tensor top_grad,
  function points_in_polygons_forward_cuda (line 1446) | void points_in_polygons_forward_cuda(const Tensor points, const Tensor p...
  function min_area_polygons_cuda (line 1462) | void min_area_polygons_cuda(const Tensor pointsets, Tensor polygons) {
  function active_rotated_filter_forward_cuda (line 1478) | void active_rotated_filter_forward_cuda(const Tensor input,
  function active_rotated_filter_backward_cuda (line 1483) | void active_rotated_filter_backward_cuda(const Tensor grad_out,
  function convex_iou_cuda (line 1505) | void convex_iou_cuda(const Tensor pointsets, const Tensor polygons,
  function convex_giou_cuda (line 1510) | void convex_giou_cuda(const Tensor pointsets, const Tensor polygons,
  function Tensor (line 1528) | Tensor diff_iou_rotated_sort_vertices_forward_cuda(Tensor vertices, Tens...
  function chamfer_distance_forward_cuda (line 1548) | void chamfer_distance_forward_cuda(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward_cuda (line 1555) | void chamfer_distance_backward_cuda(const Tensor xyz1, const Tensor xyz2,
  function prroi_pool_forward_cuda (line 1589) | void prroi_pool_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward_cuda (line 1596) | void prroi_pool_backward_cuda(Tensor grad_output, Tensor rois,
  function prroi_pool_coor_backward_cuda (line 1604) | void prroi_pool_coor_backward_cuda(Tensor output, Tensor grad_output,

FILE: mmcv/ops/csrc/parrots/deform_conv.cpp
  function deformable_im2col_impl (line 5) | void deformable_im2col_impl(Tensor data_im, Tensor data_offset,
  function deformable_col2im_impl (line 19) | void deformable_col2im_impl(Tensor data_col, Tensor data_offset,
  function deformable_col2im_coord_impl (line 33) | void deformable_col2im_coord_impl(
  function deform_conv_shape_check (line 45) | void deform_conv_shape_check(at::Tensor input, at::Tensor offset,
  function deform_conv_forward (line 140) | void deform_conv_forward(Tensor input, Tensor weight, Tensor offset,
  function deform_conv_backward_input (line 260) | void deform_conv_backward_input(Tensor input, Tensor offset, Tensor grad...
  function deform_conv_backward_parameters (line 392) | void deform_conv_backward_parameters(Tensor input, Tensor offset,

FILE: mmcv/ops/csrc/parrots/deform_conv_parrots.cpp
  function deform_conv_forward_cuda_parrots (line 11) | void deform_conv_forward_cuda_parrots(CudaContext& ctx, const SSElement&...
  function deform_conv_backward_input_cuda_parrots (line 43) | void deform_conv_backward_input_cuda_parrots(CudaContext& ctx,
  function deform_conv_backward_parameters_cuda_parrots (line 78) | void deform_conv_backward_parameters_cuda_parrots(
  function deform_conv_forward_cpu_parrots (line 113) | void deform_conv_forward_cpu_parrots(HostContext& ctx, const SSElement& ...
  function deform_conv_backward_input_cpu_parrots (line 145) | void deform_conv_backward_input_cpu_parrots(HostContext& ctx,
  function deform_conv_backward_parameters_cpu_parrots (line 180) | void deform_conv_backward_parameters_cpu_parrots(

FILE: mmcv/ops/csrc/parrots/deform_roi_pool.cpp
  function deform_roi_pool_forward_impl (line 5) | void deform_roi_pool_forward_impl(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_impl (line 14) | void deform_roi_pool_backward_impl(Tensor grad_output, Tensor input,
  function deform_roi_pool_forward (line 25) | void deform_roi_pool_forward(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward (line 34) | void deform_roi_pool_backward(Tensor grad_output, Tensor input, Tensor r...

FILE: mmcv/ops/csrc/parrots/deform_roi_pool_parrots.cpp
  function deform_roi_pool_forward_cuda_parrots (line 16) | void deform_roi_pool_forward_cuda_parrots(CudaContext& ctx,
  function deform_roi_pool_backward_cuda_parrots (line 50) | void deform_roi_pool_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/diff_iou_rotated.cpp
  function Tensor (line 5) | Tensor diff_iou_rotated_sort_vertices_forward_impl(Tensor vertices, Tens...
  function Tensor (line 11) | Tensor diff_iou_rotated_sort_vertices_forward(Tensor vertices, Tensor mask,

FILE: mmcv/ops/csrc/parrots/diff_iou_rotated_parrots.cpp
  function diff_iou_rotated_sort_vertices_forward_cuda_parrots (line 11) | void diff_iou_rotated_sort_vertices_forward_cuda_parrots(

FILE: mmcv/ops/csrc/parrots/focal_loss.cpp
  function sigmoid_focal_loss_forward_impl (line 5) | void sigmoid_focal_loss_forward_impl(Tensor input, Tensor target, Tensor...
  function sigmoid_focal_loss_backward_impl (line 11) | void sigmoid_focal_loss_backward_impl(Tensor input, Tensor target,
  function softmax_focal_loss_forward_impl (line 18) | void softmax_focal_loss_forward_impl(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_backward_impl (line 24) | void softmax_focal_loss_backward_impl(Tensor input, Tensor target,
  function sigmoid_focal_loss_forward (line 32) | void sigmoid_focal_loss_forward(Tensor input, Tensor target, Tensor weight,
  function sigmoid_focal_loss_backward (line 37) | void sigmoid_focal_loss_backward(Tensor input, Tensor target, Tensor wei...
  function softmax_focal_loss_forward (line 43) | void softmax_focal_loss_forward(Tensor input, Tensor target, Tensor weight,
  function softmax_focal_loss_backward (line 48) | void softmax_focal_loss_backward(Tensor input, Tensor target, Tensor wei...

FILE: mmcv/ops/csrc/parrots/focal_loss_parrots.cpp
  function sigmoid_focal_loss_forward_cuda_parrots (line 11) | void sigmoid_focal_loss_forward_cuda_parrots(CudaContext& ctx,
  function sigmoid_focal_loss_backward_cuda_parrots (line 29) | void sigmoid_focal_loss_backward_cuda_parrots(
  function softmax_focal_loss_forward_cuda_parrots (line 47) | void softmax_focal_loss_forward_cuda_parrots(CudaContext& ctx,
  function softmax_focal_loss_backward_cuda_parrots (line 64) | void softmax_focal_loss_backward_cuda_parrots(

FILE: mmcv/ops/csrc/parrots/furthest_point_sample.cpp
  function furthest_point_sampling_forward_impl (line 7) | void furthest_point_sampling_forward_impl(Tensor points_tensor,
  function furthest_point_sampling_with_dist_forward_impl (line 14) | void furthest_point_sampling_with_dist_forward_impl(Tensor points_tensor,
  function furthest_point_sampling_forward (line 22) | void furthest_point_sampling_forward(Tensor points_tensor, Tensor temp_t...
  function furthest_point_sampling_with_dist_forward (line 28) | void furthest_point_sampling_with_dist_forward(Tensor points_tensor,

FILE: mmcv/ops/csrc/parrots/furthest_point_sample_parrots.cpp
  function furthest_point_sample_forward_cuda_parrots (line 11) | void furthest_point_sample_forward_cuda_parrots(
  function furthest_point_sampling_with_dist_forward_cuda_parrots (line 26) | void furthest_point_sampling_with_dist_forward_cuda_parrots(

FILE: mmcv/ops/csrc/parrots/fused_bias_leakyrelu.cpp
  function fused_bias_leakyrelu_op_impl (line 105) | torch::Tensor fused_bias_leakyrelu_op_impl(const torch::Tensor& input,
  function fused_bias_leakyrelu (line 113) | torch::Tensor fused_bias_leakyrelu(const torch::Tensor& input,

FILE: mmcv/ops/csrc/parrots/fused_bias_parrots.cpp
  function fused_bias_leakyrelu_parrots (line 15) | void fused_bias_leakyrelu_parrots(CudaContext &ctx, const SSElement &attr,

FILE: mmcv/ops/csrc/parrots/gather_points.cpp
  function gather_points_forward_impl (line 4) | void gather_points_forward_impl(int b, int c, int n, int npoints,
  function gather_points_backward_impl (line 11) | void gather_points_backward_impl(int b, int c, int n, int npoints,
  function gather_points_forward (line 18) | void gather_points_forward(Tensor points_tensor, Tensor idx_tensor,
  function gather_points_backward (line 25) | void gather_points_backward(Tensor grad_out_tensor, Tensor idx_tensor,

FILE: mmcv/ops/csrc/parrots/gather_points_parrots.cpp
  function gather_points_forward_cuda_parrots (line 11) | void gather_points_forward_cuda_parrots(CudaContext& ctx, const SSElemen...
  function gather_points_backward_cuda_parrots (line 31) | void gather_points_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/group_points.cpp
  function group_points_forward_impl (line 8) | void group_points_forward_impl(int b, int c, int n, int npoints, int nsa...
  function group_points_backward_impl (line 15) | void group_points_backward_impl(int b, int c, int n, int npoints, int ns...
  function group_points_forward (line 22) | void group_points_forward(Tensor points_tensor, Tensor idx_tensor,
  function group_points_backward (line 29) | void group_points_backward(Tensor grad_out_tensor, Tensor idx_tensor,

FILE: mmcv/ops/csrc/parrots/group_points_parrots.cpp
  function group_points_forward_cuda_parrots (line 11) | void group_points_forward_cuda_parrots(CudaContext& ctx, const SSElement...
  function group_points_backward_cuda_parrots (line 31) | void group_points_backward_cuda_parrots(CudaContext& ctx, const SSElemen...

FILE: mmcv/ops/csrc/parrots/info.cpp
  function get_hiprt_version (line 9) | int get_hiprt_version() {
  function get_cudart_version (line 16) | int get_cudart_version() { return CUDART_VERSION; }
  function get_compiling_cuda_version (line 20) | std::string get_compiling_cuda_version() {
  function get_compiler_version (line 46) | std::string get_compiler_version() {

FILE: mmcv/ops/csrc/parrots/iou3d.cpp
  function iou3d_boxes_overlap_bev_forward_impl (line 15) | void iou3d_boxes_overlap_bev_forward_impl(const int num_a, const Tensor ...
  function iou3d_nms3d_forward_impl (line 22) | void iou3d_nms3d_forward_impl(const Tensor boxes, Tensor &keep,
  function iou3d_nms3d_normal_forward_impl (line 28) | void iou3d_nms3d_normal_forward_impl(const Tensor boxes, Tensor &keep,
  function iou3d_boxes_overlap_bev_forward (line 35) | void iou3d_boxes_overlap_bev_forward(Tensor boxes_a, Tensor boxes_b,
  function iou3d_nms3d_forward (line 47) | void iou3d_nms3d_forward(Tensor boxes, Tensor keep, Tensor keep_num,
  function iou3d_nms3d_normal_forward (line 57) | void iou3d_nms3d_normal_forward(Tensor boxes, Tensor keep, Tensor keep_num,

FILE: mmcv/ops/csrc/parrots/iou3d_parrots.cpp
  function iou3d_boxes_overlap_bev_forward_cuda_parrots (line 11) | void iou3d_boxes_overlap_bev_forward_cuda_parrots(
  function iou3d_nms3d_forward_cuda_parrots (line 22) | void iou3d_nms3d_forward_cuda_parrots(CudaContext& ctx, const SSElement&...
  function iou3d_nms3d_normal_forward_cuda_parrots (line 36) | void iou3d_nms3d_normal_forward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/knn.cpp
  function knn_forward_impl (line 7) | void knn_forward_impl(int b, int n, int m, int nsample, const Tensor xyz,
  function knn_forward (line 13) | void knn_forward(Tensor xyz_tensor, Tensor new_xyz_tensor, Tensor idx_te...

FILE: mmcv/ops/csrc/parrots/knn_parrots.cpp
  function knn_forward_cuda_parrots (line 11) | void knn_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/masked_conv2d.cpp
  function masked_im2col_forward_impl (line 5) | void masked_im2col_forward_impl(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_impl (line 13) | void masked_col2im_forward_impl(const Tensor col, const Tensor mask_h_idx,
  function masked_im2col_forward (line 20) | void masked_im2col_forward(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward (line 28) | void masked_col2im_forward(const Tensor col, const Tensor mask_h_idx,

FILE: mmcv/ops/csrc/parrots/masked_conv2d_parrots.cpp
  function masked_im2col_forward_cuda_parrots (line 11) | void masked_im2col_forward_cuda_parrots(CudaContext& ctx, const SSElemen...
  function masked_col2im_forward_cuda_parrots (line 33) | void masked_col2im_forward_cuda_parrots(CudaContext& ctx, const SSElemen...

FILE: mmcv/ops/csrc/parrots/min_area_polygons.cpp
  function min_area_polygons_impl (line 5) | void min_area_polygons_impl(const Tensor pointsets, Tensor polygons) {
  function min_area_polygons (line 9) | void min_area_polygons(const Tensor pointsets, Tensor polygons) {

FILE: mmcv/ops/csrc/parrots/min_area_polygons_parrots.cpp
  function min_area_polygons_cuda_parrots (line 11) | void min_area_polygons_cuda_parrots(CudaContext& ctx, const SSElement& a...

FILE: mmcv/ops/csrc/parrots/modulated_deform_conv.cpp
  function modulated_deformable_im2col_impl (line 5) | void modulated_deformable_im2col_impl(
  function modulated_deformable_col2im_impl (line 19) | void modulated_deformable_col2im_impl(
  function modulated_deformable_col2im_coord_impl (line 33) | void modulated_deformable_col2im_coord_impl(
  function modulated_deform_conv_forward (line 48) | void modulated_deform_conv_forward(
  function modulated_deform_conv_backward (line 126) | void modulated_deform_conv_backward(

FILE: mmcv/ops/csrc/parrots/modulated_deform_conv_parrots.cpp
  function modulated_deform_conv_forward_cuda_parrots (line 11) | void modulated_deform_conv_forward_cuda_parrots(
  function modulated_deform_conv_backward_cuda_parrots (line 46) | void modulated_deform_conv_backward_cuda_parrots(
  function modulated_deform_conv_forward_cpu_parrots (line 87) | void modulated_deform_conv_forward_cpu_parrots(
  function modulated_deform_conv_backward_cpu_parrots (line 122) | void modulated_deform_conv_backward_cpu_parrots(

FILE: mmcv/ops/csrc/parrots/ms_deform_attn.cpp
  function Tensor (line 15) | Tensor ms_deform_attn_impl_forward(const Tensor &value,
  function ms_deform_attn_impl_backward (line 26) | void ms_deform_attn_impl_backward(
  function Tensor (line 38) | Tensor ms_deform_attn_forward(const Tensor &value, const Tensor &spatial...
  function ms_deform_attn_backward (line 48) | void ms_deform_attn_backward(const Tensor &value, const Tensor &spatial_...

FILE: mmcv/ops/csrc/parrots/ms_deform_attn_parrots.cpp
  function ms_deform_attn_forward_parrots (line 23) | void ms_deform_attn_forward_parrots(CudaContext &ctx, const SSElement &a...
  function ms_deform_attn_backward_parrots (line 38) | void ms_deform_attn_backward_parrots(CudaContext &ctx, const SSElement &...

FILE: mmcv/ops/csrc/parrots/nms.cpp
  function Tensor (line 5) | Tensor nms_impl(Tensor boxes, Tensor scores, float iou_threshold, int of...
  function Tensor (line 9) | Tensor softnms_impl(Tensor boxes, Tensor scores, Tensor dets,
  function nms_match_impl (line 16) | std::vector<std::vector<int> > nms_match_impl(Tensor dets,
  function Tensor (line 21) | Tensor nms(Tensor boxes, Tensor scores, float iou_threshold, int offset) {
  function Tensor (line 25) | Tensor softnms(Tensor boxes, Tensor scores, Tensor dets, float iou_thres...
  function nms_match (line 31) | std::vector<std::vector<int> > nms_match(Tensor dets, float iou_threshol...

FILE: mmcv/ops/csrc/parrots/nms_parrots.cpp
  function nms_parrots (line 12) | void nms_parrots(T& ctx, const SSElement& attr,
  function softnms_parrots (line 31) | void softnms_parrots(T& ctx, const SSElement& attr,
  function nms_match_parrots (line 54) | void nms_match_parrots(T& ctx, const SSElement& attr,
  function nms_rotated_parrots (line 77) | void nms_rotated_parrots(T& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/nms_rotated.cpp
  function Tensor (line 18) | Tensor nms_rotated(const Tensor dets, const Tensor scores, const Tensor ...

FILE: mmcv/ops/csrc/parrots/pixel_group.cpp
  function pixel_group_impl (line 7) | std::vector<std::vector<float>> pixel_group_impl(
  function pixel_group (line 15) | std::vector<std::vector<float>> pixel_group(

FILE: mmcv/ops/csrc/parrots/pixel_group_parrots.cpp
  function pixel_group_parrots (line 12) | void pixel_group_parrots(T& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/points_in_boxes.cpp
  function points_in_boxes_part_forward_impl (line 4) | void points_in_boxes_part_forward_impl(int batch_size, int boxes_num,
  function points_in_boxes_all_forward_impl (line 12) | void points_in_boxes_all_forward_impl(int batch_size, int boxes_num,
  function points_in_boxes_part_forward (line 20) | void points_in_boxes_part_forward(Tensor boxes_tensor, Tensor pts_tensor,
  function points_in_boxes_all_forward (line 34) | void points_in_boxes_all_forward(Tensor boxes_tensor, Tensor pts_tensor,

FILE: mmcv/ops/csrc/parrots/points_in_boxes_parrots.cpp
  function points_in_boxes_part_forward_cuda_parrots (line 11) | void points_in_boxes_part_forward_cuda_parrots(
  function points_in_boxes_all_forward_cuda_parrots (line 23) | void points_in_boxes_all_forward_cuda_parrots(
  function points_in_boxes_forward_cpu_parrots (line 48) | void points_in_boxes_forward_cpu_parrots(HostContext& ctx,

FILE: mmcv/ops/csrc/parrots/points_in_polygons.cpp
  function points_in_polygons_forward_impl (line 4) | void points_in_polygons_forward_impl(const Tensor points, const Tensor p...
  function points_in_polygons_forward (line 11) | void points_in_polygons_forward(Tensor points, Tensor polygons, Tensor o...

FILE: mmcv/ops/csrc/parrots/points_in_polygons_parrots.cpp
  function points_in_polygons_cuda_parrots (line 11) | void points_in_polygons_cuda_parrots(CudaContext& ctx, const SSElement& ...

FILE: mmcv/ops/csrc/parrots/prroi_pool.cpp
  function prroi_pool_forward_impl (line 5) | void prroi_pool_forward_impl(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward_impl (line 12) | void prroi_pool_backward_impl(Tensor grad_output, Tensor rois,
  function prroi_pool_coor_backward_impl (line 19) | void prroi_pool_coor_backward_impl(Tensor output, Tensor grad_output,
  function prroi_pool_forward (line 28) | void prroi_pool_forward(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward (line 35) | void prroi_pool_backward(Tensor grad_output, Tensor rois, Tensor grad_in...
  function prroi_pool_coor_backward (line 42) | void prroi_pool_coor_backward(Tensor output, Tensor grad_output, Tensor ...

FILE: mmcv/ops/csrc/parrots/prroi_pool_parrots.cpp
  function prroi_pool_forward_cuda_parrots (line 11) | void prroi_pool_forward_cuda_parrots(CudaContext& ctx, const SSElement& ...
  function prroi_pool_backward_cuda_parrots (line 30) | void prroi_pool_backward_cuda_parrots(CudaContext& ctx, const SSElement&...
  function prroi_pool_coor_backward_cuda_parrots (line 49) | void prroi_pool_coor_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/psamask.cpp
  function psamask_forward_impl (line 7) | void psamask_forward_impl(const int psa_type, const Tensor input, Tensor...
  function psamask_backward_impl (line 17) | void psamask_backward_impl(const int psa_type, const Tensor grad_output,
  function psamask_forward (line 27) | void psamask_forward(const Tensor input, Tensor output, const int psa_type,
  function psamask_backward (line 35) | void psamask_backward(Tensor grad_output, const Tensor grad_input,

FILE: mmcv/ops/csrc/parrots/psamask_parrots.cpp
  function psamask_forward_cuda_parrots (line 10) | void psamask_forward_cuda_parrots(CudaContext &ctx, const SSElement &attr,
  function psamask_backward_cuda_parrots (line 31) | void psamask_backward_cuda_parrots(CudaContext &ctx, const SSElement &attr,
  function psamask_forward_cpu_parrots (line 54) | void psamask_forward_cpu_parrots(HostContext &ctx, const SSElement &attr,
  function psamask_backward_cpu_parrots (line 75) | void psamask_backward_cpu_parrots(HostContext &ctx, const SSElement &attr,

FILE: mmcv/ops/csrc/parrots/riroi_align_rotated.cpp
  function riroi_align_rotated_forward_impl (line 5) | void riroi_align_rotated_forward_impl(Tensor features, Tensor rois,
  function riroi_align_rotated_backward_impl (line 15) | void riroi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
  function riroi_align_rotated_forward (line 25) | void riroi_align_rotated_forward(Tensor features, Tensor rois, Tensor ou...
  function riroi_align_rotated_backward (line 34) | void riroi_align_rotated_backward(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/parrots/riroi_align_rotated_parrots.cpp
  function riroi_align_rotated_forward_cuda_parrots (line 10) | void riroi_align_rotated_forward_cuda_parrots(
  function riroi_align_rotated_backward_cuda_parrots (line 36) | void riroi_align_rotated_backward_cuda_parrots(

FILE: mmcv/ops/csrc/parrots/roi_align.cpp
  function roi_align_forward_impl (line 5) | void roi_align_forward_impl(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_impl (line 15) | void roi_align_backward_impl(Tensor grad_output, Tensor rois, Tensor arg...
  function roi_align_forward (line 25) | void roi_align_forward(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward (line 34) | void roi_align_backward(Tensor grad_output, Tensor rois, Tensor argmax_y,

FILE: mmcv/ops/csrc/parrots/roi_align_parrots.cpp
  function roi_align_forward_cuda_parrots (line 10) | void roi_align_forward_cuda_parrots(CudaContext& ctx, const SSElement& a...
  function roi_align_backward_cuda_parrots (line 38) | void roi_align_backward_cuda_parrots(CudaContext& ctx, const SSElement& ...
  function roi_align_forward_cpu_parrots (line 67) | void roi_align_forward_cpu_parrots(HostContext& ctx, const SSElement& attr,
  function roi_align_backward_cpu_parrots (line 95) | void roi_align_backward_cpu_parrots(HostContext& ctx, const SSElement& a...

FILE: mmcv/ops/csrc/parrots/roi_align_rotated.cpp
  function roi_align_rotated_forward_impl (line 5) | void roi_align_rotated_forward_impl(Tensor features, Tensor rois, Tensor...
  function roi_align_rotated_backward_impl (line 14) | void roi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
  function roi_align_rotated_forward (line 24) | void roi_align_rotated_forward(Tensor input, Tensor rois, Tensor output,
  function roi_align_rotated_backward (line 33) | void roi_align_rotated_backward(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/parrots/roi_align_rotated_parrots.cpp
  function roi_align_rotated_forward_cuda_parrots (line 10) | void roi_align_rotated_forward_cuda_parrots(CudaContext& ctx,
  function roi_align_rotated_backward_cuda_parrots (line 37) | void roi_align_rotated_backward_cuda_parrots(CudaContext& ctx,
  function roi_align_rotated_forward_cpu_parrots (line 65) | void roi_align_rotated_forward_cpu_parrots(HostContext& ctx,
  function roi_align_rotated_backward_cpu_parrots (line 92) | void roi_align_rotated_backward_cpu_parrots(HostContext& ctx,

FILE: mmcv/ops/csrc/parrots/roi_pool.cpp
  function roi_pool_forward_impl (line 5) | void roi_pool_forward_impl(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_impl (line 12) | void roi_pool_backward_impl(Tensor grad_output, Tensor rois, Tensor argmax,
  function roi_pool_forward (line 19) | void roi_pool_forward(Tensor input, Tensor rois, Tensor output, Tensor a...
  function roi_pool_backward (line 26) | void roi_pool_backward(Tensor grad_output, Tensor rois, Tensor argmax,

FILE: mmcv/ops/csrc/parrots/roi_pool_parrots.cpp
  function roi_pool_forward_cuda_parrots (line 10) | void roi_pool_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,
  function roi_pool_backward_cuda_parrots (line 30) | void roi_pool_backward_cuda_parrots(CudaContext& ctx, const SSElement& a...

FILE: mmcv/ops/csrc/parrots/roiaware_pool3d.cpp
  function roiaware_pool3d_forward_impl (line 4) | void roiaware_pool3d_forward_impl(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_impl (line 16) | void roiaware_pool3d_backward_impl(int boxes_num, int out_x, int out_y,
  function roiaware_pool3d_forward (line 27) | void roiaware_pool3d_forward(Tensor rois, Tensor pts, Tensor pts_feature,
  function roiaware_pool3d_backward (line 54) | void roiaware_pool3d_backward(Tensor pts_idx_of_voxels, Tensor argmax,

FILE: mmcv/ops/csrc/parrots/roiaware_pool3d_parrots.cpp
  function roiaware_pool3d_forward_cuda_parrots (line 11) | void roiaware_pool3d_forward_cuda_parrots(CudaContext& ctx,
  function roiaware_pool3d_backward_cuda_parrots (line 29) | void roiaware_pool3d_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/roipoint_pool3d.cpp
  function roipoint_pool3d_forward_impl (line 12) | void roipoint_pool3d_forward_impl(int batch_size, int pts_num, int boxes...
  function roipoint_pool3d_forward (line 23) | void roipoint_pool3d_forward(Tensor xyz, Tensor boxes3d, Tensor pts_feat...

FILE: mmcv/ops/csrc/parrots/roipoint_pool3d_parrots.cpp
  function roipoint_pool3d_forward_cuda_parrots (line 11) | void roipoint_pool3d_forward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/rotated_feature_align.cpp
  function rotated_feature_align_forward_impl (line 8) | void rotated_feature_align_forward_impl(const Tensor features,
  function rotated_feature_align_backward_impl (line 16) | void rotated_feature_align_backward_impl(const Tensor top_grad,
  function rotated_feature_align_forward (line 24) | void rotated_feature_align_forward(const Tensor features,
  function rotated_feature_align_backward (line 32) | void rotated_feature_align_backward(const Tensor top_grad,

FILE: mmcv/ops/csrc/parrots/rotated_feature_align_parrots.cpp
  function rotated_feature_align_forward_cuda_parrots (line 10) | void rotated_feature_align_forward_cuda_parrots(
  function rotated_feature_align_backward_cuda_parrots (line 27) | void rotated_feature_align_backward_cuda_parrots(
  function rotated_feature_align_forward_cpu_parrots (line 45) | void rotated_feature_align_forward_cpu_parrots(
  function rotated_feature_align_backward_cpu_parrots (line 62) | void rotated_feature_align_backward_cpu_parrots(

FILE: mmcv/ops/csrc/parrots/sync_bn.cpp
  function sync_bn_forward_mean_impl (line 5) | void sync_bn_forward_mean_impl(const Tensor input, Tensor mean) {
  function sync_bn_forward_var_impl (line 9) | void sync_bn_forward_var_impl(const Tensor input, const Tensor mean,
  function sync_bn_forward_output_impl (line 14) | void sync_bn_forward_output_impl(const Tensor input, const Tensor mean,
  function sync_bn_backward_param_impl (line 25) | void sync_bn_backward_param_impl(const Tensor grad_output, const Tensor ...
  function sync_bn_backward_data_impl (line 31) | void sync_bn_backward_data_impl(const Tensor grad_output, const Tensor w...
  function sync_bn_forward_mean (line 39) | void sync_bn_forward_mean(const Tensor input, Tensor mean) {
  function sync_bn_forward_var (line 43) | void sync_bn_forward_var(const Tensor input, const Tensor mean, Tensor v...
  function sync_bn_forward_output (line 47) | void sync_bn_forward_output(const Tensor input, const Tensor mean,
  function sync_bn_backward_param (line 58) | void sync_bn_backward_param(const Tensor grad_output, const Tensor norm,
  function sync_bn_backward_data (line 63) | void sync_bn_backward_data(const Tensor grad_output, const Tensor weight,

FILE: mmcv/ops/csrc/parrots/sync_bn_parrots.cpp
  function sync_bn_forward_mean_cuda_parrots (line 10) | void sync_bn_forward_mean_cuda_parrots(CudaContext& ctx, const SSElement...
  function sync_bn_forward_var_cuda_parrots (line 18) | void sync_bn_forward_var_cuda_parrots(CudaContext& ctx, const SSElement&...
  function sync_bn_forward_output_cuda_parrots (line 27) | void sync_bn_forward_output_cuda_parrots(CudaContext& ctx,
  function sync_bn_backward_param_cuda_parrots (line 54) | void sync_bn_backward_param_cuda_parrots(CudaContext& ctx,
  function sync_bn_backward_data_cuda_parrots (line 65) | void sync_bn_backward_data_cuda_parrots(CudaContext& ctx, const SSElemen...

FILE: mmcv/ops/csrc/parrots/three_interpolate.cpp
  function three_interpolate_forward_impl (line 7) | void three_interpolate_forward_impl(int b, int c, int m, int n,
  function three_interpolate_backward_impl (line 14) | void three_interpolate_backward_impl(int b, int c, int n, int m,
  function three_interpolate_forward (line 21) | void three_interpolate_forward(Tensor points_tensor, Tensor idx_tensor,
  function three_interpolate_backward (line 28) | void three_interpolate_backward(Tensor grad_out_tensor, Tensor idx_tensor,

FILE: mmcv/ops/csrc/parrots/three_interpolate_parrots.cpp
  function three_interpolate_forward_cuda_parrots (line 11) | void three_interpolate_forward_cuda_parrots(CudaContext& ctx,
  function three_interpolate_backward_cuda_parrots (line 33) | void three_interpolate_backward_cuda_parrots(CudaContext& ctx,

FILE: mmcv/ops/csrc/parrots/three_nn.cpp
  function three_nn_forward_impl (line 7) | void three_nn_forward_impl(int b, int n, int m, const Tensor unknown,
  function three_nn_forward (line 13) | void three_nn_forward(Tensor unknown_tensor, Tensor known_tensor,

FILE: mmcv/ops/csrc/parrots/three_nn_parrots.cpp
  function three_nn_forward_cuda_parrots (line 11) | void three_nn_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,

FILE: mmcv/ops/csrc/parrots/tin_shift.cpp
  function tin_shift_forward_impl (line 5) | void tin_shift_forward_impl(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_impl (line 9) | void tin_shift_backward_impl(Tensor grad_output, Tensor shift,
  function tin_shift_forward (line 14) | void tin_shift_forward(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward (line 18) | void tin_shift_backward(Tensor grad_output, Tensor shift, Tensor grad_in...

FILE: mmcv/ops/csrc/parrots/tin_shift_parrots.cpp
  function tin_shift_forward_cuda_parrots (line 10) | void tin_shift_forward_cuda_parrots(CudaContext &ctx, const SSElement &a...
  function tin_shift_backward_cuda_parrots (line 19) | void tin_shift_backward_cuda_parrots(CudaContext &ctx, const SSElement &...

FILE: mmcv/ops/csrc/parrots/upfirdn2d.cpp
  function upfirdn2d_op_impl (line 105) | torch::Tensor upfirdn2d_op_impl(const torch::Tensor& input,
  function upfirdn2d (line 113) | torch::Tensor upfirdn2d(const torch::Tensor& input, const torch::Tensor&...

FILE: mmcv/ops/csrc/parrots/upfirdn2d_parrots.cpp
  function upfirdn2d_parrots (line 14) | void upfirdn2d_parrots(CudaContext &ctx, const SSElement &attr,

FILE: mmcv/ops/csrc/parrots/voxelization.cpp
  function hard_voxelize_forward_impl (line 5) | int hard_voxelize_forward_impl(const at::Tensor &points, at::Tensor &vox...
  function nondeterministic_hard_voxelize_forward_impl (line 17) | int nondeterministic_hard_voxelize_forward_impl(
  function dynamic_voxelize_forward_impl (line 28) | void dynamic_voxelize_forward_impl(const at::Tensor &points, at::Tensor ...
  function hard_voxelize_forward (line 36) | void hard_voxelize_forward(const at::Tensor &points,
  function dynamic_voxelize_forward (line 62) | void dynamic_voxelize_forward(const at::Tensor &points,

FILE: mmcv/ops/csrc/parrots/voxelization_parrots.cpp
  function hard_voxelize_forward_cuda_parrots (line 11) | void hard_voxelize_forward_cuda_parrots(CudaContext& ctx, const SSElemen...
  function dynamic_voxelize_forward_cuda_parrots (line 36) | void dynamic_voxelize_forward_cuda_parrots(CudaContext& ctx,
  function hard_voxelize_forward_cpu_parrots (line 52) | void hard_voxelize_forward_cpu_parrots(HostContext& ctx, const SSElement...
  function dynamic_voxelize_forward_cpu_parrots (line 77) | void dynamic_voxelize_forward_cpu_parrots(HostContext& ctx,

FILE: mmcv/ops/csrc/pytorch/active_rotated_filter.cpp
  function active_rotated_filter_forward_impl (line 8) | void active_rotated_filter_forward_impl(const Tensor input,
  function active_rotated_filter_backward_impl (line 14) | void active_rotated_filter_backward_impl(const Tensor grad_out,
  function active_rotated_filter_forward (line 20) | void active_rotated_filter_forward(const Tensor input, const Tensor indi...
  function active_rotated_filter_backward (line 25) | void active_rotated_filter_backward(const Tensor grad_out, const Tensor ...

FILE: mmcv/ops/csrc/pytorch/assign_score_withk.cpp
  function assign_score_withk_forward_impl (line 6) | void assign_score_withk_forward_impl(int B, int N0, int N1, int M, int K...
  function assign_score_withk_backward_impl (line 15) | void assign_score_withk_backward_impl(
  function assign_score_withk_forward (line 25) | void assign_score_withk_forward(const Tensor& points, const Tensor& cent...
  function assign_score_withk_backward (line 33) | void assign_score_withk_backward(const Tensor& grad_out, const Tensor& p...

FILE: mmcv/ops/csrc/pytorch/ball_query.cpp
  function ball_query_forward_impl (line 7) | void ball_query_forward_impl(int b, int n, int m, float min_radius,
  function ball_query_forward (line 15) | void ball_query_forward(Tensor new_xyz_tensor, Tensor xyz_tensor,
  function stack_ball_query_forward_impl (line 22) | void stack_ball_query_forward_impl(float max_radius, int nsample,
  function stack_ball_query_forward (line 31) | void stack_ball_query_forward(Tensor new_xyz_tensor, Tensor new_xyz_batc...

FILE: mmcv/ops/csrc/pytorch/bbox_overlaps.cpp
  function bbox_overlaps_impl (line 16) | void bbox_overlaps_impl(const Tensor bboxes1, const Tensor bboxes2, Tens...
  function bbox_overlaps_diopi (line 23) | void bbox_overlaps_diopi(const Tensor bboxes1, const Tensor bboxes2,
  function bbox_overlaps (line 53) | void bbox_overlaps(const Tensor bboxes1, const Tensor bboxes2, Tensor ious,

FILE: mmcv/ops/csrc/pytorch/bezier_align.cpp
  function bezier_align_forward_impl (line 5) | void bezier_align_forward_impl(Tensor input, Tensor rois, Tensor output,
  function bezier_align_backward_impl (line 14) | void bezier_align_backward_impl(Tensor grad_output, Tensor rois,
  function bezier_align_forward (line 23) | void bezier_align_forward(Tensor input, Tensor rois, Tensor output,
  function bezier_align_backward (line 31) | void bezier_align_backward(Tensor grad_output, Tensor rois, Tensor grad_...

FILE: mmcv/ops/csrc/pytorch/bias_act.cpp
  function bias_act_op_impl (line 4) | torch::Tensor bias_act_op_impl(const torch::Tensor &input,
  function bias_act (line 14) | torch::Tensor bias_act(const torch::Tensor &input, const torch::Tensor &...

FILE: mmcv/ops/csrc/pytorch/border_align.cpp
  function border_align_forward_impl (line 5) | void border_align_forward_impl(const Tensor &input, const Tensor &boxes,
  function border_align_backward_impl (line 12) | void border_align_backward_impl(const Tensor &grad_output, const Tensor ...
  function border_align_forward (line 19) | void border_align_forward(const Tensor &input, const Tensor &boxes,
  function border_align_backward (line 25) | void border_align_backward(const Tensor &grad_output, const Tensor &boxes,

FILE: mmcv/ops/csrc/pytorch/box_iou_quadri.cpp
  function box_iou_quadri_impl (line 5) | void box_iou_quadri_impl(const Tensor boxes1, const Tensor boxes2, Tenso...
  function box_iou_quadri (line 14) | void box_iou_quadri(const Tensor boxes1, const Tensor boxes2, Tensor ious,

FILE: mmcv/ops/csrc/pytorch/box_iou_rotated.cpp
  function box_iou_rotated_impl (line 7) | void box_iou_rotated_impl(const Tensor boxes1, const Tensor boxes2, Tens...
  function box_iou_rotated (line 16) | void box_iou_rotated(const Tensor boxes1, const Tensor boxes2, Tensor ious,

FILE: mmcv/ops/csrc/pytorch/carafe.cpp
  function carafe_forward_impl (line 5) | void carafe_forward_impl(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_impl (line 12) | void carafe_backward_impl(Tensor top_grad, Tensor rfeatures, Tensor masks,
  function carafe_forward (line 23) | void carafe_forward(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward (line 30) | void carafe_backward(Tensor top_grad, Tensor rfeatures, Tensor masks,

FILE: mmcv/ops/csrc/pytorch/carafe_naive.cpp
  function carafe_naive_forward_impl (line 5) | void carafe_naive_forward_impl(Tensor features, Tensor masks, Tensor out...
  function carafe_naive_backward_impl (line 12) | void carafe_naive_backward_impl(Tensor top_grad, Tensor features, Tensor...
  function carafe_naive_forward (line 21) | void carafe_naive_forward(Tensor features, Tensor masks, Tensor output,
  function carafe_naive_backward (line 27) | void carafe_naive_backward(Tensor top_grad, Tensor features, Tensor masks,

FILE: mmcv/ops/csrc/pytorch/chamfer_distance.cpp
  function chamfer_distance_forward_impl (line 8) | void chamfer_distance_forward_impl(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward_impl (line 15) | void chamfer_distance_backward_impl(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_forward (line 23) | void chamfer_distance_forward(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward (line 29) | void chamfer_distance_backward(const Tensor xyz1, const Tensor xyz2,

FILE: mmcv/ops/csrc/pytorch/contour_expand.cpp
  class Point2d (line 10) | class Point2d {
    method Point2d (line 15) | Point2d() : x(0), y(0) {}
    method Point2d (line 16) | Point2d(int _x, int _y) : x(_x), y(_y) {}
  function kernel_dilate (line 19) | void kernel_dilate(const uint8_t *data, IntArrayRef data_shape,
  function contour_expand (line 89) | std::vector<std::vector<int>> contour_expand(Tensor kernel_mask,

FILE: mmcv/ops/csrc/pytorch/convex_iou.cpp
  function convex_iou_impl (line 7) | void convex_iou_impl(const Tensor pointsets, const Tensor polygons,
  function convex_iou (line 12) | void convex_iou(const Tensor pointsets, const Tensor polygons, Tensor io...
  function convex_giou_impl (line 16) | void convex_giou_impl(const Tensor pointsets, const Tensor polygons,
  function convex_giou (line 21) | void convex_giou(const Tensor pointsets, const Tensor polygons, Tensor o...

FILE: mmcv/ops/csrc/pytorch/correlation.cpp
  function correlation_forward_impl (line 7) | void correlation_forward_impl(Tensor input1, Tensor input2, Tensor output,
  function correlation_backward_impl (line 17) | void correlation_backward_impl(Tensor grad_output, Tensor input1, Tensor...
  function correlation_forward (line 29) | void correlation_forward(Tensor input1, Tensor input2, Tensor output, in...
  function correlation_backward (line 38) | void correlation_backward(Tensor grad_output, Tensor input1, Tensor input2,

FILE: mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp
  function active_rotated_filter_forward_cpu_kernel (line 8) | void active_rotated_filter_forward_cpu_kernel(
  function active_rotated_filter_backward_cpu_kernel (line 35) | void active_rotated_filter_backward_cpu_kernel(
  function ActiveRotatedFilterForwardCPULauncher (line 63) | void ActiveRotatedFilterForwardCPULauncher(const Tensor input,
  function ActiveRotatedFilterBackwardCPULauncher (line 82) | void ActiveRotatedFilterBackwardCPULauncher(const Tensor grad_out,
  function active_rotated_filter_forward_cpu (line 101) | void active_rotated_filter_forward_cpu(const Tensor input, const Tensor ...
  function active_rotated_filter_backward_cpu (line 106) | void active_rotated_filter_backward_cpu(const Tensor grad_out,

FILE: mmcv/ops/csrc/pytorch/cpu/bbox_overlaps_cpu.cpp
  function bbox_overlaps_cpu_kernel (line 8) | void bbox_overlaps_cpu_kernel(const Tensor boxes1, const Tensor boxes2,
  function bbox_overlaps_cpu (line 57) | void bbox_overlaps_cpu(const Tensor boxes1, const Tensor boxes2, Tensor ...

FILE: mmcv/ops/csrc/pytorch/cpu/bezier_align.cpp
  type PreCalc (line 12) | struct PreCalc {
  function T (line 24) | T bezier_curve(const T p0, const T p1, const T p2, const T p3, const T u) {
  function pre_calc_for_bilinear_interpolate (line 31) | void pre_calc_for_bilinear_interpolate(
  function BezierAlignForward (line 129) | void BezierAlignForward(const int nthreads, const T *input, const T *rois,
  function bilinear_interpolate_gradient (line 229) | void bilinear_interpolate_gradient(const int height, const int width, T ...
  function add (line 276) | inline void add(T *address, const T &val) {
  function BezierAlignBackward (line 281) | void BezierAlignBackward(const int nthreads, const T *grad_output,
  function BezierAlignForwardCPULauncher (line 389) | void BezierAlignForwardCPULauncher(Tensor input, Tensor rois, Tensor out...
  function BezierAlignBackwardCPULauncher (line 408) | void BezierAlignBackwardCPULauncher(Tensor grad_output, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/cpu/box_iou_quadri.cpp
  function box_iou_quadri_cpu_kernel (line 7) | void box_iou_quadri_cpu_kernel(const Tensor boxes1, const Tensor boxes2,
  function box_iou_quadri_cpu (line 29) | void box_iou_quadri_cpu(const Tensor boxes1, const Tensor boxes2, Tensor...

FILE: mmcv/ops/csrc/pytorch/cpu/box_iou_rotated.cpp
  function box_iou_rotated_cpu_kernel (line 9) | void box_iou_rotated_cpu_kernel(const Tensor boxes1, const Tensor boxes2,
  function box_iou_rotated_cpu (line 31) | void box_iou_rotated_cpu(const Tensor boxes1, const Tensor boxes2, Tenso...

FILE: mmcv/ops/csrc/pytorch/cpu/deform_conv.cpp
  function T (line 6) | T deformable_im2col_bilinear_cpu(const T *input, const int data_width,
  function T (line 40) | T get_gradient_weight_cpu(T argmax_h, T argmax_w, const int h, const int w,
  function T (line 66) | T get_coordinate_weight_cpu(T argmax_h, T argmax_w, const int height,
  function deformable_im2col_cpu_kernel (line 114) | void deformable_im2col_cpu_kernel(
  function deformable_col2im_cpu_kernel (line 167) | void deformable_col2im_cpu_kernel(
  function deformable_col2im_coord_cpu_kernel (line 224) | void deformable_col2im_coord_cpu_kernel(
  function deformable_im2col_cpu (line 291) | void deformable_im2col_cpu(Tensor data_im, Tensor data_offset,
  function deformable_col2im_cpu (line 318) | void deformable_col2im_cpu(Tensor data_col, Tensor data_offset,
  function deformable_col2im_coord_cpu (line 349) | void deformable_col2im_coord_cpu(

FILE: mmcv/ops/csrc/pytorch/cpu/modulated_deform_conv.cpp
  function T (line 6) | T dmcn_im2col_bilinear_cpu(const T *input, const int data_width,
  function T (line 36) | T dmcn_get_gradient_weight_cpu(T argmax_h, T argmax_w, const int h, cons...
  function T (line 62) | T dmcn_get_coordinate_weight_cpu(T argmax_h, T argmax_w, const int height,
  function modulated_deformable_im2col_cpu_kernel (line 110) | void modulated_deformable_im2col_cpu_kernel(
  function modulated_deformable_col2im_cpu_kernel (line 171) | void modulated_deformable_col2im_cpu_kernel(
  function modulated_deformable_col2im_coord_cpu_kernel (line 234) | void modulated_deformable_col2im_coord_cpu_kernel(
  function modulated_deformable_im2col_cpu (line 322) | void modulated_deformable_im2col_cpu(
  function modulated_deformable_col2im_cpu (line 348) | void modulated_deformable_col2im_cpu(
  function modulated_deformable_col2im_coord_cpu (line 374) | void modulated_deformable_col2im_coord_cpu(

FILE: mmcv/ops/csrc/pytorch/cpu/nms.cpp
  function Tensor (line 5) | Tensor nms_cpu(Tensor boxes, Tensor scores, float iou_threshold, int off...
  function Tensor (line 59) | Tensor softnms_cpu(Tensor boxes, Tensor scores, Tensor dets,
  function nms_match_cpu (line 168) | std::vector<std::vector<int> > nms_match_cpu(Tensor dets, float iou_thre...

FILE: mmcv/ops/csrc/pytorch/cpu/nms_quadri.cpp
  function Tensor (line 6) | Tensor nms_quadri_cpu_kernel(const Tensor dets, const Tensor scores,
  function Tensor (line 57) | Tensor nms_quadri_cpu(const Tensor dets, const Tensor scores,

FILE: mmcv/ops/csrc/pytorch/cpu/nms_rotated.cpp
  function Tensor (line 8) | Tensor nms_rotated_cpu_kernel(const Tensor dets, const Tensor scores,
  function Tensor (line 59) | Tensor nms_rotated_cpu(const Tensor dets, const Tensor scores,

FILE: mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp
  function estimate_confidence (line 9) | std::vector<std::vector<float>> estimate_confidence(int32_t* label,
  function pixel_group_cpu (line 39) | std::vector<std::vector<float>> pixel_group_cpu(

FILE: mmcv/ops/csrc/pytorch/cpu/points_in_boxes.cpp
  function lidar_to_local_coords_cpu (line 3) | inline void lidar_to_local_coords_cpu(float shift_x, float shift_y, floa...
  function check_pt_in_box3d_cpu (line 10) | inline int check_pt_in_box3d_cpu(const float *pt, const float *box3d,
  function points_in_boxes_cpu_forward (line 28) | void points_in_boxes_cpu_forward(Tensor boxes_tensor, Tensor pts_tensor,

FILE: mmcv/ops/csrc/pytorch/cpu/psamask.cpp
  function psamask_collect_forward (line 14) | void psamask_collect_forward(const int num_, const int h_feature,
  function psamask_distribute_forward (line 51) | void psamask_distribute_forward(const int num_, const int h_feature,
  function psamask_collect_backward (line 88) | void psamask_collect_backward(const int num_, const int h_feature,
  function psamask_distribute_backward (line 124) | void psamask_distribute_backward(const int num_, const int h_feature,
  function psamask_forward_cpu (line 160) | void psamask_forward_cpu(const int psa_type, const Tensor input, Tensor ...
  function psamask_backward_cpu (line 173) | void psamask_backward_cpu(const int psa_type, const Tensor grad_output,

FILE: mmcv/ops/csrc/pytorch/cpu/roi_align.cpp
  type PreCalc (line 12) | struct PreCalc {
  function pre_calc_for_bilinear_interpolate (line 24) | void pre_calc_for_bilinear_interpolate(
  function ROIAlignForward (line 111) | void ROIAlignForward(const int nthreads, const T* input, const T* rois,
  function bilinear_interpolate_gradient (line 217) | void bilinear_interpolate_gradient(const int height, const int width, T ...
  function add (line 266) | inline void add(T* address, const T& val) {
  function ROIAlignBackward (line 271) | void ROIAlignBackward(const int nthreads, const T* grad_output, const T*...
  function ROIAlignForwardCPULauncher (line 384) | void ROIAlignForwardCPULauncher(Tensor input, Tensor rois, Tensor output,
  function ROIAlignBackwardCPULauncher (line 405) | void ROIAlignBackwardCPULauncher(Tensor grad_output, Tensor rois,
  function roi_align_forward_cpu (line 434) | void roi_align_forward_cpu(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_cpu (line 443) | void roi_align_backward_cpu(Tensor grad_output, Tensor rois, Tensor argm...

FILE: mmcv/ops/csrc/pytorch/cpu/roi_align_rotated.cpp
  type PreCalc (line 12) | struct PreCalc {
  function pre_calc_for_bilinear_interpolate (line 24) | void pre_calc_for_bilinear_interpolate(
  function ROIAlignRotatedForward (line 116) | void ROIAlignRotatedForward(const int nthreads, const T* input,
  function bilinear_interpolate_gradient (line 215) | void bilinear_interpolate_gradient(const int height, const int width, T ...
  function add (line 268) | inline void add(T* address, const T& val) {
  function ROIAlignRotatedBackward (line 273) | void ROIAlignRotatedBackward(
  function ROIAlignRotatedForwardCPULauncher (line 374) | void ROIAlignRotatedForwardCPULauncher(Tensor input, Tensor rois, Tensor...
  function ROIAlignRotatedBackwardCPULauncher (line 393) | void ROIAlignRotatedBackwardCPULauncher(Tensor grad_output, Tensor rois,
  function roi_align_rotated_forward_cpu (line 419) | void roi_align_rotated_forward_cpu(Tensor input, Tensor rois, Tensor out...
  function roi_align_rotated_backward_cpu (line 428) | void roi_align_rotated_backward_cpu(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
  function T (line 7) | T bilinear_interpolate(const T* input, const int height, const int width...
  function rotated_feature_align_forward_cpu_kernel (line 49) | void rotated_feature_align_forward_cpu_kernel(
  function bilinear_interpolate_gradient (line 100) | void bilinear_interpolate_gradient(const int height, const int width, T ...
  function valueAdd (line 142) | inline void valueAdd(scalar_t* address, scalar_t val) {
  function rotated_feature_align_backward_cpu_kernel (line 148) | void rotated_feature_align_backward_cpu_kernel(
  function rotated_feature_align_forward_cpu (line 213) | void rotated_feature_align_forward_cpu(const Tensor features,
  function rotated_feature_align_backward_cpu (line 231) | void rotated_feature_align_backward_cpu(const Tensor top_grad,

FILE: mmcv/ops/csrc/pytorch/cpu/sparse_indice.cpp
  type functor (line 21) | namespace functor {
    type CreateConvIndicePairFunctor<tv::CPU, Index, IndexGrid, NDim> (line 23) | struct CreateConvIndicePairFunctor<tv::CPU, Index, IndexGrid, NDim> {
      method Index (line 24) | Index operator()(const tv::CPU& d, tv::TensorView<const Index> indic...
    type CreateSubMIndicePairFunctor<tv::CPU, Index, IndexGrid, NDim> (line 49) | struct CreateSubMIndicePairFunctor<tv::CPU, Index, IndexGrid, NDim> {
      method Index (line 50) | Index operator()(const tv::CPU& d, tv::TensorView<const Index> indic...

FILE: mmcv/ops/csrc/pytorch/cpu/sparse_maxpool.cpp
  type functor (line 20) | namespace functor {
    type SparseMaxPoolForwardFunctor<tv::CPU, scalar_t, Index> (line 22) | struct SparseMaxPoolForwardFunctor<tv::CPU, scalar_t, Index> {
    type SparseMaxPoolBackwardFunctor<tv::CPU, scalar_t, Index> (line 43) | struct SparseMaxPoolBackwardFunctor<tv::CPU, scalar_t, Index> {

FILE: mmcv/ops/csrc/pytorch/cpu/sparse_reordering.cpp
  type functor (line 20) | namespace functor {
    type SparseGatherFunctor<tv::CPU, scalar_t, Index> (line 22) | struct SparseGatherFunctor<tv::CPU, scalar_t, Index> {
    type SparseScatterAddFunctor<tv::CPU, scalar_t, Index> (line 36) | struct SparseScatterAddFunctor<tv::CPU, scalar_t, Index> {

FILE: mmcv/ops/csrc/pytorch/cpu/voxelization.cpp
  function dynamic_voxelize_forward_cpu_kernel (line 6) | void dynamic_voxelize_forward_cpu_kernel(
  function hard_voxelize_forward_cpu_kernel (line 48) | void hard_voxelize_forward_cpu_kernel(
  function dynamic_voxelize_forward_cpu (line 106) | void dynamic_voxelize_forward_cpu(const at::Tensor& points, at::Tensor& ...
  function hard_voxelize_forward_cpu (line 131) | int hard_voxelize_forward_cpu(const at::Tensor& points, at::Tensor& voxels,

FILE: mmcv/ops/csrc/pytorch/cuda/cudabind.cpp
  function assign_score_withk_forward_cuda (line 15) | void assign_score_withk_forward_cuda(int B, int N0, int N1, int M, int K...
  function assign_score_withk_backward_cuda (line 24) | void assign_score_withk_backward_cuda(
  function ball_query_forward_cuda (line 56) | void ball_query_forward_cuda(int b, int n, int m, float min_radius,
  function stack_ball_query_forward_cuda (line 77) | void stack_ball_query_forward_cuda(float max_radius, int nsample,
  function bbox_overlaps_cuda (line 98) | void bbox_overlaps_cuda(const Tensor bboxes1, const Tensor bboxes2, Tens...
  function border_align_forward_cuda (line 118) | void border_align_forward_cuda(const Tensor &input, const Tensor &boxes,
  function border_align_backward_cuda (line 125) | void border_align_backward_cuda(const Tensor &grad_output, const Tensor ...
  function carafe_forward_cuda (line 172) | void carafe_forward_cuda(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_cuda (line 180) | void carafe_backward_cuda(Tensor top_grad, Tensor rfeatures, Tensor masks,
  function carafe_naive_forward_cuda (line 215) | void carafe_naive_forward_cuda(Tensor features, Tensor masks, Tensor out...
  function carafe_naive_backward_cuda (line 222) | void carafe_naive_backward_cuda(Tensor top_grad, Tensor features, Tensor...
  function correlation_forward_cuda (line 259) | void correlation_forward_cuda(Tensor input1, Tensor input2, Tensor output,
  function correlation_backward_cuda (line 269) | void correlation_backward_cuda(Tensor grad_output, Tensor input1, Tensor...
  function deform_roi_pool_forward_cuda (line 364) | void deform_roi_pool_forward_cuda(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_cuda (line 373) | void deform_roi_pool_backward_cuda(Tensor grad_output, Tensor input,
  function sigmoid_focal_loss_forward_cuda (line 423) | void sigmoid_focal_loss_forward_cuda(Tensor input, Tensor target, Tensor...
  function sigmoid_focal_loss_backward_cuda (line 429) | void sigmoid_focal_loss_backward_cuda(Tensor input, Tensor target,
  function softmax_focal_loss_forward_cuda (line 436) | void softmax_focal_loss_forward_cuda(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_backward_cuda (line 442) | void softmax_focal_loss_backward_cuda(Tensor input, Tensor target,
  function furthest_point_sampling_forward_cuda (line 481) | void furthest_point_sampling_forward_cuda(Tensor points_tensor,
  function furthest_point_sampling_with_dist_forward_cuda (line 490) | void furthest_point_sampling_with_dist_forward_cuda(Tensor points_tensor,
  function gather_points_forward_cuda (line 561) | void gather_points_forward_cuda(int b, int c, int n, int npoints,
  function gather_points_backward_cuda (line 567) | void gather_points_backward_cuda(int b, int c, int n, int npoints,
  function group_points_forward_cuda (line 596) | void group_points_forward_cuda(int b, int c, int n, int npoints, int nsa...
  function group_points_backward_cuda (line 603) | void group_points_backward_cuda(int b, int c, int n, int npoints, int ns...
  function stack_group_points_forward_cuda (line 632) | void stack_group_points_forward_cuda(int b, int c, int m, int nsample,
  function stack_group_points_backward_cuda (line 643) | void stack_group_points_backward_cuda(int b, int c, int m, int n, int ns...
  function iou3d_boxes_overlap_bev_forward_cuda (line 687) | void iou3d_boxes_overlap_bev_forward_cuda(const int num_a, const Tensor ...
  function iou3d_nms3d_forward_cuda (line 694) | void iou3d_nms3d_forward_cuda(const Tensor boxes, Tensor &keep,
  function iou3d_nms3d_normal_forward_cuda (line 700) | void iou3d_nms3d_normal_forward_cuda(const Tensor boxes, Tensor &keep,
  function knn_forward_cuda (line 728) | void knn_forward_cuda(int b, int n, int m, int nsample, const Tensor xyz,
  function masked_im2col_forward_cuda (line 750) | void masked_im2col_forward_cuda(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_cuda (line 760) | void masked_col2im_forward_cuda(const Tensor col, const Tensor mask_h_idx,
  function Tensor (line 874) | Tensor nms_cuda(Tensor boxes, Tensor scores, float iou_threshold, int of...
  function points_in_boxes_part_forward_cuda (line 891) | void points_in_boxes_part_forward_cuda(int batch_size, int boxes_num,
  function points_in_boxes_all_forward_cuda (line 899) | void points_in_boxes_all_forward_cuda(int batch_size, int boxes_num,
  function psamask_forward_cuda (line 933) | void psamask_forward_cuda(const int psa_type, const Tensor input, Tensor...
  function psamask_backward_cuda (line 943) | void psamask_backward_cuda(const int psa_type, const Tensor grad_output,
  function roi_align_forward_cuda (line 980) | void roi_align_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_cuda (line 990) | void roi_align_backward_cuda(Tensor grad_output, Tensor rois, Tensor arg...
  function roi_align_rotated_forward_cuda (line 1027) | void roi_align_rotated_forward_cuda(Tensor input, Tensor rois, Tensor ou...
  function roi_align_rotated_backward_cuda (line 1048) | void roi_align_rotated_backward_cuda(Tensor top_grad, Tensor rois,
  function riroi_align_rotated_forward_cuda (line 1098) | void riroi_align_rotated_forward_cuda(Tensor features, Tensor rois,
  function riroi_align_rotated_backward_cuda (line 1120) | void riroi_align_rotated_backward_cuda(Tensor top_grad, Tensor rois,
  function roiaware_pool3d_forward_cuda (line 1170) | void roiaware_pool3d_forward_cuda(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_cuda (line 1182) | void roiaware_pool3d_backward_cuda(int boxes_num, int out_x, int out_y,
  function roipoint_pool3d_forward_cuda (line 1217) | void roipoint_pool3d_forward_cuda(int batch_size, int pts_num, int boxes...
  function roi_pool_forward_cuda (line 1246) | void roi_pool_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_cuda (line 1253) | void roi_pool_backward_cuda(Tensor grad_output, Tensor rois, Tensor argmax,
  function dynamic_point_to_voxel_forward_cuda (line 1281) | std::vector<torch::Tensor> dynamic_point_to_voxel_forward_cuda(
  function dynamic_point_to_voxel_backward_cuda (line 1288) | void dynamic_point_to_voxel_backward_cuda(
  function sync_bn_forward_mean_cuda (line 1336) | void sync_bn_forward_mean_cuda(const Tensor input, Tensor mean) {
  function sync_bn_forward_var_cuda (line 1340) | void sync_bn_forward_var_cuda(const Tensor input, const Tensor mean,
  function sync_bn_forward_output_cuda (line 1345) | void sync_bn_forward_output_cuda(const Tensor input, const Tensor mean,
  function sync_bn_backward_param_cuda (line 1356) | void sync_bn_backward_param_cuda(const Tensor grad_output, const Tensor ...
  function sync_bn_backward_data_cuda (line 1362) | void sync_bn_backward_data_cuda(const Tensor grad_output, const Tensor w...
  function three_interpolate_forward_cuda (line 1411) | void three_interpolate_forward_cuda(int b, int c, int m, int n,
  function three_interpolate_backward_cuda (line 1418) | void three_interpolate_backward_cuda(int b, int c, int n, int m,
  function three_nn_forward_cuda (line 1441) | void three_nn_forward_cuda(int b, int n, int m, const Tensor unknown,
  function tin_shift_forward_cuda (line 1456) | void tin_shift_forward_cuda(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_cuda (line 1460) | void tin_shift_backward_cuda(Tensor grad_output, Tensor shift,
  function hard_voxelize_forward_cuda (line 1498) | int hard_voxelize_forward_cuda(const at::Tensor &points, at::Tensor &vox...
  function nondeterministic_hard_voxelize_forward_cuda (line 1510) | int nondeterministic_hard_voxelize_forward_cuda(
  function dynamic_voxelize_forward_cuda (line 1520) | void dynamic_voxelize_forward_cuda(const at::Tensor &points, at::Tensor ...
  function rotated_feature_align_forward_cuda (line 1566) | void rotated_feature_align_forward_cuda(const Tensor features,
  function rotated_feature_align_backward_cuda (line 1574) | void rotated_feature_align_backward_cuda(const Tensor top_grad,
  function points_in_polygons_forward_cuda (line 1602) | void points_in_polygons_forward_cuda(const Tensor points, const Tensor p...
  function indice_maxpool_forward_cuda (line 1621) | torch::Tensor indice_maxpool_forward_cuda(torch::Tensor features,
  function indice_maxpool_backward_cuda (line 1642) | torch::Tensor indice_maxpool_backward_cuda(torch::Tensor features,
  function indice_conv_forward_cuda (line 1665) | torch::Tensor indice_conv_forward_cuda(torch::Tensor features,
  function indice_conv_backward_cuda (line 1689) | std::vector<torch::Tensor> indice_conv_backward_cuda(
  function fused_indice_conv_batchnorm_forward_cuda (line 1710) | torch::Tensor fused_indice_conv_batchnorm_forward_cuda(
  function min_area_polygons_cuda (line 1729) | void min_area_polygons_cuda(const Tensor pointsets, Tensor polygons) {
  function active_rotated_filter_forward_cuda (line 1745) | void active_rotated_filter_forward_cuda(const Tensor input,
  function active_rotated_filter_backward_cuda (line 1750) | void active_rotated_filter_backward_cuda(const Tensor grad_out,
  function convex_iou_cuda (line 1772) | void convex_iou_cuda(const Tensor pointsets, const Tensor polygons,
  function convex_giou_cuda (line 1777) | void convex_giou_cuda(const Tensor pointsets, const Tensor polygons,
  function Tensor (line 1795) | Tensor diff_iou_rotated_sort_vertices_forward_cuda(Tensor vertices, Tens...
  function chamfer_distance_forward_cuda (line 1815) | void chamfer_distance_forward_cuda(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward_cuda (line 1822) | void chamfer_distance_backward_cuda(const Tensor xyz1, const Tensor xyz2,
  function prroi_pool_forward_cuda (line 1856) | void prroi_pool_forward_cuda(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward_cuda (line 1863) | void prroi_pool_backward_cuda(Tensor grad_output, Tensor rois,
  function prroi_pool_coor_backward_cuda (line 1871) | void prroi_pool_coor_backward_cuda(Tensor output, Tensor grad_output,

FILE: mmcv/ops/csrc/pytorch/deform_conv.cpp
  function deformable_im2col_impl (line 5) | void deformable_im2col_impl(Tensor data_im, Tensor data_offset,
  function deformable_col2im_impl (line 19) | void deformable_col2im_impl(Tensor data_col, Tensor data_offset,
  function deformable_col2im_coord_impl (line 33) | void deformable_col2im_coord_impl(
  function deform_conv_shape_check (line 45) | void deform_conv_shape_check(at::Tensor input, at::Tensor offset,
  function deform_conv_forward (line 140) | void deform_conv_forward(Tensor input, Tensor weight, Tensor offset,
  function deform_conv_backward_input (line 262) | void deform_conv_backward_input(Tensor input, Tensor offset, Tensor grad...
  function deform_conv_backward_parameters (line 397) | void deform_conv_backward_parameters(Tensor input, Tensor offset,

FILE: mmcv/ops/csrc/pytorch/deform_roi_pool.cpp
  function deform_roi_pool_forward_impl (line 5) | void deform_roi_pool_forward_impl(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_impl (line 14) | void deform_roi_pool_backward_impl(Tensor grad_output, Tensor input,
  function deform_roi_pool_forward (line 25) | void deform_roi_pool_forward(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward (line 34) | void deform_roi_pool_backward(Tensor grad_output, Tensor input, Tensor r...

FILE: mmcv/ops/csrc/pytorch/diff_iou_rotated.cpp
  function Tensor (line 5) | Tensor diff_iou_rotated_sort_vertices_forward_impl(Tensor vertices, Tens...
  function Tensor (line 11) | Tensor diff_iou_rotated_sort_vertices_forward(Tensor vertices, Tensor mask,

FILE: mmcv/ops/csrc/pytorch/filtered_lrelu.cpp
  function filtered_lrelu_op_impl (line 4) | std::tuple<torch::Tensor, torch::Tensor, int> filtered_lrelu_op_impl(
  function filtered_lrelu (line 14) | std::tuple<torch::Tensor, torch::Tensor, int> filtered_lrelu(
  function filtered_lrelu_act_op_impl (line 24) | torch::Tensor filtered_lrelu_act_op_impl(torch::Tensor x, torch::Tensor si,
  function filtered_lrelu_act_ (line 32) | torch::Tensor filtered_lrelu_act_(torch::Tensor x, torch::Tensor si, int...

FILE: mmcv/ops/csrc/pytorch/focal_loss.cpp
  function sigmoid_focal_loss_forward_impl (line 19) | void sigmoid_focal_loss_forward_impl(Tensor input, Tensor target, Tensor...
  function sigmoid_focal_loss_backward_impl (line 25) | void sigmoid_focal_loss_backward_impl(Tensor input, Tensor target,
  function softmax_focal_loss_forward_impl (line 32) | void softmax_focal_loss_forward_impl(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_backward_impl (line 38) | void softmax_focal_loss_backward_impl(Tensor input, Tensor target,
  function sigmoid_focal_loss_forward_diopi (line 47) | void sigmoid_focal_loss_forward_diopi(Tensor input, Tensor target,
  function sigmoid_focal_loss_backward_diopi (line 87) | void sigmoid_focal_loss_backward_diopi(Tensor input, Tensor target,
  function sigmoid_focal_loss_forward (line 128) | void sigmoid_focal_loss_forward(Tensor input, Tensor target, Tensor weight,
  function sigmoid_focal_loss_backward (line 137) | void sigmoid_focal_loss_backward(Tensor input, Tensor target, Tensor wei...
  function softmax_focal_loss_forward (line 148) | void softmax_focal_loss_forward(Tensor input, Tensor target, Tensor weight,
  function softmax_focal_loss_backward (line 153) | void softmax_focal_loss_backward(Tensor input, Tensor target, Tensor wei...

FILE: mmcv/ops/csrc/pytorch/furthest_point_sample.cpp
  function furthest_point_sampling_forward_impl (line 7) | void furthest_point_sampling_forward_impl(Tensor points_tensor,
  function furthest_point_sampling_with_dist_forward_impl (line 14) | void furthest_point_sampling_with_dist_forward_impl(Tensor points_tensor,
  function furthest_point_sampling_forward (line 22) | void furthest_point_sampling_forward(Tensor points_tensor, Tensor temp_t...
  function furthest_point_sampling_with_dist_forward (line 28) | void furthest_point_sampling_with_dist_forward(Tensor points_tensor,

FILE: mmcv/ops/csrc/pytorch/fused_bias_leakyrelu.cpp
  function fused_bias_leakyrelu_op_impl (line 105) | torch::Tensor fused_bias_leakyrelu_op_impl(const torch::Tensor& input,
  function fused_bias_leakyrelu (line 113) | torch::Tensor fused_bias_leakyrelu(const torch::Tensor& input,

FILE: mmcv/ops/csrc/pytorch/fused_spconv_ops.cpp
  function fused_indice_conv_batchnorm_forward_impl (line 18) | torch::Tensor fused_indice_conv_batchnorm_forward_impl(
  function fused_indice_conv_batchnorm_forward (line 27) | torch::Tensor fused_indice_conv_batchnorm_forward(

FILE: mmcv/ops/csrc/pytorch/gather_points.cpp
  function gather_points_forward_impl (line 4) | void gather_points_forward_impl(int b, int c, int n, int npoints,
  function gather_points_backward_impl (line 11) | void gather_points_backward_impl(int b, int c, int n, int npoints,
  function gather_points_forward (line 18) | void gather_points_forward(Tensor points_tensor, Tensor idx_tensor,
  function gather_points_backward (line 25) | void gather_points_backward(Tensor grad_out_tensor, Tensor idx_tensor,

FILE: mmcv/ops/csrc/pytorch/group_points.cpp
  function group_points_forward_impl (line 8) | void group_points_forward_impl(int b, int c, int n, int npoints, int nsa...
  function group_points_backward_impl (line 15) | void group_points_backward_impl(int b, int c, int n, int npoints, int ns...
  function group_points_forward (line 22) | void group_points_forward(Tensor points_tensor, Tensor idx_tensor,
  function group_points_backward (line 29) | void group_points_backward(Tensor grad_out_tensor, Tensor idx_tensor,
  function stack_group_points_backward_impl (line 36) | void stack_group_points_backward_impl(int b, int c, int m, int n, int ns...
  function stack_group_points_backward (line 47) | void stack_group_points_backward(Tensor grad_out_tensor, Tensor idx_tensor,
  function stack_group_points_forward_impl (line 57) | void stack_group_points_forward_impl(int b, int c, int m, int nsample,
  function stack_group_points_forward (line 68) | void stack_group_points_forward(Tensor features_tensor,

FILE: mmcv/ops/csrc/pytorch/info.cpp
  function get_hiprt_version (line 9) | int get_hiprt_version() {
  function get_cudart_version (line 16) | int get_cudart_version() { return CUDART_VERSION; }
  function get_compiling_cuda_version (line 20) | std::string get_compiling_cuda_version() {
  function get_compiler_version (line 46) | std::string get_compiler_version() {

FILE: mmcv/ops/csrc/pytorch/iou3d.cpp
  function iou3d_boxes_overlap_bev_forward_impl (line 15) | void iou3d_boxes_overlap_bev_forward_impl(const int num_a, const Tensor ...
  function iou3d_nms3d_forward_impl (line 22) | void iou3d_nms3d_forward_impl(const Tensor boxes, Tensor &keep,
  function iou3d_nms3d_normal_forward_impl (line 28) | void iou3d_nms3d_normal_forward_impl(const Tensor boxes, Tensor &keep,
  function iou3d_boxes_overlap_bev_forward (line 35) | void iou3d_boxes_overlap_bev_forward(Tensor boxes_a, Tensor boxes_b,
  function iou3d_nms3d_forward (line 47) | void iou3d_nms3d_forward(Tensor boxes, Tensor keep, Tensor keep_num,
  function iou3d_nms3d_normal_forward (line 57) | void iou3d_nms3d_normal_forward(Tensor boxes, Tensor keep, Tensor keep_num,

FILE: mmcv/ops/csrc/pytorch/knn.cpp
  function knn_forward_impl (line 7) | void knn_forward_impl(int b, int n, int m, int nsample, const Tensor xyz,
  function knn_forward (line 13) | void knn_forward(Tensor xyz_tensor, Tensor new_xyz_tensor, Tensor idx_te...

FILE: mmcv/ops/csrc/pytorch/masked_conv2d.cpp
  function masked_im2col_forward_impl (line 5) | void masked_im2col_forward_impl(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_impl (line 13) | void masked_col2im_forward_impl(const Tensor col, const Tensor mask_h_idx,
  function masked_im2col_forward (line 20) | void masked_im2col_forward(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward (line 28) | void masked_col2im_forward(const Tensor col, const Tensor mask_h_idx,

FILE: mmcv/ops/csrc/pytorch/min_area_polygons.cpp
  function min_area_polygons_impl (line 5) | void min_area_polygons_impl(const Tensor pointsets, Tensor polygons) {
  function min_area_polygons (line 9) | void min_area_polygons(const Tensor pointsets, Tensor polygons) {

FILE: mmcv/ops/csrc/pytorch/mlu/ball_query_mlu.cpp
  function ball_query_forward_mlu (line 14) | void ball_query_forward_mlu(int b, int n, int m, float min_radius,

FILE: mmcv/ops/csrc/pytorch/mlu/bbox_overlaps_mlu.cpp
  function bbox_overlaps_mlu (line 15) | void bbox_overlaps_mlu(const Tensor bboxes1, const Tensor bboxes2, Tenso...

FILE: mmcv/ops/csrc/pytorch/mlu/box_iou_rotated.cpp
  function BoxIouRotatedMLUKernelLauncher (line 14) | void BoxIouRotatedMLUKernelLauncher(const Tensor boxes1, const Tensor bo...
  function box_iou_rotated_mlu (line 46) | void box_iou_rotated_mlu(const Tensor boxes1, const Tensor boxes2, Tenso...

FILE: mmcv/ops/csrc/pytorch/mlu/carafe_mlu.cpp
  function CARAFEForwardMLUKernelLauncher (line 14) | void CARAFEForwardMLUKernelLauncher(const Tensor input, const Tensor mask,
  function CARAFEBackwardMLUKernelLauncher (line 89) | void CARAFEBackwardMLUKernelLauncher(
  function carafe_forward_mlu (line 179) | void carafe_forward_mlu(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_mlu (line 186) | void carafe_backward_mlu(Tensor top_grad, Tensor rfeatures, Tensor masks,

FILE: mmcv/ops/csrc/pytorch/mlu/deform_roi_pool_mlu.cpp
  function DeformRoIPoolForwardMLUKernelLauncher (line 14) | void DeformRoIPoolForwardMLUKernelLauncher(Tensor input, Tensor rois,
  function DeformRoIPoolBackwardMLUKernelLauncher (line 61) | void DeformRoIPoolBackwardMLUKernelLauncher(
  function deform_roi_pool_forward_mlu (line 125) | void deform_roi_pool_forward_mlu(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_mlu (line 134) | void deform_roi_pool_backward_mlu(Tensor grad_output, Tensor input, Tens...

FILE: mmcv/ops/csrc/pytorch/mlu/diff_iou_rotated_mlu.cpp
  function Tensor (line 14) | Tensor diff_iou_rotated_sort_vertices_forward_mlu(Tensor vertices, Tenso...

FILE: mmcv/ops/csrc/pytorch/mlu/focal_loss_sigmoid_mlu.cpp
  function sigmoid_focal_loss_forward_mlu (line 17) | void sigmoid_focal_loss_forward_mlu(Tensor input, Tensor target, Tensor ...
  function sigmoid_focal_loss_backward_mlu (line 91) | void sigmoid_focal_loss_backward_mlu(Tensor input, Tensor target, Tensor...

FILE: mmcv/ops/csrc/pytorch/mlu/iou3d_mlu.cpp
  function IoU3DNMS3DMLUKernelLauncher (line 15) | void IoU3DNMS3DMLUKernelLauncher(Tensor boxes, Tensor &keep, Tensor &kee...
  function iou3d_nms3d_forward_mlu (line 72) | void iou3d_nms3d_forward_mlu(const Tensor boxes, Tensor &keep, Tensor &k...

FILE: mmcv/ops/csrc/pytorch/mlu/masked_conv2d_mlu.cpp
  function policyFunc (line 31) | static void policyFunc(const int mask_cnt, cnrtDim3_t *k_dim,
  function MaskedIm2colForwardMLUKernelLauncher (line 43) | void MaskedIm2colForwardMLUKernelLauncher(const Tensor im,
  function MaskedCol2imForwardMLUKernelLauncher (line 123) | void MaskedCol2imForwardMLUKernelLauncher(const Tensor col,
  function masked_im2col_forward_mlu (line 195) | void masked_im2col_forward_mlu(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_mlu (line 205) | void masked_col2im_forward_mlu(const Tensor col, const Tensor mask_h_idx,

FILE: mmcv/ops/csrc/pytorch/mlu/mlu_common_helper.cpp
  function mluOpDataType_t (line 15) | mluOpDataType_t getMluOpDataType(const caffe2::TypeMeta& data_type) {
  function mluOpTensorLayout_t (line 39) | mluOpTensorLayout_t getMluOpSuggestLayout(const at::Tensor& input) {
  function mluOpReduceMode_t (line 59) | mluOpReduceMode_t getMluOpReduceMode(const reduce_t reduce_type) {
  function mluOpHandle_t (line 135) | mluOpHandle_t mluOpGetCurrentHandle(c10::DeviceIndex device_index) {

FILE: mmcv/ops/csrc/pytorch/mlu/mlu_common_helper.h
  function reduce_t (line 48) | enum class reduce_t { SUM = 0, MEAN = 1, MAX = 2 };

FILE: mmcv/ops/csrc/pytorch/mlu/ms_deform_attn_mlu.cpp
  function Tensor (line 16) | Tensor MsDeformAttnForwardLauncher(const Tensor& value,
  function MsDeformAttnBackwardLauncher (line 49) | void MsDeformAttnBackwardLauncher(
  function Tensor (line 90) | Tensor ms_deform_attn_mlu_forward(const Tensor& value,
  function ms_deform_attn_mlu_backward (line 100) | void ms_deform_attn_mlu_backward(

FILE: mmcv/ops/csrc/pytorch/mlu/nms_mlu.cpp
  function Tensor (line 15) | Tensor NMSMLUKernelLauncher(Tensor boxes, Tensor scores, float iou_thres...
  function Tensor (line 81) | Tensor nms_mlu(Tensor boxes, Tensor scores, float iou_threshold, int off...

FILE: mmcv/ops/csrc/pytorch/mlu/nms_rotated_mlu.cpp
  function Tensor (line 14) | Tensor nms_rotated_mlu(Tensor boxes, Tensor scores, float iou_threshold) {

FILE: mmcv/ops/csrc/pytorch/mlu/psamask_mlu.cpp
  function PSAMaskForwardMLUKernelLauncher (line 14) | void PSAMaskForwardMLUKernelLauncher(const int psa_type, const Tensor x,
  function PSAMaskBackwardMLUKernelLauncher (line 44) | void PSAMaskBackwardMLUKernelLauncher(const int psa_type, const Tensor dy,
  function psamask_forward_mlu (line 77) | void psamask_forward_mlu(const int psa_type, const Tensor input, Tensor ...
  function psamask_backward_mlu (line 87) | void psamask_backward_mlu(const int psa_type, const Tensor grad_output,

FILE: mmcv/ops/csrc/pytorch/mlu/roi_align_mlu.cpp
  function ROIAlignForwardMLUKernelLauncher (line 14) | void ROIAlignForwardMLUKernelLauncher(Tensor input, Tensor rois, Tensor ...
  function ROIAlignBackwardMLUKernelLauncher (line 83) | void ROIAlignBackwardMLUKernelLauncher(Tensor grad, Tensor rois,
  function roi_align_forward_mlu (line 153) | void roi_align_forward_mlu(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_mlu (line 162) | void roi_align_backward_mlu(Tensor grad_output, Tensor rois, Tensor argm...

FILE: mmcv/ops/csrc/pytorch/mlu/roi_align_rotated_mlu.cpp
  function ROIAlignRotatedForwardMLUKernelLauncher (line 14) | void ROIAlignRotatedForwardMLUKernelLauncher(Tensor input, Tensor rois,
  function ROIAlignRotatedBackwardMLUKernelLauncher (line 51) | void ROIAlignRotatedBackwardMLUKernelLauncher(
  function roi_align_rotated_forward_mlu (line 86) | void roi_align_rotated_forward_mlu(Tensor input, Tensor rois, Tensor out...
  function roi_align_rotated_backward_mlu (line 95) | void roi_align_rotated_backward_mlu(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/mlu/roi_pool_mlu.cpp
  function policyFuncForward (line 34) | static void policyFuncForward(const int bin_num, cnrtDim3_t *k_dim,
  function ROIPoolForwardMLUKernelLauncher (line 45) | void ROIPoolForwardMLUKernelLauncher(Tensor input, Tensor rois, Tensor o...
  function policyFuncBackward (line 140) | static void policyFuncBackward(cnrtDim3_t *k_dim, cnrtFunctionType_t *k_...
  function ROIPoolBackwardMLUKernelLauncher (line 147) | void ROIPoolBackwardMLUKernelLauncher(Tensor grad_output, Tensor rois,
  function roi_pool_forward_mlu (line 252) | void roi_pool_forward_mlu(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_mlu (line 259) | void roi_pool_backward_mlu(Tensor grad_output, Tensor rois, Tensor argmax,

FILE: mmcv/ops/csrc/pytorch/mlu/roiaware_pool3d_mlu.cpp
  function RoiawarePool3dForwardMLUKernelLauncher (line 14) | void RoiawarePool3dForwardMLUKernelLauncher(
  function roiaware_pool3d_forward_mlu (line 81) | void roiaware_pool3d_forward_mlu(int boxes_num, int pts_num, int channels,
  function RoiawarePool3dBackwardMLUKernelLauncher (line 103) | void RoiawarePool3dBackwardMLUKernelLauncher(
  function roiaware_pool3d_backward_mlu (line 145) | void roiaware_pool3d_backward_mlu(int boxes_num, int out_x, int out_y,

FILE: mmcv/ops/csrc/pytorch/mlu/roipoint_pool3d_mlu.cpp
  function RoIPointPool3dForwardMLUKernelLauncher (line 14) | void RoIPointPool3dForwardMLUKernelLauncher(
  function roipoint_pool3d_forward_mlu (line 131) | void roipoint_pool3d_forward_mlu(int batch_size, int pts_num, int boxes_...

FILE: mmcv/ops/csrc/pytorch/mlu/rotated_feature_align_mlu.cpp
  function RotatedFeatureAlignForwardMLUKernelLauncher (line 14) | void RotatedFeatureAlignForwardMLUKernelLauncher(const Tensor features,
  function RotatedFeatureAlignBackwardMLUKernelLauncher (line 50) | void RotatedFeatureAlignBackwardMLUKernelLauncher(const Tensor top_grad,
  function rotated_feature_align_forward_mlu (line 86) | void rotated_feature_align_forward_mlu(const Tensor features,
  function rotated_feature_align_backward_mlu (line 94) | void rotated_feature_align_backward_mlu(const Tensor top_grad,

FILE: mmcv/ops/csrc/pytorch/mlu/scatter_points_mlu.cpp
  function dynamic_point_to_voxel_forward_mlu (line 14) | std::vector<Tensor> dynamic_point_to_voxel_forward_mlu(
  function dynamic_point_to_voxel_backward_mlu (line 76) | void dynamic_point_to_voxel_backward_mlu(

FILE: mmcv/ops/csrc/pytorch/mlu/sparse_conv_mlu.cpp
  function GetIndicePairsForwardMLUKernelLauncher (line 21) | std::vector<torch::Tensor> GetIndicePairsForwardMLUKernelLauncher(
  function IndiceConvForwardMLUKernelLauncher (line 146) | torch::Tensor IndiceConvForwardMLUKernelLauncher(
  function IndiceConvBackwardMLUKernelLauncher (line 239) | std::vector<torch::Tensor> IndiceConvBackwardMLUKernelLauncher(
  function indice_conv_forward_mlu (line 392) | torch::Tensor indice_conv_forward_mlu(torch::Tensor features,
  function indice_conv_backward_mlu (line 402) | std::vector<torch::Tensor> indice_conv_backward_mlu(

FILE: mmcv/ops/csrc/pytorch/mlu/three_nn_mlu.cpp
  function ThreeNNMLUKernelLauncher (line 14) | void ThreeNNMLUKernelLauncher(int b, int n, int m, const Tensor unknown,
  function three_nn_forward_mlu (line 55) | void three_nn_forward_mlu(int b, int n, int m, const Tensor unknown,

FILE: mmcv/ops/csrc/pytorch/mlu/tin_shift_mlu.cpp
  function TINShiftForwardMLUKernelLauncher (line 14) | void TINShiftForwardMLUKernelLauncher(Tensor input, Tensor shift,
  function TINShiftBackwardMLUKernelLauncher (line 67) | void TINShiftBackwardMLUKernelLauncher(Tensor grad_output, Tensor shift,
  function tin_shift_forward_mlu (line 123) | void tin_shift_forward_mlu(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_mlu (line 127) | void tin_shift_backward_mlu(Tensor grad_output, Tensor shift,

FILE: mmcv/ops/csrc/pytorch/mlu/voxelization_mlu.cpp
  function HardVoxelizeForwardMLUKernelLauncher (line 27) | int HardVoxelizeForwardMLUKernelLauncher(
  function hard_voxelize_forward_mlu (line 78) | int hard_voxelize_forward_mlu(const at::Tensor &points, at::Tensor &voxels,

FILE: mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp
  function modulated_deformable_im2col_impl (line 19) | void modulated_deformable_im2col_impl(
  function modulated_deformable_col2im_impl (line 33) | void modulated_deformable_col2im_impl(
  function modulated_deformable_col2im_coord_impl (line 47) | void modulated_deformable_col2im_coord_impl(
  function modulated_deform_conv_forward_fallthrough (line 62) | void modulated_deform_conv_forward_fallthrough(
  function modulated_deform_conv_backward_fallthrough (line 140) | void modulated_deform_conv_backward_fallthrough(
  function modulated_deform_conv_forward_diopi (line 254) | void modulated_deform_conv_forward_diopi(
  function modulated_deform_conv_backward_diopi (line 312) | void modulated_deform_conv_backward_diopi(
  function modulated_deform_conv_forward (line 395) | void modulated_deform_conv_forward(
  function modulated_deform_conv_backward (line 414) | void modulated_deform_conv_backward(

FILE: mmcv/ops/csrc/pytorch/ms_deform_attn.cpp
  function Tensor (line 15) | Tensor ms_deform_attn_impl_forward(const Tensor &value,
  function ms_deform_attn_impl_backward (line 26) | void ms_deform_attn_impl_backward(
  function Tensor (line 38) | Tensor ms_deform_attn_forward(const Tensor &value, const Tensor &spatial...
  function ms_deform_attn_backward (line 48) | void ms_deform_attn_backward(const Tensor &value, const Tensor &spatial_...

FILE: mmcv/ops/csrc/pytorch/musa/musabind.cpp
  function assign_score_withk_forward_musa (line 14) | void assign_score_withk_forward_musa(int B, int N0, int N1, int M, int K...
  function assign_score_withk_backward_musa (line 23) | void assign_score_withk_backward_musa(
  function ball_query_forward_musa (line 55) | void ball_query_forward_musa(int b, int n, int m, float min_radius,
  function stack_ball_query_forward_musa (line 76) | void stack_ball_query_forward_musa(float max_radius, int nsample,
  function bbox_overlaps_musa (line 97) | void bbox_overlaps_musa(const Tensor bboxes1, const Tensor bboxes2, Tens...
  function border_align_forward_musa (line 117) | void border_align_forward_musa(const Tensor &input, const Tensor &boxes,
  function border_align_backward_musa (line 124) | void border_align_backward_musa(const Tensor &grad_output, const Tensor ...
  function carafe_forward_musa (line 173) | void carafe_forward_musa(Tensor features, Tensor masks, Tensor rfeatures,
  function carafe_backward_musa (line 181) | void carafe_backward_musa(Tensor top_grad, Tensor rfeatures, Tensor masks,
  function carafe_naive_forward_musa (line 217) | void carafe_naive_forward_musa(Tensor features, Tensor masks, Tensor out...
  function carafe_naive_backward_musa (line 224) | void carafe_naive_backward_musa(Tensor top_grad, Tensor features, Tensor...
  function correlation_forward_musa (line 261) | void correlation_forward_musa(Tensor input1, Tensor input2, Tensor output,
  function correlation_backward_musa (line 271) | void correlation_backward_musa(Tensor grad_output, Tensor input1, Tensor...
  function deform_roi_pool_forward_musa (line 366) | void deform_roi_pool_forward_musa(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_musa (line 375) | void deform_roi_pool_backward_musa(Tensor grad_output, Tensor input,
  function sigmoid_focal_loss_forward_musa (line 425) | void sigmoid_focal_loss_forward_musa(Tensor input, Tensor target, Tensor...
  function sigmoid_focal_loss_backward_musa (line 431) | void sigmoid_focal_loss_backward_musa(Tensor input, Tensor target,
  function softmax_focal_loss_forward_musa (line 438) | void softmax_focal_loss_forward_musa(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_backward_musa (line 444) | void softmax_focal_loss_backward_musa(Tensor input, Tensor target,
  function furthest_point_sampling_forward_musa (line 483) | void furthest_point_sampling_forward_musa(Tensor points_tensor,
  function furthest_point_sampling_with_dist_forward_musa (line 492) | void furthest_point_sampling_with_dist_forward_musa(Tensor points_tensor,
  function gather_points_forward_musa (line 563) | void gather_points_forward_musa(int b, int c, int n, int npoints,
  function gather_points_backward_musa (line 569) | void gather_points_backward_musa(int b, int c, int n, int npoints,
  function group_points_forward_musa (line 598) | void group_points_forward_musa(int b, int c, int n, int npoints, int nsa...
  function group_points_backward_musa (line 605) | void group_points_backward_musa(int b, int c, int n, int npoints, int ns...
  function stack_group_points_forward_musa (line 634) | void stack_group_points_forward_musa(int b, int c, int m, int nsample,
  function stack_group_points_backward_musa (line 645) | void stack_group_points_backward_musa(int b, int c, int m, int n, int ns...
  function iou3d_boxes_overlap_bev_forward_musa (line 689) | void iou3d_boxes_overlap_bev_forward_musa(const int num_a, const Tensor ...
  function iou3d_nms3d_forward_musa (line 696) | void iou3d_nms3d_forward_musa(const Tensor boxes, Tensor &keep,
  function iou3d_nms3d_normal_forward_musa (line 702) | void iou3d_nms3d_normal_forward_musa(const Tensor boxes, Tensor &keep,
  function knn_forward_musa (line 730) | void knn_forward_musa(int b, int n, int m, int nsample, const Tensor xyz,
  function masked_im2col_forward_musa (line 752) | void masked_im2col_forward_musa(const Tensor im, const Tensor mask_h_idx,
  function masked_col2im_forward_musa (line 762) | void masked_col2im_forward_musa(const Tensor col, const Tensor mask_h_idx,
  function Tensor (line 876) | Tensor nms_musa(Tensor boxes, Tensor scores, float iou_threshold, int of...
  function points_in_boxes_part_forward_musa (line 893) | void points_in_boxes_part_forward_musa(int batch_size, int boxes_num,
  function points_in_boxes_all_forward_musa (line 901) | void points_in_boxes_all_forward_musa(int batch_size, int boxes_num,
  function psamask_forward_musa (line 935) | void psamask_forward_musa(const int psa_type, const Tensor input, Tensor...
  function psamask_backward_musa (line 945) | void psamask_backward_musa(const int psa_type, const Tensor grad_output,
  function roi_align_forward_musa (line 982) | void roi_align_forward_musa(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_musa (line 992) | void roi_align_backward_musa(Tensor grad_output, Tensor rois, Tensor arg...
  function roi_align_rotated_forward_musa (line 1029) | void roi_align_rotated_forward_musa(Tensor input, Tensor rois, Tensor ou...
  function roi_align_rotated_backward_musa (line 1050) | void roi_align_rotated_backward_musa(Tensor top_grad, Tensor rois,
  function riroi_align_rotated_forward_musa (line 1100) | void riroi_align_rotated_forward_musa(Tensor features, Tensor rois,
  function riroi_align_rotated_backward_musa (line 1122) | void riroi_align_rotated_backward_musa(Tensor top_grad, Tensor rois,
  function roiaware_pool3d_forward_musa (line 1172) | void roiaware_pool3d_forward_musa(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_musa (line 1184) | void roiaware_pool3d_backward_musa(int boxes_num, int out_x, int out_y,
  function roipoint_pool3d_forward_musa (line 1219) | void roipoint_pool3d_forward_musa(int batch_size, int pts_num, int boxes...
  function roi_pool_forward_musa (line 1248) | void roi_pool_forward_musa(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_musa (line 1255) | void roi_pool_backward_musa(Tensor grad_output, Tensor rois, Tensor argmax,
  function dynamic_point_to_voxel_forward_musa (line 1283) | std::vector<torch::Tensor> dynamic_point_to_voxel_forward_musa(
  function dynamic_point_to_voxel_backward_musa (line 1290) | void dynamic_point_to_voxel_backward_musa(
  function sync_bn_forward_mean_musa (line 1338) | void sync_bn_forward_mean_musa(const Tensor input, Tensor mean) {
  function sync_bn_forward_var_musa (line 1342) | void sync_bn_forward_var_musa(const Tensor input, const Tensor mean,
  function sync_bn_forward_output_musa (line 1347) | void sync_bn_forward_output_musa(const Tensor input, const Tensor mean,
  function sync_bn_backward_param_musa (line 1358) | void sync_bn_backward_param_musa(const Tensor grad_output, const Tensor ...
  function sync_bn_backward_data_musa (line 1364) | void sync_bn_backward_data_musa(const Tensor grad_output, const Tensor w...
  function three_interpolate_forward_musa (line 1413) | void three_interpolate_forward_musa(int b, int c, int m, int n,
  function three_interpolate_backward_musa (line 1420) | void three_interpolate_backward_musa(int b, int c, int n, int m,
  function three_nn_forward_musa (line 1443) | void three_nn_forward_musa(int b, int n, int m, const Tensor unknown,
  function tin_shift_forward_musa (line 1458) | void tin_shift_forward_musa(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_musa (line 1462) | void tin_shift_backward_musa(Tensor grad_output, Tensor shift,
  function hard_voxelize_forward_musa (line 1502) | int hard_voxelize_forward_musa(const at::Tensor &points, at::Tensor &vox...
  function nondeterministic_hard_voxelize_forward_musa (line 1514) | int nondeterministic_hard_voxelize_forward_musa(
  function dynamic_voxelize_forward_musa (line 1524) | void dynamic_voxelize_forward_musa(const at::Tensor &points, at::Tensor ...
  function rotated_feature_align_forward_musa (line 1570) | void rotated_feature_align_forward_musa(const Tensor features,
  function rotated_feature_align_backward_musa (line 1578) | void rotated_feature_align_backward_musa(const Tensor top_grad,
  function points_in_polygons_forward_musa (line 1606) | void points_in_polygons_forward_musa(const Tensor points, const Tensor p...
  function indice_maxpool_forward_musa (line 1625) | torch::Tensor indice_maxpool_forward_musa(torch::Tensor features,
  function indice_maxpool_backward_musa (line 1646) | torch::Tensor indice_maxpool_backward_musa(torch::Tensor features,
  function indice_conv_forward_musa (line 1669) | torch::Tensor indice_conv_forward_musa(torch::Tensor features,
  function indice_conv_backward_musa (line 1693) | std::vector<torch::Tensor> indice_conv_backward_musa(
  function fused_indice_conv_batchnorm_forward_musa (line 1714) | torch::Tensor fused_indice_conv_batchnorm_forward_musa(
  function min_area_polygons_musa (line 1733) | void min_area_polygons_musa(const Tensor pointsets, Tensor polygons) {
  function active_rotated_filter_forward_musa (line 1749) | void active_rotated_filter_forward_musa(const Tensor input,
  function active_rotated_filter_backward_musa (line 1754) | void active_rotated_filter_backward_musa(const Tensor grad_out,
  function convex_iou_musa (line 1776) | void convex_iou_musa(const Tensor pointsets, const Tensor polygons,
  function convex_giou_musa (line 1781) | void convex_giou_musa(const Tensor pointsets, const Tensor polygons,
  function Tensor (line 1799) | Tensor diff_iou_rotated_sort_vertices_forward_musa(Tensor vertices, Tens...
  function chamfer_distance_forward_musa (line 1822) | void chamfer_distance_forward_musa(const Tensor xyz1, const Tensor xyz2,
  function chamfer_distance_backward_musa (line 1829) | void chamfer_distance_backward_musa(const Tensor xyz1, const Tensor xyz2,
  function prroi_pool_forward_musa (line 1864) | void prroi_pool_forward_musa(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward_musa (line 1871) | void prroi_pool_backward_musa(Tensor grad_output, Tensor rois,
  function prroi_pool_coor_backward_musa (line 1879) | void prroi_pool_coor_backward_musa(Tensor output, Tensor grad_output,

FILE: mmcv/ops/csrc/pytorch/nms.cpp
  function Tensor (line 20) | Tensor nms_impl(Tensor boxes, Tensor scores, float iou_threshold, int of...
  function Tensor (line 24) | Tensor softnms_impl(Tensor boxes, Tensor scores, Tensor dets,
  function nms_match_impl (line 31) | std::vector<std::vector<int> > nms_match_impl(Tensor dets,
  function Tensor (line 37) | Tensor nms_diopi(Tensor boxes, Tensor scores, float iou_threshold, int o...
  function Tensor (line 76) | Tensor nms(Tensor boxes, Tensor scores, float iou_threshold, int offset) {
  function Tensor (line 84) | Tensor softnms(Tensor boxes, Tensor scores, Tensor dets, float iou_thres...
  function nms_match (line 90) | std::vector<std::vector<int> > nms_match(Tensor dets, float iou_threshol...

FILE: mmcv/ops/csrc/pytorch/nms_quadri.cpp
  function Tensor (line 20) | Tensor nms_quadri(const Tensor dets, const Tensor scores, const Tensor o...

FILE: mmcv/ops/csrc/pytorch/nms_rotated.cpp
  function Tensor (line 34) | Tensor nms_rotated(const Tensor dets, const Tensor scores, const Tensor ...

FILE: mmcv/ops/csrc/pytorch/npu/active_rotated_filter_npu.cpp
  function active_rotated_filter_forward_npu (line 12) | void active_rotated_filter_forward_npu(const Tensor input, const Tensor ...
  function active_rotated_filter_backward_npu (line 22) | void active_rotated_filter_backward_npu(const Tensor grad_out,

FILE: mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp
  function assign_score_withk_forward_npu (line 6) | void assign_score_withk_forward_npu(int B, int N0, int N1, int M, int K,...
  function assign_score_withk_backward_npu (line 25) | void assign_score_withk_backward_npu(

FILE: mmcv/ops/csrc/pytorch/npu/ball_query_npu.cpp
  function ball_query_forward_npu (line 6) | void ball_query_forward_npu(int b, int n, int m, float min_radius,

FILE: mmcv/ops/csrc/pytorch/npu/bbox_overlaps_npu.cpp
  function bbox_overlaps_npu (line 10) | void bbox_overlaps_npu(const Tensor bboxes1, const Tensor bboxes2, Tenso...

FILE: mmcv/ops/csrc/pytorch/npu/border_align_npu.cpp
  function border_align_forward_npu (line 10) | void border_align_forward_npu(const Tensor &input, const Tensor &boxes,
  function border_align_backward_npu (line 46) | void border_align_backward_npu(const Tensor &grad_output, const Tensor &...

FILE: mmcv/ops/csrc/pytorch/npu/box_iou_quadri_npu.cpp
  function box_iou_quadri_npu (line 9) | void box_iou_quadri_npu(const Tensor boxes1, const Tensor boxes2, Tensor...

FILE: mmcv/ops/csrc/pytorch/npu/box_iou_rotated_npu.cpp
  function box_iou_rotated_npu (line 9) | void box_iou_rotated_npu(const Tensor boxes1, const Tensor boxes2, Tenso...

FILE: mmcv/ops/csrc/pytorch/npu/boxes_overlap_bev_npu.cpp
  function iou3d_boxes_overlap_bev_forward_npu (line 15) | void iou3d_boxes_overlap_bev_forward_npu(const int num_a, const Tensor b...

FILE: mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp
  function chamfer_distance_forward_npu (line 6) | void chamfer_distance_forward_npu(Tensor XYZ1, Tensor XYZ2, Tensor dist1,
  function chamfer_distance_backward_npu (line 23) | void chamfer_distance_backward_npu(Tensor xyz1, Tensor xyz2, Tensor idx1,

FILE: mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp
  function deform_roi_pool_forward_npu (line 18) | void deform_roi_pool_forward_npu(Tensor input, Tensor rois, Tensor offset,
  function deform_roi_pool_backward_npu (line 38) | void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tens...

FILE: mmcv/ops/csrc/pytorch/npu/diff_iou_rotated_npu.cpp
  function Tensor (line 5) | Tensor diff_iou_rotated_sort_vertices_npu(Tensor vertices, Tensor mask,

FILE: mmcv/ops/csrc/pytorch/npu/focal_loss_npu.cpp
  function sigmoid_focal_loss_forward_npu (line 5) | void sigmoid_focal_loss_forward_npu(Tensor input, Tensor target, Tensor ...
  function sigmoid_focal_loss_backward_npu (line 40) | void sigmoid_focal_loss_backward_npu(Tensor input, Tensor target, Tensor...
  function softmax_focal_loss_forward_npu (line 77) | void softmax_focal_loss_forward_npu(Tensor input, Tensor target, Tensor ...
  function softmax_focal_loss_backward_npu (line 126) | void softmax_focal_loss_backward_npu(Tensor input, Tensor target, Tensor...

FILE: mmcv/ops/csrc/pytorch/npu/furthest_point_sample_npu.cpp
  function furthest_point_sampling_forward_npu (line 6) | void furthest_point_sampling_forward_npu(Tensor points_tensor,

FILE: mmcv/ops/csrc/pytorch/npu/furthest_point_sampling_with_dist_npu.cpp
  function furthest_point_sampling_with_dist_npu (line 5) | void furthest_point_sampling_with_dist_npu(Tensor points_tensor,

FILE: mmcv/ops/csrc/pytorch/npu/fused_bias_leakyrelu_npu.cpp
  function Tensor (line 11) | Tensor fused_bias_leakyrelu_npu(const Tensor &input, const Tensor &bias,

FILE: mmcv/ops/csrc/pytorch/npu/gather_points_npu.cpp
  function gather_points_forward_npu (line 6) | void gather_points_forward_npu(int b, int c, int n, int npoints,
  function gather_points_backward_npu (line 24) | void gather_points_backward_npu(int b, int c, int n, int npoints,

FILE: mmcv/ops/csrc/pytorch/npu/group_points_npu.cpp
  function group_points_forward_npu (line 6) | void group_points_forward_npu(int b, int c, int n, int npoints, int nsam...
  function group_points_backward_npu (line 41) | void group_points_backward_npu(int b, int c, int n, int npoints, int nsa...

FILE: mmcv/ops/csrc/pytorch/npu/knn_npu.cpp
  function knn_forward_npu (line 8) | void knn_forward_npu(int b, int n, int m, int nsample, const Tensor xyz,

FILE: mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp
  function check_support (line 13) | void check_support(const Tensor &value, const Tensor &attention_weights) {
  function Tensor (line 28) | Tensor ms_deform_attn_forward_npu(const Tensor &value,
  function ms_deform_attn_backward_npu (line 79) | void ms_deform_attn_backward_npu(

FILE: mmcv/ops/csrc/pytorch/npu/nms3d_normal_npu.cpp
  function iou3d_nms3d_normal_forward_npu (line 5) | void iou3d_nms3d_normal_forward_npu(const Tensor boxes, Tensor &keep,

FILE: mmcv/ops/csrc/pytorch/npu/nms3d_npu.cpp
  function iou3d_nms3d_forward_npu (line 8) | void iou3d_nms3d_forward_npu(const Tensor boxes, Tensor &keep, Tensor &n...

FILE: mmcv/ops/csrc/pytorch/npu/nms_npu.cpp
  function Tensor (line 7) | Tensor nms_npu(Tensor boxes, Tensor scores, float iou_threshold, int off...

FILE: mmcv/ops/csrc/pytorch/npu/nms_rotated_npu.cpp
  function Tensor (line 6) | Tensor nms_rotated_npu(const Tensor dets, const Tensor scores,

FILE: mmcv/ops/csrc/pytorch/npu/points_in_box_npu.cpp
  function points_in_boxes_part_forward_impl_npu (line 6) | void points_in_boxes_part_forward_impl_npu(int batch_size, int boxes_num,

FILE: mmcv/ops/csrc/pytorch/npu/points_in_box_npu_all.cpp
  function points_in_boxes_all_forward_impl_npu (line 6) | void points_in_boxes_all_forward_impl_npu(int batch_size, int boxes_num,

FILE: mmcv/ops/csrc/pytorch/npu/points_in_polygons_npu.cpp
  function points_in_polygons_npu (line 8) | void points_in_polygons_npu(const Tensor points, Tensor polygons, Tensor...

FILE: mmcv/ops/csrc/pytorch/npu/psa_mask_npu.cpp
  function psamask_forward_npu (line 6) | void psamask_forward_npu(const int psa_type, const Tensor x, Tensor y,
  function psamask_backward_npu (line 40) | void psamask_backward_npu(const int psa_type, const Tensor y_grad,

FILE: mmcv/ops/csrc/pytorch/npu/roi_align_npu.cpp
  function roi_align_forward_npu (line 7) | void roi_align_forward_npu(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_npu (line 34) | void roi_align_backward_npu(Tensor grad_output, Tensor rois, Tensor argm...

FILE: mmcv/ops/csrc/pytorch/npu/roi_align_rotated_npu.cpp
  function roi_align_rotated_forward_npu (line 7) | void roi_align_rotated_forward_npu(Tensor input, Tensor rois, Tensor out...
  function roi_align_rotated_backward_npu (line 36) | void roi_align_rotated_backward_npu(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
  function roi_pool_forward_npu (line 6) | void roi_pool_forward_npu(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_npu (line 47) | void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,

FILE: mmcv/ops/csrc/pytorch/npu/roiaware_pool3d_npu.cpp
  function roiaware_pool3d_forward_npu (line 5) | void roiaware_pool3d_forward_npu(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_npu (line 35) | void roiaware_pool3d_backward_npu(int boxes_num, int out_x, int out_y,

FILE: mmcv/ops/csrc/pytorch/npu/roipoint_pool3d_forward.cpp
  function roipoint_pool3d_forward_impl_npu (line 7) | void roipoint_pool3d_forward_impl_npu(int batch_size, int pts_num,

FILE: mmcv/ops/csrc/pytorch/npu/rotated_feature_align_npu.cpp
  function rotated_feature_align_forward_npu (line 16) | void rotated_feature_align_forward_npu(const Tensor features,
  function rotated_feature_align_backward_npu (line 32) | void rotated_feature_align_backward_npu(const Tensor top_grad,

FILE: mmcv/ops/csrc/pytorch/npu/stack_ball_query_npu.cpp
  function stack_ball_query_forward_npu (line 6) | void stack_ball_query_forward_npu(float max_radius, int nsample,

FILE: mmcv/ops/csrc/pytorch/npu/stack_group_points_npu.cpp
  function stack_group_points_forward_npu (line 6) | void stack_group_points_forward_npu(int b, int c, int n, int nsample,

FILE: mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp
  function three_interpolate_forward_npu (line 8) | void three_interpolate_forward_npu(int b, int c, int m, int n,
  function three_interpolate_backward_npu (line 30) | void three_interpolate_backward_npu(int b, int c, int n, int m,

FILE: mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp
  function three_nn_forward_npu (line 8) | void three_nn_forward_npu(int b, int n, int m, const Tensor unknown,

FILE: mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp
  function hard_voxelize_forward_npu (line 19) | int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels,
  function dynamic_voxelize_forward_npu (line 61) | void dynamic_voxelize_forward_npu(const at::Tensor &points, at::Tensor &...

FILE: mmcv/ops/csrc/pytorch/pixel_group.cpp
  function pixel_group_impl (line 7) | std::vector<std::vector<float>> pixel_group_impl(
  function pixel_group (line 15) | std::vector<std::vector<float>> pixel_group(

FILE: mmcv/ops/csrc/pytorch/points_in_boxes.cpp
  function points_in_boxes_part_forward_impl (line 4) | void points_in_boxes_part_forward_impl(int batch_size, int boxes_num,
  function points_in_boxes_all_forward_impl (line 12) | void points_in_boxes_all_forward_impl(int batch_size, int boxes_num,
  function points_in_boxes_part_forward (line 20) | void points_in_boxes_part_forward(Tensor boxes_tensor, Tensor pts_tensor,
  function points_in_boxes_all_forward (line 34) | void points_in_boxes_all_forward(Tensor boxes_tensor, Tensor pts_tensor,

FILE: mmcv/ops/csrc/pytorch/points_in_polygons.cpp
  function points_in_polygons_forward_impl (line 4) | void points_in_polygons_forward_impl(const Tensor points, const Tensor p...
  function points_in_polygons_forward (line 11) | void points_in_polygons_forward(Tensor points, Tensor polygons, Tensor o...

FILE: mmcv/ops/csrc/pytorch/prroi_pool.cpp
  function prroi_pool_forward_impl (line 5) | void prroi_pool_forward_impl(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward_impl (line 12) | void prroi_pool_backward_impl(Tensor grad_output, Tensor rois,
  function prroi_pool_coor_backward_impl (line 19) | void prroi_pool_coor_backward_impl(Tensor output, Tensor grad_output,
  function prroi_pool_forward (line 28) | void prroi_pool_forward(Tensor input, Tensor rois, Tensor output,
  function prroi_pool_backward (line 35) | void prroi_pool_backward(Tensor grad_output, Tensor rois, Tensor grad_in...
  function prroi_pool_coor_backward (line 42) | void prroi_pool_coor_backward(Tensor output, Tensor grad_output, Tensor ...

FILE: mmcv/ops/csrc/pytorch/psamask.cpp
  function psamask_forward_impl (line 7) | void psamask_forward_impl(const int psa_type, const Tensor input, Tensor...
  function psamask_backward_impl (line 17) | void psamask_backward_impl(const int psa_type, const Tensor grad_output,
  function psamask_forward (line 27) | void psamask_forward(const Tensor input, Tensor output, const int psa_type,
  function psamask_backward (line 35) | void psamask_backward(Tensor grad_output, const Tensor grad_input,

FILE: mmcv/ops/csrc/pytorch/pybind.cpp
  function PYBIND11_MODULE (line 473) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: mmcv/ops/csrc/pytorch/riroi_align_rotated.cpp
  function riroi_align_rotated_forward_impl (line 5) | void riroi_align_rotated_forward_impl(Tensor features, Tensor rois,
  function riroi_align_rotated_backward_impl (line 15) | void riroi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
  function riroi_align_rotated_forward (line 25) | void riroi_align_rotated_forward(Tensor features, Tensor rois, Tensor ou...
  function riroi_align_rotated_backward (line 34) | void riroi_align_rotated_backward(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/roi_align.cpp
  function roi_align_forward_impl (line 20) | void roi_align_forward_impl(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_impl (line 30) | void roi_align_backward_impl(Tensor grad_output, Tensor rois, Tensor arg...
  function roi_align_forward_diopi (line 41) | void roi_align_forward_diopi(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward_diopi (line 88) | void roi_align_backward_diopi(Tensor grad_output, Tensor rois, Tensor ar...
  function roi_align_forward (line 139) | void roi_align_forward(Tensor input, Tensor rois, Tensor output,
  function roi_align_backward (line 154) | void roi_align_backward(Tensor grad_output, Tensor rois, Tensor argmax_y,

FILE: mmcv/ops/csrc/pytorch/roi_align_rotated.cpp
  function roi_align_rotated_forward_impl (line 5) | void roi_align_rotated_forward_impl(Tensor input, Tensor rois, Tensor ou...
  function roi_align_rotated_backward_impl (line 14) | void roi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
  function roi_align_rotated_forward (line 24) | void roi_align_rotated_forward(Tensor input, Tensor rois, Tensor output,
  function roi_align_rotated_backward (line 33) | void roi_align_rotated_backward(Tensor top_grad, Tensor rois,

FILE: mmcv/ops/csrc/pytorch/roi_pool.cpp
  function roi_pool_forward_impl (line 5) | void roi_pool_forward_impl(Tensor input, Tensor rois, Tensor output,
  function roi_pool_backward_impl (line 12) | void roi_pool_backward_impl(Tensor grad_output, Tensor rois, Tensor argmax,
  function roi_pool_forward (line 19) | void roi_pool_forward(Tensor input, Tensor rois, Tensor output, Tensor a...
  function roi_pool_backward (line 26) | void roi_pool_backward(Tensor grad_output, Tensor rois, Tensor argmax,

FILE: mmcv/ops/csrc/pytorch/roiaware_pool3d.cpp
  function roiaware_pool3d_forward_impl (line 4) | void roiaware_pool3d_forward_impl(int boxes_num, int pts_num, int channels,
  function roiaware_pool3d_backward_impl (line 16) | void roiaware_pool3d_backward_impl(int boxes_num, int out_x, int out_y,
  function roiaware_pool3d_forward (line 27) | void roiaware_pool3d_forward(Tensor rois, Tensor pts, Tensor pts_feature,
  function roiaware_pool3d_backward (line 54) | void roiaware_pool3d_backward(Tensor pts_idx_of_voxels, Tensor argmax,

FILE: mmcv/ops/csrc/pytorch/roipoint_pool3d.cpp
  function roipoint_pool3d_forward_impl (line 12) | void roipoint_pool3d_forward_impl(int batch_size, int pts_num, int boxes...
  function roipoint_pool3d_forward (line 23) | void roipoint_pool3d_forward(Tensor xyz, Tensor boxes3d, Tensor pts_feat...

FILE: mmcv/ops/csrc/pytorch/rotated_feature_align.cpp
  function rotated_feature_align_forward_impl (line 8) | void rotated_feature_align_forward_impl(const Tensor features,
  function rotated_feature_align_backward_impl (line 16) | void rotated_feature_align_backward_impl(const Tensor top_grad,
  function rotated_feature_align_forward (line 24) | void rotated_feature_align_forward(const Tensor features,
  function rotated_feature_align_backward (line 32) | void rotated_feature_align_backward(const Tensor top_grad,

FILE: mmcv/ops/csrc/pytorch/scatter_points.cpp
  function dynamic_point_to_voxel_forward_impl (line 7) | std::vector<torch::Tensor> dynamic_point_to_voxel_forward_impl(
  function dynamic_point_to_voxel_backward_impl (line 14) | void dynamic_point_to_voxel_backward_impl(
  function reduce_t (line 24) | inline reduce_t convert_reduce_type(const std::string &reduce_type) {
  function dynamic_point_to_voxel_forward (line 36) | std::vector<torch::Tensor> dynamic_point_to_voxel_forward(
  function dynamic_point_to_voxel_backward (line 43) | void dynamic_point_to_voxel_backward(torch::Tensor &grad_feats,

FILE: mmcv/ops/csrc/pytorch/sparse_pool_ops.cpp
  function indice_maxpool_forward_impl (line 18) | torch::Tensor indice_maxpool_forward_impl(torch::Tensor features,
  function indice_maxpool_forward (line 26) | torch::Tensor indice_maxpool_forward(torch::Tensor features,
  function indice_maxpool_backward_impl (line 32) | torch::Tensor indice_maxpool_backward_impl(torch::Tensor features,
  function indice_maxpool_backward (line 41) | torch::Tensor indice_maxpool_backward(torch::Tensor features,

FILE: mmcv/ops/csrc/pytorch/spconv_ops.cpp
  function get_indice_pairs_forward_cuda (line 27) | std::vector<torch::Tensor> get_indice_pairs_forward_cuda(
  function get_indice_pairs_forward_mlu (line 47) | std::vector<torch::Tensor> get_indice_pairs_forward_mlu(
  function get_indice_pairs_forward_musa (line 67) | std::vector<torch::Tensor> get_indice_pairs_forward_musa(
  function get_indice_pairs_backward_cuda (line 87) | std::vector<torch::Tensor> get_indice_pairs_backward_cuda(
  function get_indice_pairs_backward_musa (line 108) | std::vector<torch::Tensor> get_indice_pairs_backward_musa(
  function get_indice_pairs_forward (line 121) | std::vector<torch::Tensor> get_indice_pairs_forward(
  function get_indice_pairs_backward (line 155) | std::vector<torch::Tensor> get_indice_pairs_backward(
  function indice_conv_forward_impl (line 186) | torch::Tensor indice_conv_forward_impl(torch::Tensor features,
  function indice_conv_forward (line 197) | torch::Tensor indice_conv_forward(torch::Tensor features, torch::Tensor ...
  function indice_conv_backward_impl (line 205) | std::vector<torch::Tensor> indice_conv_backward_impl(
  function indice_conv_backward (line 213) | std::vector<torch::Tensor> indice_conv_backward(

FILE: mmcv/ops/csrc/pytorch/spconv_utils.h
  function namespace (line 28) | namespace tv {

FILE: mmcv/ops/csrc/pytorch/sync_bn.cpp
  function sync_bn_forward_mean_impl (line 5) | void sync_bn_forward_mean_impl(const Tensor input, Tensor mean) {
  function sync_bn_forward_var_impl (line 9) | void sync_bn_forward_var_impl(const Tensor input, const Tensor mean,
  function sync_bn_forward_output_impl (line 14) | void sync_bn_forward_output_impl(const Tensor input, const Tensor mean,
  function sync_bn_backward_param_impl (line 25) | void sync_bn_backward_param_impl(const Tensor grad_output, const Tensor ...
  function sync_bn_backward_data_impl (line 31) | void sync_bn_backward_data_impl(const Tensor grad_output, const Tensor w...
  function sync_bn_forward_mean (line 39) | void sync_bn_forward_mean(const Tensor input, Tensor mean) {
  function sync_bn_forward_var (line 43) | void sync_bn_forward_var(const Tensor input, const Tensor mean, Tensor v...
  function sync_bn_forward_output (line 47) | void sync_bn_forward_output(const Tensor input, const Tensor mean,
  function sync_bn_backward_param (line 58) | void sync_bn_backward_param(const Tensor grad_output, const Tensor norm,
  function sync_bn_backward_data (line 63) | void sync_bn_backward_data(const Tensor grad_output, const Tensor weight,

FILE: mmcv/ops/csrc/pytorch/three_interpolate.cpp
  function three_interpolate_forward_impl (line 7) | void three_interpolate_forward_impl(int b, int c, int m, int n,
  function three_interpolate_backward_impl (line 14) | void three_interpolate_backward_impl(int b, int c, int n, int m,
  function three_interpolate_forward (line 21) | void three_interpolate_forward(Tensor points_tensor, Tensor idx_tensor,
  function three_interpolate_backward (line 28) | void three_interpolate_backward(Tensor grad_out_tensor, Tensor idx_tensor,

FILE: mmcv/ops/csrc/pytorch/three_nn.cpp
  function three_nn_forward_impl (line 7) | void three_nn_forward_impl(int b, int n, int m, const Tensor unknown,
  function three_nn_forward (line 13) | void three_nn_forward(Tensor unknown_tensor, Tensor known_tensor,

FILE: mmcv/ops/csrc/pytorch/tin_shift.cpp
  function tin_shift_forward_impl (line 5) | void tin_shift_forward_impl(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward_impl (line 9) | void tin_shift_backward_impl(Tensor grad_output, Tensor shift,
  function tin_shift_forward (line 14) | void tin_shift_forward(Tensor input, Tensor shift, Tensor output) {
  function tin_shift_backward (line 18) | void tin_shift_backward(Tensor grad_output, Tensor shift, Tensor grad_in...

FILE: mmcv/ops/csrc/pytorch/upfirdn2d.cpp
  function upfirdn2d_op_impl (line 105) | torch::Tensor upfirdn2d_op_impl(torch::Tensor input, torch::Tensor filter,
  function upfirdn2d (line 113) | torch::Tensor upfirdn2d(torch::Tensor input, torch::Tensor filter, int upx,

FILE: mmcv/ops/csrc/pytorch/voxelization.cpp
  function hard_voxelize_forward_impl (line 19) | int hard_voxelize_forward_impl(const at::Tensor &points, at::Tensor &vox...
  function nondeterministic_hard_voxelize_forward_impl (line 31) | int nondeterministic_hard_voxelize_forward_impl(
  function dynamic_voxelize_forward_impl (line 42) | void dynamic_voxelize_forward_impl(const at::Tensor &points, at::Tensor ...
  function hard_voxelize_forward_diopi (line 51) | void hard_voxelize_forward_diopi(const at::Tensor &points,
  function dynamic_voxelize_forward_diopi (line 138) | void dynamic_voxelize_forward_diopi(const at::Tensor &points,
  function hard_voxelize_forward (line 192) | void hard_voxelize_forward(const at::Tensor &points,
  function dynamic_voxelize_forward (line 224) | void dynamic_voxelize_forward(const at::Tensor &points,

FILE: mmcv/ops/deform_conv.py
  class DeformConv2dFunction (line 25) | class DeformConv2dFunction(Function):
    method symbolic (line 28) | def symbolic(g,
    method _npu_backward (line 53) | def _npu_backward(ctx, grad_output):
    method forward (line 72) | def forward(ctx,
    method backward (line 146) | def backward(
    method _output_size (line 210) | def _output_size(ctx, input, weight):
  class DeformConv2d (line 229) | class DeformConv2d(nn.Module):
    method __init__ (line 265) | def __init__(self,
    method reset_parameters (line 306) | def reset_parameters(self):
    method forward (line 313) | def forward(self, x: Tensor, offset: Tensor) -> Tensor:
    method __repr__ (line 352) | def __repr__(self):
  class DeformConv2dPack (line 368) | class DeformConv2dPack(DeformConv2d):
    method __init__ (line 395) | def __init__(self, *args, **kwargs):
    method init_offset (line 407) | def init_offset(self):
    method forward (line 411) | def forward(self, x: Tensor) -> Tensor:  # type: ignore
    method _load_from_state_dict (line 417) | def _load_from_state_dict(self, state_dict, prefix, local_metadata, st...
  class DeformConv2dPack_MLU (line 451) | class DeformConv2dPack_MLU(DeformConv2d):
    method __init__ (line 476) | def __init__(self, *args, **kwargs):
    method init_offset (line 492) | def init_offset(self):
    method forward (line 496) | def forward(self, x: Tensor) -> Tensor:  # type: ignore

FILE: mmcv/ops/deform_roi_pool.py
  class DeformRoIPoolFunction (line 15) | class DeformRoIPoolFunction(Function):
    method symbolic (line 18) | def symbolic(g, input, rois, offset, output_size, spatial_scale,
    method forward (line 34) | def forward(ctx,
    method backward (line 71) | def backward(
  class DeformRoIPool (line 98) | class DeformRoIPool(nn.Module):
    method __init__ (line 100) | def __init__(self,
    method forward (line 111) | def forward(self,
  class DeformRoIPoolPack (line 120) | class DeformRoIPoolPack(DeformRoIPool):
    method __init__ (line 122) | def __init__(self,
    method forward (line 146) | def forward(self, input: Tensor, rois: Tensor) -> Tensor:  # type: ignore
  class ModulatedDeformRoIPoolPack (line 160) | class ModulatedDeformRoIPoolPack(DeformRoIPool):
    method __init__ (line 162) | def __init__(self,
    method forward (line 197) | def forward(self, input: Tensor, rois: Tensor) -> Tensor:  # type: ignore

FILE: mmcv/ops/deprecated_wrappers.py
  class Conv2d_deprecated (line 9) | class Conv2d_deprecated(Conv2d):
    method __init__ (line 11) | def __init__(self, *args, **kwargs):
  class ConvTranspose2d_deprecated (line 19) | class ConvTranspose2d_deprecated(ConvTranspose2d):
    method __init__ (line 21) | def __init__(self, *args, **kwargs):
  class MaxPool2d_deprecated (line 29) | class MaxPool2d_deprecated(MaxPool2d):
    method __init__ (line 31) | def __init__(self, *args, **kwargs):
  class Linear_deprecated (line 39) | class Linear_deprecated(Linear):
    method __init__ (line 41) | def __init__(self, *args, **kwargs):

FILE: mmcv/ops/diff_iou_rotated.py
  class SortVertices (line 17) | class SortVertices(Function):
    method forward (line 20) | def forward(ctx, vertices, mask, num_valid):
    method backward (line 28) | def backward(ctx, gradout):
  function box_intersection (line 32) | def box_intersection(corners1: Tensor,
  function box1_in_box2 (line 75) | def box1_in_box2(corners1: Tensor, corners2: Tensor) -> Tensor:
  function box_in_box (line 108) | def box_in_box(corners1: Tensor, corners2: Tensor) -> Tuple[Tensor, Tens...
  function build_vertices (line 125) | def build_vertices(corners1: Tensor, corners2: Tensor, c1_in_2: Tensor,
  function sort_indices (line 157) | def sort_indices(vertices: Tensor, mask: Tensor) -> Tensor:
  function calculate_area (line 184) | def calculate_area(idx_sorted: Tensor,
  function oriented_box_intersection_2d (line 206) | def oriented_box_intersection_2d(corners1: Tensor,
  function box2corners (line 227) | def box2corners(box: Tensor) -> Tensor:
  function diff_iou_rotated_2d (line 255) | def diff_iou_rotated_2d(box1: Tensor, box2: Tensor) -> Tensor:
  function diff_iou_rotated_3d (line 276) | def diff_iou_rotated_3d(box3d1: Tensor, box3d2: Tensor) -> Tensor:

FILE: mmcv/ops/filtered_lrelu.py
  function filtered_lrelu (line 26) | def filtered_lrelu(input: torch.Tensor,
  function _filtered_lrelu_ref (line 149) | def _filtered_lrelu_ref(input: torch.Tensor,
  function _filtered_lrelu_cuda (line 238) | def _filtered_lrelu_cuda(up: int = 1,
  function _filtered_lrelu_musa (line 441) | def _filtered_lrelu_musa(up: int = 1,

FILE: mmcv/ops/focal_loss.py
  class SigmoidFocalLossFunction (line 17) | class SigmoidFocalLossFunction(Function):
    method forward (line 20) | def forward(ctx,
    method backward (line 57) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
  class SigmoidFocalLoss (line 79) | class SigmoidFocalLoss(nn.Module):
    method __init__ (line 81) | def __init__(self,
    method forward (line 92) | def forward(
    method __repr__ (line 100) | def __repr__(self):
  class SoftmaxFocalLossFunction (line 108) | class SoftmaxFocalLossFunction(Function):
    method forward (line 111) | def forward(ctx,
    method backward (line 159) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
  class SoftmaxFocalLoss (line 182) | class SoftmaxFocalLoss(nn.Module):
    method __init__ (line 184) | def __init__(self,
    method forward (line 195) | def forward(
    method __repr__ (line 203) | def __repr__(self):

FILE: mmcv/ops/furthest_point_sample.py
  class FurthestPointSampling (line 13) | class FurthestPointSampling(Function):
    method forward (line 18) | def forward(ctx, points_xyz: torch.Tensor,
    method backward (line 54) | def backward(xyz, a=None):
  class FurthestPointSamplingWithDist (line 58) | class FurthestPointSamplingWithDist(Function):
    method forward (line 63) | def forward(ctx, points_dist: torch.Tensor,
    method backward (line 87) | def backward(xyz, a=None):

FILE: mmcv/ops/fused_bias_leakyrelu.py
  class FusedBiasLeakyReLUFunctionBackward (line 108) | class FusedBiasLeakyReLUFunctionBackward(Function):
    method forward (line 116) | def forward(ctx, grad_output: torch.Tensor, out: torch.Tensor,
    method backward (line 143) | def backward(ctx, gradgrad_input: torch.Tensor,
  class FusedBiasLeakyReLUFunction (line 162) | class FusedBiasLeakyReLUFunction(Function):
    method forward (line 165) | def forward(ctx, input: torch.Tensor, bias: nn.Parameter,
    method backward (line 184) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
  class FusedBiasLeakyReLU (line 193) | class FusedBiasLeakyReLU(nn.Module):
    method __init__ (line 217) | def __init__(self,
    method forward (line 227) | def forward(self, input: torch.Tensor) -> torch.Tensor:
  function fused_bias_leakyrelu (line 232) | def fused_bias_leakyrelu(input: torch.Tensor,
  function bias_leakyrelu_ref (line 269) | def bias_leakyrelu_ref(x: torch.Tensor,

FILE: mmcv/ops/gather_points.py
  class GatherPoints (line 12) | class GatherPoints(Function):
    method forward (line 16) | def forward(ctx, features: torch.Tensor,
    method backward (line 42) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, None]:

FILE: mmcv/ops/group_points.py
  class QueryAndGroup (line 18) | class QueryAndGroup(nn.Module):
    method __init__ (line 41) | def __init__(self,
    method forward (line 69) | def forward(
  class GroupAll (line 145) | class GroupAll(nn.Module):
    method __init__ (line 152) | def __init__(self, use_xyz: bool = True):
    method forward (line 156) | def forward(self,
  class GroupingOperation (line 184) | class GroupingOperation(Function):
    method forward (line 188) | def forward(
    method backward (line 252) | def backward(ctx, grad_out: torch.Tensor) -> Tuple:

FILE: mmcv/ops/info.py
  function get_compiler_version (line 7) | def get_compiler_version():
  function get_compiling_cuda_version (line 10) | def get_compiling_cuda_version():
  function get_compiler_version (line 17) | def get_compiler_version():
  function get_compiling_cuda_version (line 20) | def get_compiling_cuda_version():

FILE: mmcv/ops/iou3d.py
  function boxes_overlap_bev (line 16) | def boxes_overlap_bev(boxes_a: Tensor, boxes_b: Tensor) -> Tensor:
  function boxes_iou3d (line 35) | def boxes_iou3d(boxes_a: Tensor, boxes_b: Tensor) -> Tensor:
  function nms3d (line 69) | def nms3d(boxes: Tensor, scores: Tensor, iou_threshold: float) -> Tensor:
  function nms3d_normal (line 93) | def nms3d_normal(boxes: Tensor, scores: Tensor,
  function _xyxyr2xywhr (line 119) | def _xyxyr2xywhr(boxes: Tensor) -> Tensor:
  function boxes_iou_bev (line 137) | def boxes_iou_bev(boxes_a: Tensor, boxes_b: Tensor) -> Tensor:
  function nms_bev (line 159) | def nms_bev(boxes: Tensor,
  function nms_normal_bev (line 204) | def nms_normal_bev(boxes: Tensor, scores: Tensor, thresh: float) -> Tensor:

FILE: mmcv/ops/knn.py
  class KNN (line 12) | class KNN(Function):
    method forward (line 22) | def forward(ctx,
    method backward (line 92) | def backward(ctx, a=None):

FILE: mmcv/ops/masked_conv.py
  class MaskedConv2dFunction (line 17) | class MaskedConv2dFunction(Function):
    method symbolic (line 20) | def symbolic(g, features, mask, weight, bias, padding, stride=1):
    method forward (line 31) | def forward(ctx,
    method backward (line 105) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
  class MaskedConv2d (line 112) | class MaskedConv2d(nn.Conv2d):
    method __init__ (line 119) | def __init__(self,
    method forward (line 131) | def forward(self,

FILE: mmcv/ops/merge_cells.py
  class BaseMergeCell (line 13) | class BaseMergeCell(nn.Module):
    method __init__ (line 45) | def __init__(self,
    method _build_input_conv (line 80) | def _build_input_conv(self, channel, conv_cfg, norm_cfg):
    method _binary_op (line 91) | def _binary_op(self, x1, x2):
    method _resize (line 94) | def _resize(self, x, size):
    method forward (line 115) | def forward(self,
  class SumCell (line 136) | class SumCell(BaseMergeCell):
    method __init__ (line 138) | def __init__(self, in_channels: int, out_channels: int, **kwargs):
    method _binary_op (line 141) | def _binary_op(self, x1, x2):
  class ConcatCell (line 145) | class ConcatCell(BaseMergeCell):
    method __init__ (line 147) | def __init__(self, in_channels: int, out_channels: int, **kwargs):
    method _binary_op (line 150) | def _binary_op(self, x1, x2):
  class GlobalPoolingCell (line 155) | class GlobalPoolingCell(BaseMergeCell):
    method __init__ (line 157) | def __init__(self,
    method _binary_op (line 164) | def _binary_op(self, x1, x2):

FILE: mmcv/ops/min_area_polygons.py
  function min_area_polygons (line 9) | def min_area_polygons(pointsets: torch.Tensor) -> torch.Tensor:

FILE: mmcv/ops/modulated_deform_conv.py
  class ModulatedDeformConv2dFunction (line 22) | class ModulatedDeformConv2dFunction(Function):
    method symbolic (line 25) | def symbolic(g, input, offset, mask, weight, bias, stride, padding,
    method _calculate_sort_index (line 40) | def _calculate_sort_index(kernel_h, kernel_w, deformable_group):
    method _npu_forward (line 53) | def _npu_forward(ctx, input_tensor, offset, mask, weight, bias):
    method _npu_backward (line 83) | def _npu_backward(ctx, grad_output):
    method forward (line 105) | def forward(ctx,
    method backward (line 174) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
    method _output_size (line 217) | def _output_size(ctx, input, weight):
  class ModulatedDeformConv2d (line 236) | class ModulatedDeformConv2d(nn.Module):
    method __init__ (line 240) | def __init__(self,
    method init_weights (line 272) | def init_weights(self):
    method forward (line 281) | def forward(self, x: torch.Tensor, offset: torch.Tensor,
  class ModulatedDeformConv2dPack (line 290) | class ModulatedDeformConv2dPack(ModulatedDeformConv2d):
    method __init__ (line 309) | def __init__(self, *args, **kwargs):
    method init_weights (line 321) | def init_weights(self) -> None:
    method forward (line 327) | def forward(self, x: torch.Tensor) -> torch.Tensor:  # type: ignore
    method _load_from_state_dict (line 337) | def _load_from_state_dict(self, state_dict, prefix, local_metadata, st...
  class ModulatedDeformConv2dPack_MLU (line 372) | class ModulatedDeformConv2dPack_MLU(ModulatedDeformConv2d):
    method __init__ (line 391) | def __init__(self, *args, **kwargs):
    method init_weights (line 406) | def init_weights(self):
    method forward (line 412) | def forward(self, x):

FILE: mmcv/ops/multi_scale_deform_attn.py
  class MultiScaleDeformableAttnFunction (line 23) | class MultiScaleDeformableAttnFunction(Function):
    method forward (line 26) | def forward(ctx, value: torch.Tensor, value_spatial_shapes: torch.Tensor,
    method backward (line 79) | def backward(ctx, grad_output: torch.Tensor) -> tuple:
  function multi_scale_deformable_attn_pytorch (line 110) | def multi_scale_deformable_attn_pytorch(
  class MultiScaleDeformableAttention (line 173) | class MultiScaleDeformableAttention(BaseModule):
    method __init__ (line 202) | def __init__(self,
    method init_weights (line 252) | def init_weights(self) -> None:
    method forward (line 276) | def forward(self,

FILE: mmcv/ops/nms.py
  class NMSop (line 15) | class NMSop(torch.autograd.Function):
    method forward (line 18) | def forward(ctx: Any, bboxes: Tensor, scores: Tensor, iou_threshold: f...
  class SoftNMSop (line 37) | class SoftNMSop(torch.autograd.Function):
    method forward (line 40) | def forward(ctx: Any, boxes: Tensor, scores: Tensor, iou_threshold: fl...
    method symbolic (line 56) | def symbolic(g, boxes, scores, iou_threshold, sigma, min_score, method,
  function nms (line 77) | def nms(boxes: array_like_type,
  function soft_nms (line 137) | def soft_nms(boxes: array_like_type,
  function batched_nms (line 216) | def batched_nms(boxes: Tensor,
  function nms_match (line 337) | def nms_match(dets: array_like_type,
  function nms_rotated (line 375) | def nms_rotated(dets: Tensor,
  function nms_quadri (line 448) | def nms_quadri(dets: Tensor,

FILE: mmcv/ops/pixel_group.py
  function pixel_group (line 13) | def pixel_group(

FILE: mmcv/ops/point_sample.py
  function bilinear_grid_sample (line 12) | def bilinear_grid_sample(im: Tensor,
  function normalize (line 90) | def normalize(grid: Tensor) -> Tensor:
  function denormalize (line 103) | def denormalize(grid: Tensor) -> Tensor:
  function generate_grid (line 116) | def generate_grid(num_grid: int, size: Tuple[int, int],
  function rel_roi_point_to_abs_img_point (line 138) | def rel_roi_point_to_abs_img_point(rois: Tensor,
  function get_shape_from_feature_map (line 170) | def get_shape_from_feature_map(x: Tensor) -> Tensor:
  function abs_img_point_to_rel_img_point (line 185) | def abs_img_point_to_rel_img_point(abs_img_points: Tensor,
  function rel_roi_point_to_rel_img_point (line 218) | def rel_roi_point_to_rel_img_point(rois: Tensor,
  function point_sample (line 245) | def point_sample(input: Tensor,
  class SimpleRoIAlign (line 277) | class SimpleRoIAlign(nn.Module):
    method __init__ (line 279) | def __init__(self,
    method forward (line 300) | def forward(self, features: Tensor, rois: Tensor) -> Tensor:
    method __repr__ (line 328) | def __repr__(self) -> str:

FILE: mmcv/ops/points_in_boxes.py
  function points_in_boxes_part (line 13) | def points_in_boxes_part(points: Tensor, boxes: Tensor) -> Tensor:
  function points_in_boxes_cpu (line 68) | def points_in_boxes_cpu(points: Tensor, boxes: Tensor) -> Tensor:
  function points_in_boxes_all (line 106) | def points_in_boxes_all(points: Tensor, boxes: Tensor) -> Tensor:

FILE: mmcv/ops/points_in_polygons.py
  function points_in_polygons (line 9) | def points_in_polygons(points: Tensor, polygons: Tensor) -> Tensor:

FILE: mmcv/ops/points_sampler.py
  function calc_square_dist (line 11) | def calc_square_dist(point_feat_a: Tensor,
  function get_sampler_cls (line 34) | def get_sampler_cls(sampler_type: str) -> nn.Module:
  class PointsSampler (line 57) | class PointsSampler(nn.Module):
    method __init__ (line 71) | def __init__(self,
    method forward (line 88) | def forward(self, points_xyz: Tensor, features: Tensor) -> Tensor:
  class DFPSSampler (line 134) | class DFPSSampler(nn.Module):
    method __init__ (line 137) | def __init__(self) -> None:
    method forward (line 140) | def forward(self, points: Tensor, features: Tensor, npoint: int) -> Te...
  class FFPSSampler (line 146) | class FFPSSampler(nn.Module):
    method __init__ (line 149) | def __init__(self) -> None:
    method forward (line 152) | def forward(self, points: Tensor, features: Tensor, npoint: int) -> Te...
  class FSSampler (line 163) | class FSSampler(nn.Module):
    method __init__ (line 166) | def __init__(self) -> None:
    method forward (line 169) | def forward(self, points: Tensor, features: Tensor, npoint: int) -> Te...

FILE: mmcv/ops/prroi_pool.py
  class PrRoIPoolFunction (line 18) | class PrRoIPoolFunction(Function):
    method symbolic (line 21) | def symbolic(g, features, rois, output_size, spatial_scale):
    method forward (line 31) | def forward(ctx,
    method backward (line 67) | def backward(
  class PrRoIPool (line 101) | class PrRoIPool(nn.Module):
    method __init__ (line 126) | def __init__(self,
    method forward (line 134) | def forward(self, features: torch.Tensor,
    method __repr__ (line 148) | def __repr__(self):

FILE: mmcv/ops/psa_mask.py
  class PSAMaskFunction (line 15) | class PSAMaskFunction(Function):
    method symbolic (line 18) | def symbolic(g, input, psa_type, mask_size):
    method forward (line 26) | def forward(ctx, input: torch.Tensor, psa_type: str,
    method backward (line 52) | def backward(
  class PSAMask (line 78) | class PSAMask(nn.Module):
    method __init__ (line 80) | def __init__(self, psa_type: str, mask_size: Optional[tuple] = None):
    method forward (line 91) | def forward(self, input: torch.Tensor) -> torch.Tensor:
    method __repr__ (line 94) | def __repr__(self):

FILE: mmcv/ops/riroi_align_rotated.py
  class RiRoIAlignRotatedFunction (line 15) | class RiRoIAlignRotatedFunction(Function):
    method forward (line 18) | def forward(ctx: Any,
    method backward (line 61) | def backward(
  class RiRoIAlignRotated (line 99) | class RiRoIAlignRotated(nn.Module):
    method __init__ (line 120) | def __init__(self,
    method forward (line 134) | def forward(self, features: torch.Tensor,

FILE: mmcv/ops/roi_align.py
  class RoIAlignFunction (line 17) | class RoIAlignFunction(Function):
    method symbolic (line 20) | def symbolic(g, input, rois, output_size, spatial_scale, sampling_ratio,
    method forward (line 62) | def forward(ctx: Any,
    method backward (line 108) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:
  class RoIAlign (line 131) | class RoIAlign(nn.Module):
    method __init__ (line 174) | def __init__(self,
    method forward (line 190) | def forward(self, input: torch.Tensor, rois: torch.Tensor) -> torch.Te...
    method __repr__ (line 213) | def __repr__(self):

FILE: mmcv/ops/roi_align_rotated.py
  class RoIAlignRotatedFunction (line 16) | class RoIAlignRotatedFunction(Function):
    method symbolic (line 19) | def symbolic(g, input, rois, output_size, spatial_scale, sampling_ratio,
    method forward (line 44) | def forward(ctx: Any,
    method backward (line 78) | def backward(
  class RoIAlignRotated (line 111) | class RoIAlignRotated(nn.Module):
    method __init__ (line 160) | def __init__(self,
    method forward (line 174) | def forward(self, input: torch.Tensor, rois: torch.Tensor) -> torch.Te...
    method __repr__ (line 180) | def __repr__(self):

FILE: mmcv/ops/roi_pool.py
  class RoIPoolFunction (line 16) | class RoIPoolFunction(Function):
    method symbolic (line 19) | def symbolic(g, input, rois, output_size, spatial_scale):
    method forward (line 28) | def forward(ctx: Any,
    method backward (line 58) | def backward(
  class RoIPool (line 79) | class RoIPool(nn.Module):
    method __init__ (line 81) | def __init__(self,
    method forward (line 89) | def forward(self, input: torch.Tensor, rois: torch.Tensor) -> torch.Te...
    method __repr__ (line 92) | def __repr__(self):

FILE: mmcv/ops/roiaware_pool3d.py
  class RoIAwarePool3d (line 15) | class RoIAwarePool3d(nn.Module):
    method __init__ (line 30) | def __init__(self,
    method forward (line 42) | def forward(self, rois: torch.Tensor, pts: torch.Tensor,
  class RoIAwarePool3dFunction (line 61) | class RoIAwarePool3dFunction(Function):
    method forward (line 64) | def forward(ctx: Any, rois: torch.Tensor, pts: torch.Tensor,
    method backward (line 118) | def backward(

FILE: mmcv/ops/roipoint_pool3d.py
  class RoIPointPool3d (line 12) | class RoIPointPool3d(nn.Module):
    method __init__ (line 23) | def __init__(self, num_sampled_points: int = 512):
    method forward (line 27) | def forward(self, points: torch.Tensor, point_features: torch.Tensor,
  class RoIPointPool3dFunction (line 45) | class RoIPointPool3dFunction(Function):
    method forward (line 48) | def forward(
    method backward (line 86) | def backward(ctx: Any, grad_out: torch.Tensor) -> torch.Tensor:

FILE: mmcv/ops/rotated_feature_align.py
  class RotatedFeatureAlignFunction (line 15) | class RotatedFeatureAlignFunction(Function):
    method symbolic (line 26) | def symbolic(g, features, best_rbboxes, spatial_scale, points):
    method forward (line 36) | def forward(ctx: Any, features: torch.Tensor, best_rbboxes: torch.Tensor,
    method backward (line 66) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:
  function rotated_feature_align (line 90) | def rotated_feature_align(features: torch.Tensor,

FILE: mmcv/ops/saconv.py
  class SAConv2d (line 15) | class SAConv2d(ConvAWS2d):
    method __init__ (line 41) | def __init__(self,
    method init_weights (line 85) | def init_weights(self):
    method forward (line 94) | def forward(self, x):

FILE: mmcv/ops/scatter_points.py
  class _DynamicScatter (line 16) | class _DynamicScatter(Function):
    method forward (line 19) | def forward(ctx: Any,
    method backward (line 50) | def backward(ctx: Any,
  class DynamicScatter (line 67) | class DynamicScatter(nn.Module):
    method __init__ (line 83) | def __init__(self, voxel_size: List, point_cloud_range: List,
    method forward_single (line 91) | def forward_single(
    method forward (line 110) | def forward(self, points: torch.Tensor,
    method __repr__ (line 142) | def __repr__(self):

FILE: mmcv/ops/sparse_conv.py
  function _calculate_fan_in_and_fan_out_hwio (line 28) | def _calculate_fan_in_and_fan_out_hwio(tensor):
  class SparseConvolution (line 49) | class SparseConvolution(SparseModule):
    method __init__ (line 51) | def __init__(self,
    method reset_parameters (line 107) | def reset_parameters(self):
    method forward (line 114) | def forward(self, input):
  class SparseConv2d (line 208) | class SparseConv2d(SparseConvolution):
    method __init__ (line 210) | def __init__(self,
  class SparseConv3d (line 234) | class SparseConv3d(SparseConvolution):
    method __init__ (line 236) | def __init__(self,
  class SparseConv4d (line 260) | class SparseConv4d(SparseConvolution):
    method __init__ (line 262) | def __init__(self,
  class SparseConvTranspose2d (line 286) | class SparseConvTranspose2d(SparseConvolution):
    method __init__ (line 288) | def __init__(self,
  class SparseConvTranspose3d (line 313) | class SparseConvTranspose3d(SparseConvolution):
    method __init__ (line 315) | def __init__(self,
  class SparseInverseConv2d (line 340) | class SparseInverseConv2d(SparseConvolution):
    method __init__ (line 342) | def __init__(self,
  class SparseInverseConv3d (line 359) | class SparseInverseConv3d(SparseConvolution):
    method __init__ (line 361) | def __init__(self,
  class SubMConv2d (line 378) | class SubMConv2d(SparseConvolution):
    method __init__ (line 380) | def __init__(self,
  class SubMConv3d (line 405) | class SubMConv3d(SparseConvolution):
    method __init__ (line 407) | def __init__(self,
  class SubMConv4d (line 432) | class SubMConv4d(SparseConvolution):
    method __init__ (line 434) | def __init__(self,

FILE: mmcv/ops/sparse_functional.py
  class SparseConvFunction (line 22) | class SparseConvFunction(Function):
    method forward (line 30) | def forward(ctx: Any, features: torch.Tensor, filters: torch.nn.Parame...
    method backward (line 50) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:
  class SparseInverseConvFunction (line 59) | class SparseInverseConvFunction(Function):
    method forward (line 62) | def forward(ctx: Any, features: torch.Tensor, filters: torch.nn.Parame...
    method backward (line 82) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:
  class SubMConvFunction (line 91) | class SubMConvFunction(Function):
    method forward (line 94) | def forward(ctx: Any, features: torch.Tensor, filters: torch.nn.Parame...
    method backward (line 114) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:
  class SparseMaxPoolFunction (line 123) | class SparseMaxPoolFunction(Function):
    method forward (line 126) | def forward(ctx, features: torch.Tensor, indice_pairs: torch.Tensor,
    method backward (line 146) | def backward(ctx: Any, grad_output: torch.Tensor) -> tuple:

FILE: mmcv/ops/sparse_modules.py
  function is_spconv_module (line 24) | def is_spconv_module(module: nn.Module) -> bool:
  function is_sparse_conv (line 29) | def is_sparse_conv(module: nn.Module) -> bool:
  function _mean_update (line 34) | def _mean_update(vals: Union[int, List], m_vals: Union[int, List],
  class SparseModule (line 49) | class SparseModule(nn.Module):
  class SparseSequential (line 55) | class SparseSequential(SparseModule):
    method __init__ (line 89) | def __init__(self, *args, **kwargs):
    method __getitem__ (line 105) | def __getitem__(self, idx: int) -> torch.Tensor:
    method __len__ (line 115) | def __len__(self):
    method sparity_dict (line 119) | def sparity_dict(self):
    method add (line 122) | def add(self, module: Any, name: Optional[str] = None) -> None:
    method forward (line 129) | def forward(self, input: torch.Tensor) -> torch.Tensor:
    method fused (line 143) | def fused(self):
  class ToDense (line 190) | class ToDense(SparseModule):
    method forward (line 193) | def forward(self, x: SparseConvTensor):
  class RemoveGrid (line 197) | class RemoveGrid(SparseModule):
    method forward (line 200) | def forward(self, x: SparseConvTensor):

FILE: mmcv/ops/sparse_ops.py
  function get_conv_output_size (line 28) | def get_conv_output_size(input_size, kernel_size, stride, padding, dilat...
  function get_deconv_output_size (line 41) | def get_deconv_output_size(input_size, kernel_size, stride, padding, dil...
  function get_indice_pairs (line 54) | def get_indice_pairs(indices,
  function indice_conv (line 116) | def indice_conv(features,
  function fused_indice_conv (line 132) | def fused_indice_conv(features, filters, bias, indice_pairs, indice_pair...
  function indice_conv_backward (line 143) | def indice_conv_backward(features,
  function indice_maxpool (line 158) | def indice_maxpool(features, indice_pairs, indice_pair_num, num_activate...
  function indice_maxpool_backward (line 167) | def indice_maxpool_backward(features, out_features, out_bp, indice_pairs,

FILE: mmcv/ops/sparse_pool.py
  class SparseMaxPool (line 23) | class SparseMaxPool(SparseModule):
    method __init__ (line 25) | def __init__(self,
    method forward (line 49) | def forward(self, input):
  class SparseMaxPool2d (line 77) | class SparseMaxPool2d(SparseMaxPool):
    method __init__ (line 79) | def __init__(self, kernel_size, stride=1, padding=0, dilation=1):
  class SparseMaxPool3d (line 83) | class SparseMaxPool3d(SparseMaxPool):
    method __init__ (line 85) | def __init__(self, kernel_size, stride=1, padding=0, dilation=1):

FILE: mmcv/ops/sparse_structure.py
  function scatter_nd (line 7) | def scatter_nd(indices: torch.Tensor, updates: torch.Tensor,
  class SparseConvTensor (line 24) | class SparseConvTensor:
    method __init__ (line 26) | def __init__(self,
    method spatial_size (line 42) | def spatial_size(self):
    method find_indice_pair (line 45) | def find_indice_pair(self, key):
    method dense (line 52) | def dense(self, channels_first: bool = True) -> torch.Tensor:
    method sparity (line 64) | def sparity(self):

FILE: mmcv/ops/sync_bn.py
  class SyncBatchNormFunction (line 22) | class SyncBatchNormFunction(Function):
    method symbolic (line 25) | def symbolic(g, input, running_mean, running_var, weight, bias, momentum,
    method forward (line 41) | def forward(self, input: torch.Tensor, running_mean: torch.Tensor,
    method backward (line 144) | def backward(self, grad_output: torch.Tensor) -> tuple:
  class SyncBatchNorm (line 174) | class SyncBatchNorm(Module):
    method __init__ (line 208) | def __init__(self,
    method reset_running_stats (line 245) | def reset_running_stats(self):
    method reset_parameters (line 251) | def reset_parameters(self):
    method forward (line 257) | def forward(self, input: torch.Tensor) -> torch.Tensor:
    method __repr__ (line 285) | def __repr__(self):

FILE: mmcv/ops/three_interpolate.py
  class ThreeInterpolate (line 12) | class ThreeInterpolate(Function):
    method forward (line 20) | def forward(ctx: Any, features: torch.Tensor, indices: torch.Tensor,
    method backward (line 48) | def backward(

FILE: mmcv/ops/three_nn.py
  class ThreeNN (line 11) | class ThreeNN(Function):
    method forward (line 19) | def forward(ctx: Any, target: torch.Tensor,
    method backward (line 62) | def backward(ctx, a=None, b=None):

FILE: mmcv/ops/tin_shift.py
  class TINShiftFunction (line 17) | class TINShiftFunction(Function):
    method forward (line 20) | def forward(ctx, input, shift):
    method backward (line 39) | def backward(ctx, grad_output):
  class TINShift (line 52) | class TINShift(nn.Module):
    method forward (line 66) | def forward(self, input, shift):

FILE: mmcv/ops/upfirdn2d.py
  function _parse_scaling (line 21) | def _parse_scaling(scaling):
  function _parse_padding (line 32) | def _parse_padding(padding):
  function _get_filter_size (line 45) | def _get_filter_size(filter):
  function upfirdn2d (line 58) | def upfirdn2d(input: torch.Tensor,
  function _upfirdn2d_ref (line 136) | def _upfirdn2d_ref(input: torch.Tensor,
  function _upfirdn2d_cuda (line 221) | def _upfirdn2d_cuda(up: int = 1,
  function _upfirdn2d_musa (line 316) | def _upfirdn2d_musa(up: int = 1,
  function filter2d (line 408) | def filter2d(input: torch.Tensor,
  function upsample2d (line 457) | def upsample2d(input: torch.Tensor,
  function downsample2d (line 511) | def downsample2d(input: torch.Tensor,

FILE: mmcv/ops/voxelize.py
  class _Voxelization (line 15) | class _Voxelization(Function):
    method forward (line 18) | def forward(
  class Voxelization (line 98) | class Voxelization(nn.Module):
    method __init__ (line 116) | def __init__(self,
    method forward (line 165) | def forward(self, input: torch.Tensor) -> torch.Tensor:
    method __repr__ (line 175) | def __repr__(self):

FILE: mmcv/transforms/base.py
  class BaseTransform (line 6) | class BaseTransform(metaclass=ABCMeta):
    method __call__ (line 9) | def __call__(self,
    method transform (line 15) | def transform(self,

FILE: mmcv/transforms/formatting.py
  function to_tensor (line 12) | def to_tensor(
  class ToTensor (line 43) | class ToTensor(BaseTransform):
    method __init__ (line 58) | def __init__(self, keys: Sequence[str]) -> None:
    method transform (line 61) | def transform(self, results: dict) -> dict:
    method __repr__ (line 83) | def __repr__(self) -> str:
  class ImageToTensor (line 88) | class ImageToTensor(BaseTransform):
    method __init__ (line 107) | def __init__(self, keys: dict) -> None:
    method transform (line 110) | def transform(self, results: dict) -> dict:
    method __repr__ (line 127) | def __repr__(self) -> str:

FILE: mmcv/transforms/loading.py
  class LoadImageFromFile (line 14) | class LoadImageFromFile(BaseTransform):
    method __init__ (line 53) | def __init__(self,
    method transform (line 81) | def transform(self, results: dict) -> Optional[dict]:
    method __repr__ (line 119) | def __repr__(self):
  class LoadAnnotations (line 135) | class LoadAnnotations(BaseTransform):
    method __init__ (line 225) | def __init__(
    method _load_bboxes (line 258) | def _load_bboxes(self, results: dict) -> None:
    method _load_labels (line 274) | def _load_labels(self, results: dict) -> None:
    method _load_seg_map (line 290) | def _load_seg_map(self, results: dict) -> None:
    method _load_kps (line 312) | def _load_kps(self, results: dict) -> None:
    method transform (line 328) | def transform(self, results: dict) -> dict:
    method __repr__ (line 350) | def __repr__(self) -> str:

FILE: mmcv/transforms/processing.py
  class Normalize (line 22) | class Normalize(BaseTransform):
    method __init__ (line 51) | def __init__(self,
    method transform (line 59) | def transform(self, results: dict) -> dict:
    method __repr__ (line 76) | def __repr__(self) -> str:
  class Resize (line 83) | class Resize(BaseTransform):
    method __init__ (line 132) | def __init__(self,
    method _resize_img (line 167) | def _resize_img(self, results: dict) -> None:
    method _resize_bboxes (line 196) | def _resize_bboxes(self, results: dict) -> None:
    method _resize_seg (line 208) | def _resize_seg(self, results: dict) -> None:
    method _resize_keypoints (line 225) | def _resize_keypoints(self, results: dict) -> None:
    method transform (line 239) | def transform(self, results: dict) -> dict:
    method __repr__ (line 263) | def __repr__(self):
  class Pad (line 275) | class Pad(BaseTransform):
    method __init__ (line 333) | def __init__(self,
    method _pad_img (line 358) | def _pad_img(self, results: dict) -> None:
    method _pad_seg (line 390) | def _pad_seg(self, results: dict) -> None:
    method transform (line 404) | def transform(self, results: dict) -> dict:
    method __repr__ (line 417) | def __repr__(self):
  class CenterCrop (line 428) | class CenterCrop(BaseTransform):
    method __init__ (line 468) | def __init__(self,
    method _crop_img (line 494) | def _crop_img(self, results: dict, bboxes: np.ndarray) -> None:
    method _crop_seg_map (line 508) | def _crop_seg_map(self, results: dict, bboxes: np.ndarray) -> None:
    method _crop_bboxes (line 519) | def _crop_bboxes(self, results: dict, bboxes: np.ndarray) -> None:
    method _crop_keypoints (line 540) | def _crop_keypoints(self, results: dict, bboxes: np.ndarray) -> None:
    method transform (line 569) | def transform(self, results: dict) -> dict:
    method __repr__ (line 616) | def __repr__(self) -> str:
  class RandomGrayscale (line 626) | class RandomGrayscale(BaseTransform):
    method __init__ (line 656) | def __init__(self,
    method _random_prob (line 671) | def _random_prob(self):
    method transform (line 674) | def transform(self, results: dict) -> dict:
    method __repr__ (line 711) | def __repr__(self) -> str:
  class MultiScaleFlipAug (line 721) | class MultiScaleFlipAug(BaseTransform):
    method __init__ (line 789) | def __init__(
    method transform (line 827) | def transform(self, results: dict) -> Dict:
    method __repr__ (line 867) | def __repr__(self) -> str:
  class TestTimeAug (line 877) | class TestTimeAug(BaseTransform):
    method __init__ (line 933) | def __init__(self, transforms: list):
    method transform (line 950) | def transform(self, results: dict) -> dict:
    method __repr__ (line 992) | def __repr__(self) -> str:
  class RandomChoiceResize (line 1001) | class RandomChoiceResize(BaseTransform):
    method __init__ (line 1055) | def __init__(
    method _random_select (line 1073) | def _random_select(self) -> Tuple[int, int]:
    method transform (line 1086) | def transform(self, results: dict) -> dict:
    method __repr__ (line 1104) | def __repr__(self) -> str:
  class RandomFlip (line 1112) | class RandomFlip(BaseTransform):
    method __init__ (line 1169) | def __init__(self,
    method _flip_bbox (line 1198) | def _flip_bbox(self, bboxes: np.ndarray, img_shape: Tuple[int, int],
    method _flip_keypoints (line 1231) | def _flip_keypoints(
    method _flip_seg_map (line 1267) | def _flip_seg_map(self, seg_map: dict, direction: str) -> np.ndarray:
    method _choose_direction (line 1295) | def _choose_direction(self) -> str:
    method _flip (line 1318) | def _flip(self, results: dict) -> None:
    method _flip_on_direction (line 1344) | def _flip_on_direction(self, results: dict) -> None:
    method transform (line 1356) | def transform(self, results: dict) -> dict:
    method __repr__ (line 1372) | def __repr__(self) -> str:
  class RandomResize (line 1381) | class RandomResize(BaseTransform):
    method __init__ (line 1457) | def __init__(
    method _random_sample (line 1473) | def _random_sample(scales: Sequence[Tuple[int, int]]) -> tuple:
    method _random_sample_ratio (line 1494) | def _random_sample_ratio(scale: tuple, ratio_range: Tuple[float,
    method _random_scale (line 1519) | def _random_scale(self) -> tuple:
    method transform (line 1540) | def transform(self, results: dict) -> dict:
    method __repr__ (line 1557) | def __repr__(self) -> str:

FILE: mmcv/transforms/utils.py
  class cache_randomness (line 15) | class cache_randomness:
    method __init__ (line 28) | def __init__(self, func):
    method __set_name__ (line 44) | def __set_name__(self, owner, name):
    method __call__ (line 54) | def __call__(self, *args, **kwargs):
    method __get__ (line 82) | def __get__(self, obj, cls):
  function avoid_cache_randomness (line 90) | def avoid_cache_randomness(cls):
  function cache_random_params (line 136) | def cache_random_params(transforms: Union[BaseTransform, Iterable]):

FILE: mmcv/transforms/wrappers.py
  function nullcontext (line 32) | def nullcontext(resource=None):
  class Compose (line 40) | class Compose(BaseTransform):
    method __init__ (line 58) | def __init__(self, transforms: Union[Transform, Sequence[Transform]]):
    method __iter__ (line 74) | def __iter__(self):
    method transform (line 78) | def transform(self, results: Dict) -> Optional[Dict]:
    method __repr__ (line 93) | def __repr__(self):
  class KeyMapper (line 103) | class KeyMapper(BaseTransform):
    method __init__ (line 179) | def __init__(self,
    method __iter__ (line 207) | def __iter__(self):
    method _map_input (line 211) | def _map_input(self, data: Dict,
    method _map_output (line 259) | def _map_output(self, data: Dict,
    method _apply_transforms (line 311) | def _apply_transforms(self, inputs: Dict) -> Dict:
    method transform (line 330) | def transform(self, results: Dict) -> Dict:
    method __repr__ (line 343) | def __repr__(self) -> str:
  class TransformBroadcaster (line 354) | class TransformBroadcaster(KeyMapper):
    method __init__ (line 449) | def __init__(self,
    method scatter_sequence (line 461) | def scatter_sequence(self, data: Dict) -> List[Dict]:
    method transform (line 495) | def transform(self, results: Dict):
    method __repr__ (line 531) | def __repr__(self) -> str:
  class RandomChoice (line 543) | class RandomChoice(BaseTransform):
    method __init__ (line 566) | def __init__(self,
    method __iter__ (line 582) | def __iter__(self):
    method random_pipeline_index (line 586) | def random_pipeline_index(self) -> int:
    method transform (line 591) | def transform(self, results: Dict) -> Optional[Dict]:
    method __repr__ (line 596) | def __repr__(self) -> str:
  class RandomApply (line 604) | class RandomApply(BaseTransform):
    method __init__ (line 621) | def __init__(self,
    method __iter__ (line 629) | def __iter__(self):
    method random_apply (line 633) | def random_apply(self) -> bool:
    method transform (line 638) | def transform(self, results: Dict) -> Optional[Dict]:
    method __repr__ (line 645) | def __repr__(self) -> str:

FILE: mmcv/utils/env.py
  function collect_env (line 13) | def collect_env():

FILE: mmcv/utils/ext_loader.py
  function load_ext (line 12) | def load_ext(name, funcs):
  function get_fake_func (line 42) | def get_fake_func(name, e):
  function load_ext (line 50) | def load_ext(name, funcs):
  function check_ops_exist (line 70) | def check_ops_exist() -> bool:

FILE: mmcv/utils/parrots_jit.py
  function jit (line 12) | def jit(func=None,
  function skip_no_elena (line 36) | def skip_no_elena(func):

FILE: mmcv/version.py
  function parse_version_info (line 5) | def parse_version_info(version_str: str, length: int = 4) -> tuple:

FILE: mmcv/video/io.py
  class Cache (line 13) | class Cache:
    method __init__ (line 15) | def __init__(self, capacity):
    method capacity (line 22) | def capacity(self):
    method size (line 26) | def size(self):
    method put (line 29) | def put(self, key, val):
    method get (line 36) | def get(self, key, default=None):
  class VideoReader (line 41) | class VideoReader:
    method __init__ (line 62) | def __init__(self, filename, cache_capacity=10):
    method vcap (line 78) | def vcap(self):
    method opened (line 83) | def opened(self):
    method width (line 88) | def width(self):
    method height (line 93) | def height(self):
    method resolution (line 98) | def resolution(self):
    method fps (line 103) | def fps(self):
    method frame_cnt (line 108) | def frame_cnt(self):
    method fourcc (line 113) | def fourcc(self):
    method position (line 118) | def position(self):
    method _get_real_position (line 122) | def _get_real_position(self):
    method _set_real_position (line 125) | def _set_real_position(self, frame_id):
    method read (line 132) | def read(self):
    method get_frame (line 158) | def get_frame(self, frame_id):
    method current_frame (line 185) | def current_frame(self):
    method cvt2frames (line 196) | def cvt2frames(self,
    method __len__ (line 238) | def __len__(self):
    method __getitem__ (line 241) | def __getitem__(self, index):
    method __iter__ (line 254) | def __iter__(self):
    method __next__ (line 258) | def __next__(self):
    method __enter__ (line 267) | def __enter__(self):
    method __exit__ (line 270) | def __exit__(self, exc_type, exc_value, traceback):
  function frames2video (line 274) | def frames2video(frame_dir: str,

FILE: mmcv/video/optflow.py
  function flowread (line 13) | def flowread(flow_or_path: Union[np.ndarray, str],
  function flowwrite (line 66) | def flowwrite(flow: np.ndarray,
  function quantize_flow (line 101) | def quantize_flow(flow: np.ndarray,
  function dequantize_flow (line 131) | def dequantize_flow(dx: np.ndarray,
  function flow_warp (line 158) | def flow_warp(img: np.ndarray,
  function flow_from_bytes (line 222) | def flow_from_bytes(content: bytes) -> np.ndarray:
  function sparse_flow_from_bytes (line 252) | def sparse_flow_from_bytes(content: bytes) -> Tuple[np.ndarray, np.ndarr...

FILE: mmcv/video/processing.py
  function convert_video (line 12) | def convert_video(in_file: str,
  function resize_video (line 56) | def resize_video(in_file: str,
  function cut_video (line 94) | def cut_video(in_file: str,
  function concat_video (line 129) | def concat_video(video_list: List,

FILE: mmcv/visualization/color.py
  class Color (line 9) | class Color(Enum):
  function color_val (line 24) | def color_val(color: Union[Color, str, tuple, int, np.ndarray]) -> tuple:

FILE: mmcv/visualization/image.py
  function imshow (line 14) | def imshow(img: Union[str, np.ndarray],
  function imshow_bboxes (line 37) | def imshow_bboxes(img: Union[str, np.ndarray],
  function imshow_det_bboxes (line 91) | def imshow_det_bboxes(img: Union[str, np.ndarray],

FILE: mmcv/visualization/optflow.py
  function flowshow (line 11) | def flowshow(flow: Union[np.ndarray, str],
  function flow2rgb (line 26) | def flow2rgb(flow: np.ndarray,
  function make_color_wheel (line 80) | def make_color_wheel(bins: Optional[Union[list, tuple]] = None) -> np.nd...

FILE: setup.py
  function choose_requirement (line 31) | def choose_requirement(primary, secondary):
  function get_version (line 43) | def get_version():
  function parse_requirements (line 50) | def parse_requirements(fname='requirements/runtime.txt', with_version=Tr...
  function get_extensions (line 141) | def get_extensions():

FILE: tests/test_arraymisc.py
  function test_quantize (line 9) | def test_quantize():
  function test_dequantize (line 34) | def test_dequantize():
  function test_joint (line 57) | def test_joint():

FILE: tests/test_cnn/test_build_layers.py
  function test_build_conv_layer (line 24) | def test_build_conv_layer():
  function test_infer_norm_abbr (line 88) | def test_infer_norm_abbr():
  function test_build_norm_layer (line 125) | def test_build_norm_layer():
  function test_build_activation_layer (line 185) | def test_build_activation_layer():
  function test_build_padding_layer (line 234) | def test_build_padding_layer():
  function test_upsample_layer (line 270) | def test_upsample_layer():
  function test_pixel_shuffle_pack (line 336) | def test_pixel_shuffle_pack():
  function test_is_norm (line 344) | def test_is_norm():
  function test_infer_plugin_abbr (line 379) | def test_infer_plugin_abbr():
  function test_build_plugin_layer (line 396) | def test_build_plugin_layer():

FILE: tests/test_cnn/test_context_block.py
  function test_context_block (line 8) | def test_context_block():

FILE: tests/test_cnn/test_conv2d_adaptive_padding.py
  function test_conv2d_samepadding (line 7) | def test_conv2d_samepadding():

FILE: tests/test_cnn/test_conv_module.py
  class ExampleConv (line 16) | class ExampleConv(nn.Module):
    method __init__ (line 18) | def __init__(self,
    method forward (line 44) | def forward(self, x):
    method init_weights (line 48) | def init_weights(self):
  function test_conv_module (line 52) | def test_conv_module():
  function test_bias (line 186) | def test_bias():
  function conv_forward (line 222) | def conv_forward(self, x):
  function bn_forward (line 226) | def bn_forward(self, x):
  function relu_forward (line 230) | def relu_forward(self, x):
  function test_order (line 237) | def test_order():

FILE: tests/test_cnn/test_depthwise_seperable_conv_module.py
  function test_depthwise_separable_conv (line 9) | def test_depthwise_separable_conv():

FILE: tests/test_cnn/test_flops_counter.py
  class ExampleModel (line 52) | class ExampleModel(nn.Module):
    method __init__ (line 54) | def __init__(self):
    method forward (line 58) | def forward(self, imgs):
  function input_constructor (line 63) | def input_constructor(x):
  function test_flops_counter (line 67) | def test_flops_counter():
  function test_flops_to_string (line 122) | def test_flops_to_string():
  function test_params_to_string (line 140) | def test_params_to_string():

FILE: tests/test_cnn/test_fuse_conv_bn.py
  function test_fuse_conv_bn (line 8) | def test_fuse_conv_bn():

FILE: tests/test_cnn/test_generalized_attention.py
  function test_context_block (line 7) | def test_context_block():

FILE: tests/test_cnn/test_hsigmoid.py
  function test_hsigmoid (line 8) | def test_hsigmoid():

FILE: tests/test_cnn/test_hswish.py
  function test_hswish (line 8) | def test_hswish():

FILE: tests/test_cnn/test_non_local.py
  function test_nonlocal (line 10) | def test_nonlocal():
  function test_nonlocal3d (line 24) | def test_nonlocal3d():
  function test_nonlocal2d (line 89) | def test_nonlocal2d():
  function test_nonlocal1d (line 156) | def test_nonlocal1d():

FILE: tests/test_cnn/test_rfsearch/test_operator.py
  function test_rfsearch_operator_3x3 (line 22) | def test_rfsearch_operator_3x3():
  function test_rfsearch_operator_5x5 (line 71) | def test_rfsearch_operator_5x5():
  function test_rfsearch_operator_5x5_branch3 (line 119) | def test_rfsearch_operator_5x5_branch3():
  function test_rfsearch_operator_1x5 (line 171) | def test_rfsearch_operator_1x5():
  function test_rfsearch_operator_5x5_d2x2 (line 223) | def test_rfsearch_operator_5x5_d2x2():
  function test_rfsearch_operator_5x5_d1x2 (line 276) | def test_rfsearch_operator_5x5_d1x2():

FILE: tests/test_cnn/test_rfsearch/test_search.py
  function test_rfsearchhook (line 8) | def test_rfsearchhook():

FILE: tests/test_cnn/test_scale.py
  function test_scale (line 8) | def test_scale():
  function test_layer_scale (line 26) | def test_layer_scale():

FILE: tests/test_cnn/test_silu.py
  function test_silu (line 7) | def test_silu():

FILE: tests/test_cnn/test_swish.py
  function test_swish (line 8) | def test_swish():

FILE: tests/test_cnn/test_transformer.py
  function test_adaptive_padding (line 16) | def test_adaptive_padding():
  function test_patch_embed (line 108) | def test_patch_embed():
  function test_patch_merging (line 322) | def test_patch_merging():
  function test_multiheadattention (line 472) | def test_multiheadattention():
  function test_ffn (line 537) | def test_ffn():
  function test_basetransformerlayer_cuda (line 564) | def test_basetransformerlayer_cuda():
  function test_basetransformerlayer (line 586) | def test_basetransformerlayer(embed_dims):
  function test_transformerlayersequence (line 635) | def test_transformerlayersequence():
  function test_drop_path (line 677) | def test_drop_path():

FILE: tests/test_cnn/test_wrappers.py
  function test_conv2d (line 23) | def test_conv2d(in_w, in_h, in_channel, out_channel, kernel_size, stride,
  function test_conv3d (line 80) | def test_conv3d(in_w, in_h, in_t, in_channel, out_channel, kernel_size, ...
  function test_conv_transposed_2d (line 138) | def test_conv_transposed_2d(in_w, in_h, in_channel, out_channel, kernel_...
  function test_conv_transposed_3d (line 197) | def test_conv_transposed_3d(in_w, in_h, in_t, in_channel, out_channel,
  function test_max_pool_2d (line 254) | def test_max_pool_2d(in_w, in_h, in_channel, out_channel, kernel_size, s...
  function test_max_pool_3d (line 281) | def test_max_pool_3d(in_w, in_h, in_t, in_channel, out_channel, kernel_s...
  function test_linear (line 307) | def test_linear(in_w, in_h, in_feature, out_feature):
  function test_nn_op_forward_called (line 337) | def test_nn_op_forward_called():
  function test_fx_compatibility (line 384) | def test_fx_compatibility():

FILE: tests/test_image/test_colorspace.py
  function test_bgr2gray (line 12) | def test_bgr2gray():
  function test_rgb2gray (line 24) | def test_rgb2gray():
  function test_gray2bgr (line 36) | def test_gray2bgr():
  function test_gray2rgb (line 44) | def test_gray2rgb():
  function test_bgr2rgb (line 52) | def test_bgr2rgb():
  function test_rgb2bgr (line 61) | def test_rgb2bgr():
  function test_bgr2hsv (line 70) | def test_bgr2hsv():
  function test_convert_input_type_range (line 92) | def test_convert_input_type_range():
  function test_convert_output_type_range (line 109) | def test_convert_output_type_range():
  function assert_image_almost_equal (line 126) | def assert_image_almost_equal(x, y, atol=1):
  function test_rgb2ycbcr (line 132) | def test_rgb2ycbcr():
  function test_bgr2ycbcr (line 190) | def test_bgr2ycbcr():
  function test_ycbcr2rgb (line 244) | def test_ycbcr2rgb():
  function test_ycbcr2bgr (line 282) | def test_ycbcr2bgr():
  function test_bgr2hls (line 315) | def test_bgr2hls():
  function test_imconvert (line 353) | def test_imconvert(src, dst, ref):

FILE: tests/test_image/test_geometric.py
  class TestGeometric (line 12) | class TestGeometric:
    method setup_class (line 15) | def setup_class(cls):
    method test_imresize (line 21) | def test_imresize(self):
    method test_imresize_to_multiple (line 50) | def test_imresize_to_multiple(self):
    method test_imresize_like (line 99) | def test_imresize_like(self):
    method test_rescale_size (line 104) | def test_rescale_size(self):
    method test_imrescale (line 132) | def test_imrescale(self):
    method test_imflip (line 159) | def test_imflip(self):
    method test_imflip_ (line 213) | def test_imflip_(self):
    method test_imcrop (line 291) | def test_imcrop(self):
    method test_impad (line 333) | def test_impad(self):
    method test_impad_to_multiple (line 468) | def test_impad_to_multiple(self):
    method test_cutout (line 479) | def test_cutout(self):
    method test_imrotate (line 513) | def test_imrotate(self):
    method test_imshear (line 537) | def test_imshear(self):
    method test_imtranslate (line 578) | def test_imtranslate(self):

FILE: tests/test_image/test_image_misc.py
  function test_tensor2imgs (line 15) | def test_tensor2imgs():

FILE: tests/test_image/test_io.py
  class TestIO (line 23) | class TestIO:
    method setup_class (line 26) | def setup_class(cls):
    method teardown_class (line 48) | def teardown_class(cls):
    method assert_img_equal (line 52) | def assert_img_equal(self, img, ref_img, ratio_thr=0.999):
    method test_imread (line 59) | def test_imread(self):
    method test_imfrombytes (line 292) | def test_imfrombytes(self):
    method test_imwrite (line 386) | def test_imwrite(self):
    method test_no_turbojpeg (line 426) | def test_no_turbojpeg(self):
    method test_no_pillow (line 433) | def test_no_pillow(self):

FILE: tests/test_image/test_photometric.py
  class TestPhotometric (line 12) | class TestPhotometric:
    method setup_class (line 15) | def setup_class(cls):
    method test_imnormalize (line 22) | def test_imnormalize(self):
    method test_imnormalize_ (line 32) | def test_imnormalize_(self):
    method test_imdenormalize (line 44) | def test_imdenormalize(self):
    method test_iminvert (line 53) | def test_iminvert(self):
    method test_solarize (line 60) | def test_solarize(self):
    method test_posterize (line 70) | def test_posterize(self):
    method test_adjust_color (line 80) | def test_adjust_color(self, nb_rand_test=100):
    method test_imequalize (line 128) | def test_imequalize(self, nb_rand_test=100):
    method test_adjust_brightness (line 156) | def test_adjust_brightness(self, nb_rand_test=100):
    method test_adjust_contrast (line 186) | def test_adjust_contrast(self, nb_rand_test=100):
    method test_auto_contrast (line 221) | def test_auto_contrast(self, nb_rand_test=100):
    method test_adjust_sharpness (line 266) | def test_adjust_sharpness(self, nb_rand_test=100):
    method test_adjust_lighting (line 311) | def test_adjust_lighting(self):
    method test_lut_transform (line 333) | def test_lut_transform(self):
    method test_clahe (line 364) | def test_clahe(self):
    method test_adjust_hue (line 398) | def test_adjust_hue(self):

FILE: tests/test_ops/test_active_rotated_filter.py
  function test_active_rotated_filter (line 259) | def test_active_rotated_filter(device):

FILE: tests/test_ops/test_assign_score_withk.py
  function test_paconv_assign_scores (line 23) | def test_paconv_assign_scores(device):

FILE: tests/test_ops/test_ball_query.py
  function test_ball_query (line 28) | def test_ball_query(device):
  function test_stack_ball_query (line 85) | def test_stack_ball_query(device):

FILE: tests/test_ops/test_bbox.py
  class TestBBox (line 11) | class TestBBox:
    method _test_bbox_overlaps (line 13) | def _test_bbox_overlaps(self, device='cpu', dtype=torch.float):
    method test_bbox_overlaps_float (line 72) | def test_bbox_overlaps_float(self, device):
    method test_bbox_overlaps_half (line 93) | def test_bbox_overlaps_half(self, device):

FILE: tests/test_ops/test_bezier_align.py
  function test_bezieralign (line 44) | def test_bezieralign(device, dtype):

FILE: tests/test_ops/test_bias_act.py
  class TestBiasAct (line 17) | class TestBiasAct:
    method setup_class (line 20) | def setup_class(cls):
    method test_bias_act_cpu (line 24) | def test_bias_act_cpu(self):
    method test_bias_act_cuda (line 68) | def test_bias_act_cuda(self):
    method test_bias_act_musa (line 136) | def test_bias_act_musa(self):
    method test_easy_dict (line 203) | def test_easy_dict(self):
    method test_easy_dict_musa (line 215) | def test_easy_dict_musa(self):

FILE: tests/test_ops/test_bilinear_grid_sample.py
  class TestBilinearGridSample (line 7) | class TestBilinearGridSample:
    method _test_bilinear_grid_sample (line 9) | def _test_bilinear_grid_sample(self,
    method test_bilinear_grid_sample (line 28) | def test_bilinear_grid_sample(self):

FILE: tests/test_ops/test_border_align.py
  function _test_border_align_allclose (line 53) | def _test_border_align_allclose(device, dtype, pool_size):
  function test_border_align (line 102) | def test_border_align(device, dtype, pool_size):

FILE: tests/test_ops/test_box_iou_quadri.py
  class TestBoxIoUQuadri (line 9) | class TestBoxIoUQuadri:
    method test_box_iou_quadri_cuda (line 26) | def test_box_iou_quadri_cuda(self, device):
    method test_box_iou_quadri_iof_cuda (line 68) | def test_box_iou_quadri_iof_cuda(self, device):

FILE: tests/test_ops/test_box_iou_rotated.py
  class TestBoxIoURotated (line 11) | class TestBoxIoURotated:
    method test_box_iou_rotated_cpu (line 13) | def test_box_iou_rotated_cpu(self):
    method test_box_iou_rotated (line 68) | def test_box_iou_rotated(self, device):
    method test_box_iou_rotated_iof_cpu (line 105) | def test_box_iou_rotated_iof_cpu(self):
    method test_box_iou_rotated_iof (line 159) | def test_box_iou_rotated_iof(self, device):

FILE: tests/test_ops/test_carafe.py
  class TestCarafe (line 10) | class TestCarafe:
    method test_carafe_naive_gradcheck (line 12) | def test_carafe_naive_gradcheck(self):
    method test_carafe_gradcheck (line 24) | def test_carafe_gradcheck(self):
    method test_carafe_allclose (line 50) | def test_carafe_allclose(self, device):

FILE: tests/test_ops/test_cc_attention.py
  class Loss (line 9) | class Loss(nn.Module):
    method __init__ (line 11) | def __init__(self):
    method forward (line 14) | def forward(self, input, target):
  class TestCrissCrossAttention (line 20) | class TestCrissCrossAttention:
    method test_cc_attention (line 22) | def test_cc_attention(self):

FILE: tests/test_ops/test_chamfer_distance.py
  function chamfer_distance_forward_groundtruth (line 10) | def chamfer_distance_forward_groundtruth(xyz1, xyz2, dtype):
  function torch_to_np_type (line 39) | def torch_to_np_type(dtype):
  function test_chamfer_distance_npu_dynamic_shape (line 62) | def test_chamfer_distance_npu_dynamic_shape(dtype, device, shape):

FILE: tests/test_ops/test_contour_expand.py
  function test_contour_expand (line 6) | def test_contour_expand():

FILE: tests/test_ops/test_conv_gradfix.py
  class TestCond2d (line 11) | class TestCond2d:
    method setup_class (line 14) | def setup_class(cls):
    method test_conv2d_cuda (line 19) | def test_conv2d_cuda(self):
    method test_conv2d_musa (line 28) | def test_conv2d_musa(self):
  class TestCond2dTansposed (line 37) | class TestCond2dTansposed:
    method setup_class (line 40) | def setup_class(cls):
    method test_conv2d_transposed_cuda (line 45) | def test_conv2d_transposed_cuda(self):
    method test_conv2d_transposed_musa (line 56) | def test_conv2d_transposed_musa(self):

FILE: tests/test_ops/test_convex_iou.py
  function test_convex_iou (line 40) | def test_convex_iou():
  function test_convex_giou (line 50) | def test_convex_giou():
  function test_convex_miou (line 61) | def test_convex_miou():

FILE: tests/test_ops/test_corner_pool.py
  function test_corner_pool_device_and_dtypes_cpu (line 12) | def test_corner_pool_device_and_dtypes_cpu():

FILE: tests/test_ops/test_correlation.py
  function assert_equal_tensor (line 16) | def assert_equal_tensor(tensor_a, tensor_b):
  class TestCorrelation (line 21) | class TestCorrelation:
    method _test_correlation (line 23) | def _test_correlation(self, dtype=torch.float):
    method test_correlation (line 52) | def test_correlation(self):

FILE: tests/test_ops/test_deform_conv.py
  class TestDeformconv (line 50) | class TestDeformconv:
    method _test_deformconv (line 52) | def _test_deformconv(self,
    method _test_amp_deformconv (line 127) | def _test_amp_deformconv(self,
    method test_deformconv_float (line 226) | def test_deformconv_float(self, device, threshold):
    method test_deformconv_double (line 249) | def test_deformconv_double(self, device):
    method test_deformconv_half (line 269) | def test_deformconv_half(self, device, threshold):

FILE: tests/test_ops/test_deform_roi_pool.py
  class TestDeformRoIPool (line 42) | class TestDeformRoIPool:
    method test_deform_roi_pool_gradcheck (line 44) | def test_deform_roi_pool_gradcheck(self):
    method test_modulated_deform_roi_pool_gradcheck (line 72) | def test_modulated_deform_roi_pool_gradcheck(self):
    method _test_deform_roi_pool_allclose (line 101) | def _test_deform_roi_pool_allclose(self, device, dtype=torch.float):
    method test_deform_roi_pool_allclose (line 156) | def test_deform_roi_pool_allclose(self, device, dtype):

FILE: tests/test_ops/test_diff_iou_rotated.py
  function test_diff_iou_rotated_2d (line 28) | def test_diff_iou_rotated_2d(device):
  function test_diff_iou_rotated_3d (line 61) | def test_diff_iou_rotated_3d(device):

FILE: tests/test_ops/test_filtered_lrelu.py
  class TestFilteredLrelu (line 11) | class TestFilteredLrelu:
    method setup_class (line 14) | def setup_class(cls):
    method test_filtered_lrelu_cpu (line 20) | def test_filtered_lrelu_cpu(self):
    method test_filtered_lrelu_cuda (line 123) | def test_filtered_lrelu_cuda(self):
    method test_filtered_lrelu_musa (line 228) | def test_filtered_lrelu_musa(self):

FILE: tests/test_ops/test_focal_loss.py
  class Testfocalloss (line 41) | class Testfocalloss:
    method _test_softmax (line 43) | def _test_softmax(self, dtype=torch.float):
    method _test_sigmoid (line 64) | def _test_sigmoid(self, device, dtype=torch.float):
    method _test_grad_softmax (line 83) | def _test_grad_softmax(self, dtype=torch.float):
    method _test_grad_sigmoid (line 105) | def _test_grad_sigmoid(self, dtype=torch.float):
    method test_softmax_float (line 127) | def test_softmax_float(self):
    method test_softmax_half (line 130) | def test_softmax_half(self):
    method test_sigmoid_float (line 151) | def test_sigmoid_float(self, device):
    method test_sigmoid_half (line 172) | def test_sigmoid_half(self, device):
    method test_grad_softmax_float (line 175) | def test_grad_softmax_float(self):
    method test_grad_sigmoid_float (line 178) | def test_grad_sigmoid_float(self):

FILE: tests/test_ops/test_furthest_point_sample.py
  function test_fps (line 23) | def test_fps(device):
  function test_fps_with_dist (line 52) | def test_fps_with_dist(device):

FILE: tests/test_ops/test_fused_bias_leakyrelu.py
  class TestFusedBiasLeakyReLU (line 15) | class TestFusedBiasLeakyReLU:
    method setup_class (line 18) | def setup_class(cls):
    method test_gradient (line 49) | def test_gradient(self, device):
    method test_gradgradient (line 80) | def test_gradgradient(self, device):

FILE: tests/test_ops/test_gather_points.py
  class TestGatherPoints (line 9) | class TestGatherPoints:
    method test_gather_points_all_close (line 25) | def test_gather_points_all_close(self, device):

FILE: tests/test_ops/test_group_points.py
  function test_grouping_points (line 34) | def test_grouping_points(dtype, device):
  function test_stack_grouping_points (line 142) | def test_stack_grouping_points(dtype, device):

FILE: tests/test_ops/test_info.py
  class TestInfo (line 5) | class TestInfo:
    method test_info (line 7) | def test_info(self):

FILE: tests/test_ops/test_iou3d.py
  function test_boxes_overlap_bev (line 17) | def test_boxes_overlap_bev(device):
  function test_boxes_iou3d (line 56) | def test_boxes_iou3d(device):
  function test_nms3d (line 95) | def test_nms3d(device):
  function test_nms3d_normal (line 142) | def test_nms3d_normal(device):

FILE: tests/test_ops/test_knn.py
  function test_knn (line 19) | def test_knn(device):

FILE: tests/test_ops/test_masked_conv2d.py
  class TestMaskedConv2d (line 13) | class TestMaskedConv2d:
    method test_masked_conv2d_all_close (line 29) | def test_masked_conv2d_all_close(self, device):

FILE: tests/test_ops/test_merge_cells.py
  function test_sum_cell (line 22) | def test_sum_cell(inputs_x, inputs_y):
  function test_concat_cell (line 36) | def test_concat_cell(inputs_x, inputs_y):
  function test_global_pool_cell (line 50) | def test_global_pool_cell(inputs_x, inputs_y):
  function test_resize_methods (line 61) | def test_resize_methods(target_size):

FILE: tests/test_ops/test_min_area_polygons.py
  function test_min_area_polygons (line 33) | def test_min_area_polygons(device):

FILE: tests/test_ops/test_modulated_deform_conv.py
  class TestMdconv (line 49) | class TestMdconv:
    method _test_mdconv (line 51) | def _test_mdconv(self, dtype=torch.float, device='cuda'):
    method _test_amp_mdconv (line 87) | def _test_amp_mdconv(self, input_dtype=torch.float, device='cuda'):
    method test_mdconv_float (line 145) | def test_mdconv_float(self, device):
    method test_mdconv_double (line 159) | def test_mdconv_double(self, device):
    method test_mdconv_half (line 176) | def test_mdconv_half(self, device):

FILE: tests/test_ops/test_ms_deformable_attn.py
  function test_multiscale_deformable_attention (line 51) | def test_multiscale_deformable_attention(device):
  function test_forward_multi_scale_deformable_attn_pytorch (line 100) | def test_forward_multi_scale_deformable_attn_pytorch():
  function test_forward_equal_with_pytorch_double (line 120) | def test_forward_equal_with_pytorch_double():
  function test_forward_equal_with_pytorch_npu (line 153) | def test_forward_equal_with_pytorch_npu():
  function test_forward_equal_with_pytorch_float (line 200) | def test_forward_equal_with_pytorch_float(device):
  function test_forward_equal_with_autocast (line 243) | def test_forward_equal_with_autocast(device):
  function test_gradient_numerical (line 323) | def test_gradient_numerical(channels,
  function test_backward_equal_with_pytorch_npu (line 376) | def test_backward_equal_with_pytorch_npu():

FILE: tests/test_ops/test_nms.py
  class Testnms (line 10) | class Testnms:
    method test_nms_allclose (line 26) | def test_nms_allclose(self, device):
    method test_softnms_allclose (line 46) | def test_softnms_allclose(self):
    method test_nms_match (line 115) | def test_nms_match(self):
    method test_batched_nms (line 151) | def test_batched_nms(self):

FILE: tests/test_ops/test_nms_quadri.py
  class TestNMSQuadri (line 9) | class TestNMSQuadri:
    method test_ml_nms_quadri (line 22) | def test_ml_nms_quadri(self, device):
    method test_nms_quadri (line 56) | def test_nms_quadri(self, device):
    method test_batched_nms (line 87) | def test_batched_nms(self, device):

FILE: tests/test_ops/test_nms_rotated.py
  class TestNmsRotated (line 10) | class TestNmsRotated:
    method test_ml_nms_rotated (line 30) | def test_ml_nms_rotated(self, device):
    method test_nms_rotated (line 80) | def test_nms_rotated(self, device):
    method test_batched_nms (line 109) | def test_batched_nms(self):

FILE: tests/test_ops/test_onnx.py
  function run_before_and_after_test (line 16) | def run_before_and_after_test():
  class WrapFunction (line 28) | class WrapFunction(nn.Module):
    method __init__ (line 30) | def __init__(self, wrapped_function):
    method forward (line 34) | def forward(self, *args, **kwargs):
  function test_roialign (line 38) | def test_roialign():
  function test_roipool (line 108) | def test_roipool():
  function _test_symbolic (line 171) | def _test_symbolic(model, inputs, symbol_name):
  function test_border_align (line 187) | def test_border_align():
  function test_carafe (line 196) | def test_carafe():
  function test_deform_conv (line 204) | def test_deform_conv():
  function test_modulated_deform_conv (line 212) | def test_modulated_deform_conv():
  function test_deform_roi_pool (line 221) | def test_deform_roi_pool():
  function test_masked_conv (line 239) | def test_masked_conv():
  function test_pr_roi_pool (line 248) | def test_pr_roi_pool():
  function test_psa_mask (line 260) | def test_psa_mask():
  function test_roi_align_rotated (line 268) | def test_roi_align_rotated():
  function test_roi_feaeture_align (line 282) | def test_roi_feaeture_align():

FILE: tests/test_ops/test_pixel_group.py
  function test_pixel_group (line 6) | def test_pixel_group():

FILE: tests/test_ops/test_points_in_polygons.py
  function test_points_in_polygons (line 24) | def test_points_in_polygons(device):

FILE: tests/test_ops/test_prroi_pool.py
  class TestPrRoiPool (line 38) | class TestPrRoiPool:
    method test_roipool_gradcheck (line 50) | def test_roipool_gradcheck(self, device):
    method _test_roipool_allclose (line 70) | def _test_roipool_allclose(self, device, dtype=torch.float):
    method test_roipool_allclose_float (line 105) | def test_roipool_allclose_float(self, device):

FILE: tests/test_ops/test_psa_mask.py
  class Loss (line 11) | class Loss(nn.Module):
    method __init__ (line 13) | def __init__(self):
    method forward (line 16) | def forward(self, input, target):
  class TestPSAMask (line 22) | class TestPSAMask:
    method test_psa_mask_collect (line 42) | def test_psa_mask_collect(self, device):
    method test_psa_mask_distribute (line 98) | def test_psa_mask_distribute(self, device):

FILE: tests/test_ops/test_riroi_align_rotated.py
  function test_roialign_rotated_gradcheck (line 58) | def test_roialign_rotated_gradcheck():
  function test_roialign_rotated_allclose (line 73) | def test_roialign_rotated_allclose():

FILE: tests/test_ops/test_roi_align.py
  function _test_roialign_gradcheck (line 45) | def _test_roialign_gradcheck(device, dtype):
  function _test_roialign_allclose (line 69) | def _test_roialign_allclose(device, dtype):
  function test_roialign_float (line 117) | def test_roialign_float(device, dtype):
  function test_roialign_float64 (line 128) | def test_roialign_float64(device):

FILE: tests/test_ops/test_roi_align_rotated.py
  function _test_roialign_rotated_gradcheck (line 54) | def _test_roialign_rotated_gradcheck(device, dtype):
  function _test_roialign_rotated_allclose (line 78) | def _test_roialign_rotated_allclose(device, dtype):
  function test_roialign_rotated (line 150) | def test_roialign_rotated(device, dtype):

FILE: tests/test_ops/test_roi_pool.py
  class TestRoiPool (line 36) | class TestRoiPool:
    method test_roipool_gradcheck (line 38) | def test_roipool_gradcheck(self):
    method _test_roipool_allclose (line 61) | def _test_roipool_allclose(self, device, dtype=torch.float):
    method test_roipool_allclose (line 108) | def test_roipool_allclose(self, device, dtype):

FILE: tests/test_ops/test_roiaware_pool3d.py
  function test_RoIAwarePool3d (line 34) | def test_RoIAwarePool3d(device, dtype):
  function test_points_in_boxes_part (line 79) | def test_points_in_boxes_part(device):
  function test_points_in_boxes_cpu (line 112) | def test_points_in_boxes_cpu():
  function test_points_in_boxes_all (line 148) | def test_points_in_boxes_all():

FILE: tests/test_ops/test_roipoint_pool3d.py
  function test_roipoint (line 36) | def test_roipoint(device, dtype):

FILE: tests/test_ops/test_rotated_feature_align.py
  function test_rotated_feature_align (line 34) | def test_rotated_feature_align(device):

FILE: tests/test_ops/test_saconv.py
  function test_sacconv (line 8) | def test_sacconv():

FILE: tests/test_ops/test_scatter_points.py
  function test_dynamic_scatter (line 27) | def test_dynamic_scatter(device):

FILE: tests/test_ops/test_spconv.py
  function make_sparse_convmodule (line 16) | def make_sparse_convmodule(in_channels,
  function test_make_sparse_convmodule (line 95) | def test_make_sparse_convmodule(device):

FILE: tests/test_ops/test_syncbn.py
  class TestSyncBN (line 19) | class TestSyncBN:
    method dist_init (line 21) | def dist_init(self):
    method _test_syncbn_train (line 51) | def _test_syncbn_train(self, size=1, half=False, device='cuda'):
    method _test_syncbn_empty_train (line 178) | def _test_syncbn_empty_train(self, size=1, half=False, device='cuda'):
    method test_syncbn_1 (line 299) | def test_syncbn_1(self):
    method test_syncbn_2 (line 302) | def test_syncbn_2(self):
    method test_syncbn_4 (line 305) | def test_syncbn_4(self):
    method test_syncbn_1_half (line 308) | def test_syncbn_1_half(self):
    method test_syncbn_2_half (line 311) | def test_syncbn_2_half(self):
    method test_syncbn_4_half (line 314) | def test_syncbn_4_half(self):
    method test_syncbn_empty_1 (line 317) | def test_syncbn_empty_1(self):
    method test_syncbn_empty_2 (line 320) | def test_syncbn_empty_2(self):
    method test_syncbn_empty_4 (line 323) | def test_syncbn_empty_4(self):
    method test_syncbn_empty_1_half (line 326) | def test_syncbn_empty_1_half(self):
    method test_syncbn_empty_2_half (line 329) | def test_syncbn_empty_2_half(self):
    method test_syncbn_empty_4_half (line 332) | def test_syncbn_empty_4_half(self):

FILE: tests/test_ops/test_three_interpolate.py
  function test_three_interpolate (line 31) | def test_three_interpolate(dtype, device):

FILE: tests/test_ops/test_three_nn.py
  function test_three_nn (line 59) | def test_three_nn(device, dtype, rtol):

FILE: tests/test_ops/test_tin_shift.py
  function _test_tinshift_gradcheck (line 136) | def _test_tinshift_gradcheck(device, dtype):
  function _test_tinshift_allclose (line 158) | def _test_tinshift_allclose(device, dtype):
  function _test_tinshift_assert (line 182) | def _test_tinshift_assert(device, dtype):
  function test_tinshift (line 227) | def test_tinshift(device, dtype):

FILE: tests/test_ops/test_upfirdn2d.py
  class TestUpFirDn2d (line 13) | class TestUpFirDn2d:
    method setup_class (line 22) | def setup_class(cls):
    method test_upfirdn2d (line 33) | def test_upfirdn2d(self):

FILE: tests/test_ops/test_voxelization.py
  function _get_voxel_points_indices (line 11) | def _get_voxel_points_indices(points, coors, voxel):
  function test_voxelization (line 23) | def test_voxelization(device_type):
  function test_voxelization_nondeterministic (line 68) | def test_voxelization_nondeterministic():
  function test_voxelization_mlu (line 157) | def test_voxelization_mlu(device_type):
  function test_voxelization_npu (line 192) | def test_voxelization_npu(device_type):
  function test_voxelization_musa (line 227) | def test_voxelization_musa(device_type):

FILE: tests/test_transforms/test_transforms_formatting.py
  function test_to_tensor (line 16) | def test_to_tensor():
  class TestToTensor (line 50) | class TestToTensor:
    method test_init (line 52) | def test_init(self):
    method test_transform (line 56) | def test_transform(self):
    method test_repr (line 72) | def test_repr(self):
  class TestImageToTensor (line 79) | class TestImageToTensor:
    method test_init (line 81) | def test_init(self):
    method test_transform (line 85) | def test_transform(self):
    method test_repr (line 98) | def test_repr(self):

FILE: tests/test_transforms/test_transforms_loading.py
  class TestLoadImageFromFile (line 11) | class TestLoadImageFromFile:
    method test_load_img (line 13) | def test_load_img(self):
  class TestLoadAnnotations (line 62) | class TestLoadAnnotations:
    method setup_class (line 64) | def setup_class(cls):
    method test_init (line 81) | def test_init(self):
    method test_load_bboxes (line 90) | def test_load_bboxes(self):
    method test_load_labels (line 103) | def test_load_labels(self):
    method test_load_kps (line 115) | def test_load_kps(self):
    method test_load_seg_map (line 128) | def test_load_seg_map(self):
    method test_repr (line 140) | def test_repr(self):

FILE: tests/test_transforms/test_transforms_processing.py
  class TestNormalize (line 25) | class TestNormalize:
    method test_normalize (line 27) | def test_normalize(self):
    method test_repr (line 44) | def test_repr(self):
  class TestResize (line 54) | class TestResize:
    method test_resize (line 56) | def test_resize(self):
    method test_repr (line 112) | def test_repr(self):
  class TestPad (line 120) | class TestPad:
    method test_pad (line 122) | def test_pad(self):
    method test_repr (line 229) | def test_repr(self):
  class TestCenterCrop (line 236) | class TestCenterCrop:
    method setup_class (line 239) | def setup_class(cls):
    method reset_results (line 247) | def reset_results(results, original_img, gt_semantic_map):
    method test_error (line 258) | def test_error(self):
    method test_repr (line 279) | def test_repr(self):
    method test_transform (line 285) | def test_transform(self):
    method test_torchvision_compare (line 453) | def test_torchvision_compare(self):
  class TestRandomGrayscale (line 472) | class TestRandomGrayscale:
    method setup_class (line 475) | def setup_class(cls):
    method test_repr (line 478) | def test_repr(self):
    method test_error (line 488) | def test_error(self):
    method test_transform (line 494) | def test_transform(self):
  class MockPackTaskInputs (line 531) | class MockPackTaskInputs(BaseTransform):
    method __init__ (line 533) | def __init__(self) -> None:
    method transform (line 536) | def transform(self, results):
  class TestMultiScaleFlipAug (line 541) | class TestMultiScaleFlipAug:
    method setup_class (line 544) | def setup_class(cls):
    method test_error (line 549) | def test_error(self):
    method test_multi_scale_flip_aug (line 567) | def test_multi_scale_flip_aug(self):
  class TestRandomChoiceResize (line 663) | class TestRandomChoiceResize:
    method setup_class (line 666) | def setup_class(cls):
    method reset_results (line 671) | def reset_results(self, results):
    method test_repr (line 675) | def test_repr(self):
    method test_error (line 682) | def test_error(self):
    method test_random_multiscale_resize (line 688) | def test_random_multiscale_resize(self):
  class TestRandomFlip (line 748) | class TestRandomFlip:
    method test_init (line 750) | def test_init(self):
    method test_transform (line 774) | def test_transform(self):
    method test_repr (line 850) | def test_repr(self):
  class TestRandomResize (line 856) | class TestRandomResize:
    method test_init (line 858) | def test_init(self):
    method test_repr (line 865) | def test_repr(self):
    method test_transform (line 873) | def test_transform(self):
  class TestTestTimeAug (line 931) | class TestTestTimeAug:
    method test_init (line 933) | def test_init(self):
    method test_transform (line 953) | def test_transform(self):
    method test_repr (line 998) | def test_repr(self):

FILE: tests/test_transforms/test_transforms_wrapper.py
  class AddToValue (line 16) | class AddToValue(BaseTransform):
    method __init__ (line 19) | def __init__(self, addend=0) -> None:
    method add (line 23) | def add(self, results, addend):
    method transform (line 41) | def transform(self, results):
    method __repr__ (line 44) | def __repr__(self) -> str:
  class RandomAddToValue (line 51) | class RandomAddToValue(AddToValue):
    method __init__ (line 54) | def __init__(self, repeat=1) -> None:
    method get_random_addend (line 59) | def get_random_addend(self):
    method transform (line 62) | def transform(self, results):
    method __repr__ (line 67) | def __repr__(self) -> str:
  class SumTwoValues (line 74) | class SumTwoValues(BaseTransform):
    method transform (line 77) | def transform(self, results):
    method __repr__ (line 88) | def __repr__(self) -> str:
  function test_compose (line 93) | def test_compose():
  function test_cache_random_parameters (line 120) | def test_cache_random_parameters():
  function test_key_mapper (line 158) | def test_key_mapper():
  function test_transform_broadcaster (line 320) | def test_transform_broadcaster():
  function test_random_choice (line 423) | def test_random_choice():
  function test_random_apply (line 467) | def test_random_apply():
  function test_utils (line 505) | def test_utils():

FILE: tests/test_utils/test_env.py
  function test_collect_env (line 9) | def test_collect_env():

FILE: tests/test_utils/test_parrots_jit.py
  class TestJit (line 13) | class TestJit:
    method test_add_dict (line 15) | def test_add_dict(self):
    method test_add_list (line 35) | def test_add_list(self):
    method test_jit_cache (line 65) | def test_jit_cache(self):
    method test_jit_shape (line 98) | def test_jit_shape(self):
    method test_jit_kwargs (line 119) | def test_jit_kwargs(self):
    method test_jit_derivate (line 135) | def test_jit_derivate(self):
    method test_jit_optimize (line 162) | def test_jit_optimize(self):
    method test_jit_coderize (line 179) | def test_jit_coderize(self):
    method test_jit_value_dependent (line 197) | def test_jit_value_dependent(self):
    method test_jit_check_input (line 216) | def test_jit_check_input(self):
    method test_jit_partial_shape (line 227) | def test_jit_partial_shape(self):
    method test_instance_method (line 256) | def test_instance_method(self):

FILE: tests/test_video/test_optflow.py
  function test_flowread (line 14) | def test_flowread():
  function test_flowwrite (line 51) | def test_flowwrite():
  function test_quantize_flow (line 77) | def test_quantize_flow():
  function test_dequantize_flow (line 104) | def test_dequantize_flow():
  function test_flow2rgb (line 128) | def test_flow2rgb():
  function test_flow_warp (line 144) | def test_flow_warp():
  function test_make_color_wheel (line 173) | def test_make_color_wheel():
  function test_flow_from_bytes (line 251) | def test_flow_from_bytes():
  function test_sparse_flow_from_bytes (line 267) | def test_sparse_flow_from_bytes():

FILE: tests/test_video/test_processing.py
  class TestVideoEditor (line 12) | class TestVideoEditor:
    method setup_class (line 15) | def setup_class(cls):
    method test_cut_concat_video (line 20) | def test_cut_concat_video(self):
    method test_resize_video (line 39) | def test_resize_video(self):

FILE: tests/test_video/test_reader.py
  class TestCache (line 13) | class TestCache:
    method test_init (line 15) | def test_init(self):
    method test_put (line 22) | def test_put(self):
    method test_get (line 34) | def test_get(self):
  class TestVideoReader (line 42) | class TestVideoReader:
    method setup_class (line 45) | def setup_class(cls):
    method test_load (line 50) | def test_load(self):
    method test_read (line 72) | def test_read(self):
    method test_slice (line 93) | def test_slice(self):
    method test_current_frame (line 120) | def test_current_frame(self):
    method test_position (line 127) | def test_position(self):
    method test_iterator (line 136) | def test_iterator(self):
    method test_with (line 143) | def test_with(self):
    method test_cvt2frames (line 148) | def test_cvt2frames(self):
    method test_frames2video (line 180) | def test_frames2video(self):

FILE: tests/test_visualization.py
  function test_color (line 8) | def test_color():