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.LinearQuant
torch_npu.contrib.module.LinearWeightQuant
父主题:
Ascend Extension for PyTorch自定义API