/jVddlmZddlZddlZddlmZddlmZGddeZ dS)) annotationsN)Base)expectc&eZdZeddZdS) ShrinkTestreturnNonec tjddgdgdd}tj|gdtjdtjjdggtjdtjjdgg}tj|dtjd d g }tj gd tj }tj gdtj }t||g|gddS)NShrinkxyg?)lambdbias)nodesnameinputsoutputsz backend-test ) producer_name opset_imports)ggg?g@)dtype)grrrg? test_shrink)rrr) onnxhelper make_node make_graphmake_tensor_value_info TensorProtoFLOATmake_model_gen_version make_opsetidnparrayfloat32r)nodegraphmodelr r s h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/model/shrink.pyexportzShrinkTest.exports>{$$  E E %   &&& 2238H8NQRPSTT 2238H8NQRPSTT '    22 (;33B;;<3   H000 C C C H///rz B B BuaS1#MBBBBBBN)rr )__name__ __module__ __qualname__ staticmethodr,r-r+rr s8CCC\CCCr-r) __future__rnumpyr%ronnx.backend.test.case.baseronnx.backend.test.case.modelrrr2r-r+r7s#""""" ,,,,,,//////CCCCCCCCCCr-