Skip to content

unary

torch_to_nnef.op.aten.unary

generic_unary

generic_unary(aten_op_id, node, op_helper, **kwargs)

Map PyTorch generic operators to NNEF (direct map).

List is

'aten:relu', 'aten:sigmoid', 'aten:log', 'aten:exp', 'aten:sin', 'aten:cos', 'aten:tan', 'aten:asin', 'aten:acos', 'aten:atan', 'aten:sinh', 'aten:cosh', 'aten:tanh', 'aten:asinh', 'aten:acosh', 'aten:atanh', 'aten:sign', 'aten:neg', 'aten:floor', 'aten:ceil', 'aten:sqrt', 'aten:rsqrt', 'aten:log2', 'aten:rcp', 'aten:not', 'aten:eq', 'aten:ne', 'aten:add', 'aten:sub', 'aten:lt', 'aten:gt', 'aten:le', 'aten:ge', 'aten:and', 'aten:or', 'aten:and', 'aten:or', 'aten:_relu', 'aten:greater', 'aten:greater_equal', 'aten:less', 'aten:less_equal', 'aten:logical_not', 'aten:logical_and', 'aten:logical_or', 'aten:reciprocal', 'aten:minimum', 'aten:maximum'