o
    wi>                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s   e Zd ZedddZdS )
SingleRelureturnNonec               
   C  s   t jjddgdgdd} t jj| gdt jdt jjddggt jdt jjddggd	}t jj|d
t jddgd}t	j
ddt	j}t	|d}t||g|gdd d S )NReluxytest)namer         )nodesr   inputsoutputszbackend-test 	   )producer_nameopset_importsr   test_single_relu_model)r   r   r   )onnxhelper	make_node
make_graphmake_tensor_value_infoTensorProtoFLOATmake_model_gen_versionmake_opsetidnprandomrandnastypefloat32maximumr   )nodegraphmodelr	   r
    r)   e/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/model/single-relu.pyexport   s"   
zSingleRelu.exportN)r   r   )__name__
__module____qualname__staticmethodr+   r)   r)   r)   r*   r      s    r   )

__future__r   numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r)   r)   r)   r*   <module>   s   