Adding support for Asin (#122)

Author: dstarkenburg

src/load.jl

@@ -99,6 +99,9 @@ function load_node!(tape::Tape, ::OpConfig{:ONNX, :ConstantOfShape}, args::VarVe
     return push_call!(tape, makeshape, args...; attrs...)
 end
 
+function load_node!(tape::Tape, ::OpConfig{:ONNX, :Asin}, args::VarVec, attrs::AttrDict)
+    return push_call!(tape, _asin, args[1])
+end
 
 function load_node!(tape::Tape, nd::NodeProto, backend::Symbol)
     args = [tape.c.name2var[name] for name in nd.input]

src/ops.jl

@@ -88,6 +88,7 @@ _cos(x) = cos.(x)
 _abs(x) = abs.(x)
 _acos(x) = acos.(x)
 _acosh(x) = acosh.(x)
+_asin(x) = asin.(x)
 mul(xs...) = .*(xs...)
 relu(x) = NNlib.relu.(x)
 leakyrelu(x;a = 0.01) = NNlib.leakyrelu.(x,a)

src/save.jl

@@ -176,6 +176,11 @@ function save_node!(g::GraphProto, ::OpConfig{:ONNX, typeof(makeshape)}, op::Uml
     push!(g.node, nd)
 end
 
+function save_node!(g::GraphProto, ::OpConfig{:ONNX, typeof(_asin)}, op::Umlaut.Call)
+    nd = NodeProto("Asin", op)
+    push!(g.node, nd)
+end
+
 function save_node!(g::GraphProto, ::OpConfig{:ONNX, typeof(*)}, op::Umlaut.Call)
     nd = NodeProto(
         input=[onnx_name(v) for v in reverse(op.args)],

test/saveload.jl

@@ -60,6 +60,14 @@ import ONNX: NodeProto, ValueInfoProto, AttributeProto, onnx_name
         ort_test(ONNX._acosh, A)
     end
 
+    @testset "Asin" begin
+        # ONNXRunTime has no implementation for Asin(x::Float64), using Float32
+        A = rand(Float32, 3, 4)
+        # Asin defined for |A| <= 1
+        A = A .% 1
+        ort_test(ONNX._asin, A)
+    end
+
     @testset "Transpose" begin
         # ort_test() checks for expected output of functions, errors on Transpose 
         # because of array shape; manually testing!