torch_npu.contrib
- 概述
- （beta）torch_npu.contrib.npu_fused_attention_with_layernorm
- （beta）torch_npu.contrib.npu_fused_attention
- （beta）torch_npu.contrib.Prefetcher
- （beta）torch_npu.contrib.DCNv2
- （beta）torch_npu.contrib.BiLSTM
- （beta）torch_npu.contrib.Swish
- （beta）torch_npu.contrib.NpuFairseqDropout
- （beta）torch_npu.contrib.npu_giou
- （beta）torch_npu.contrib.npu_ptiou
- （beta）torch_npu.contrib.npu_iou
- （beta）torch_npu.contrib.function.fuse_add_softmax_dropout
- （beta）torch_npu.contrib.function.npu_diou
- （beta）torch_npu.contrib.function.npu_ciou
- （beta）torch_npu.contrib.module.NpuCachedDropout
- （beta）torch_npu.contrib.module.MultiheadAttention
- （beta）torch_npu.contrib.function.npu_single_level_responsible_flags
- （beta）torch_npu.contrib.function.npu_bbox_coder_encode_xyxy2xywh
- （beta）torch_npu.contrib.function.npu_fast_condition_index_put
- （beta）torch_npu.contrib.function.matmul_transpose
- （beta）torch_npu.contrib.function.npu_multiclass_nms
- （beta）torch_npu.contrib.function.npu_batched_multiclass_nms
- （beta）torch_npu.contrib.function.roll
- （beta）torch_npu.contrib.module.Mish
- （beta）torch_npu.contrib.module.SiLU
- （beta）torch_npu.contrib.module.ChannelShuffle
- （beta）torch_npu.contrib.module.LabelSmoothingCrossEntropy
- （beta）torch_npu.contrib.module.ModulatedDeformConv
- （beta）torch_npu.contrib.module.NpuDropPath
- （beta）torch_npu.contrib.module.Focus
- （beta）torch_npu.contrib.module.PSROIPool
- （beta）torch_npu.contrib.module.ROIAlign
- （beta）torch_npu.contrib.module.FusedColorJitter
- （beta）torch_npu.contrib.function.npu_bbox_coder_decode_xywh2xyxy
- （beta）torch_npu.contrib.function.npu_bbox_coder_encode_yolo
- （beta）torch_npu.contrib.module.npu_modules.DropoutWithByteMask
- （beta）torch_npu.contrib.function.dropout_with_byte_mask
- torch_npu.contrib.module.LinearA8W8Quant
- torch_npu.contrib.module.LinearWeightQuant