File size: 493 Bytes
186701e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from enum import Enum
import torch
import torch.nn.functional as F
class MMYOLOBackend(Enum):
AX620A = 'ax620a'
COREML = 'coreml'
HORIZONX3 = 'horizonx3'
NCNN = 'ncnn'
ONNXRUNTIME = 'onnxruntime'
OPENVINO = 'openvino'
PPLNN = 'pplnn'
RKNN = 'rknn'
TENSORRT8 = 'tensorrt8'
TENSORRT7 = 'tensorrt7'
TORCHSCRIPT = 'torchscript'
TVM = 'tvm'
def HSigmoid__forward(self, x: torch.Tensor) -> torch.Tensor:
return F.hardsigmoid(x, inplace=True)
|