U ‰dœã@sddd„ZdS)FcKsH|rd|d<nd|d<d|ks&tdƒ‚|jj||d|dgf|ŽdS) ÚdropoutZCUDNNZengineÚDEFAULTZis_testzArgument 'is_test' is requiredÚ_Z_maské)ÚAssertionErrorÚnetZDropout)ÚmodelZblob_inZblob_outZ use_cudnnÚkwargs©r úA/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/dropout.pyr s ÿÿÿrN)F)rr r r r Ú ó