vortex_torch.utils

Classes

ElementwiseBinaryOpType(*values)
ElementwiseOpType(*values)
Mode(*values)
ReduceType(*values)

class vortex_torch.utils.Mode(*values)

Bases: Enum

profile = 0

execute = 1

class vortex_torch.utils.ReduceType(*values)

Bases: Enum

Mean = 0

Max = 1

Min = 2

L2Norm = 3

Sum = 4

class vortex_torch.utils.ElementwiseBinaryOpType(*values)

Bases: Enum

Maximum = 0

Minimum = 1

Add = 2

Mul = 3

class vortex_torch.utils.ElementwiseOpType(*values)

Bases: Enum

Relu = 0

Sigmoid = 1

Silu = 2

Abs = 3

Add_Mul = 4