o wiA@sHddlmZddlZddlZddlmZddlmZGdddeZ dS)) annotationsN)Base)expectc@seZdZedddZdS)SumreturnNonecCstgdtj}tgdtj}tgdtj}tgdtj}tjjdgddgd}t||||g|gd d tjjdd gdgd}t||g|gd d t||}tjjdd d gdgd}t|||g|gdd dS)N)r)r)r r )r r)data_0data_1data_2result)inputsoutputstest_sum_example)rrnamertest_sum_one_inputrtest_sum_two_inputs) nparrayastypefloat32onnxhelper make_noderadd)rrrrnoder"\/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/sum.pyexports<  z Sum.exportN)rr)__name__ __module__ __qualname__ staticmethodr$r"r"r"r#r sr) __future__rnumpyrronnx.backend.test.case.baseronnx.backend.test.case.noderrr"r"r"r#s