o wiW@sHddlmZddlZddlZddlmZddlmZGdddeZ dS)) annotationsN)Base)expectc@seZdZedddZdS)MeanreturnNonecCstgdtj}tgdtj}tgdtj}tgdtj}tjjdgddgd}t||||g|gd d tjjdd gdgd}t||g|gd d tt ||d }tjjdd dgdgd}t|||g|gdd dS)N)r)r)r r )r rr r)data_0data_1data_2result)inputsoutputstest_mean_example)rrnamer test_mean_one_inputg@rtest_mean_two_inputs) nparrayastypefloat32onnxhelper make_noderdivideadd)r rrrnoder!]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/mean.pyexports< z Mean.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