NgJddlZddlZddlmZddlmZGddeZdS)N)Base)expectc&eZdZeddZdS) ShrinkTestreturnNc 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 _/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/model/shrink.pyexportzShrinkTest.export s>{$$  E E %   &&& 2238H8NQRPSTT 2238H8NQRPSTT '    22 (;33B;;<3   H000 C C C H///rz B B BuaS1#MBBBBBB)rN)__name__ __module__ __qualname__ staticmethodr*r+r)rr s8CCC\CCCr+r)numpyr#ronnx.backend.test.case.baseronnx.backend.test.case.modelrrr0r+r)r4s~  ,,,,,,//////CCCCCCCCCCr+