math

torch_to_nnef.op.aten.math

atan2

atan2(node, op_helper, **kwargs)

aten::atan2.

bitwise_and

bitwise_and(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:bitwise_and', 'aten:bitwise_cpu' to NNEF.

bitwise_not

bitwise_not(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:bitwise_not', 'aten:bitwise_not_cpu' to NNEF.

bitwise_or

bitwise_or(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:bitwise_or' to NNEF.

bitwise_xor

bitwise_xor(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:bitwise_xor' to NNEF.

cumsum

cumsum(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:cumsum' to NNEF using a scan fragment (Tract).

• Implemented via tract_core_scan inside fragment cumsum (axis=0).
• For arbitrary dim, transpose input to bring that axis to 0, apply fragment, then transpose back.

div

div(node, op_helper, inference_target, torch_graph, **kwargs)

Map PyTorch: 'aten:div' to NNEF.

expm1

expm1(node, op_helper, **kwargs)

aten::exp1m.

floor_divide

floor_divide(node, op_helper, inference_target, torch_graph, **kwargs)

Map PyTorch: 'aten:floor_divide' to NNEF.

fmod

fmod(node, op_helper, **kwargs)

aten::fmod.

equivalent

a - a.div(b, rounding_mode="trunc") * b

log10

log10(node, op_helper, **kwargs)

Mul val may not be good enough.

log1p

log1p(node, op_helper, **kwargs)

aten::log1p.

logical_xor

logical_xor(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:logical_xor' to NNEF.

mul

mul(node, op_helper, torch_graph, **kwargs)

Map PyTorch: 'aten:mul' to NNEF.

pow_

pow_(node, op_helper, **kwargs)

Map PyTorch: 'aten:pow' to NNEF.

remainder

remainder(node, op_helper, torch_graph, inference_target, **kwargs)

Map PyTorch: 'aten:remainder' to NNEF.

round_

round_(inference_target, **kwargs)

Map PyTorch: 'aten:round' to NNEF.

rsub

rsub(node, op_helper, torch_graph, **kwargs)

Map PyTorch: 'aten:rsub' to NNEF.

trunc

trunc(node, op_helper, **kwargs)

Map PyTorch: 'aten:trunc' to NNEF.

var

var(node, op_helper, **kwargs)

aten::var.