U
    ‰dq  ã                   @   s(   d dl mZmZ ddd„Zddd„ZdS )	é    )Ú	add_if_opÚadd_while_opNc                 C   s"   t | j|||j|r|jndƒ dS )Ú	ConditionN)r   Únet)ÚmodelÚ	cond_blobÚexternal_blobsZ
then_modelZ
else_model© r	   úE/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/control_ops.pyÚcond   s    ûr   c                 C   s"   t | j|||j|r|jndƒ dS )ZLoopN)r   r   )r   r   r   Z
loop_modelZ
cond_modelr	   r	   r
   Úloop   s    ûr   )N)N)Zcaffe2.python.control_ops_utilr   r   r   r   r	   r	   r	   r
   Ú<module>   s   

