U ‰dïã@sDddlmZddlmZmZGdd„deejƒZGdd„deƒZdS)é)Úcontext)ÚModifierContextÚUseModifierBasec@s eZdZdZdd„Zdd„ZdS)ÚNormalizerContextzK provide context to allow param_info to have different normalizers cCs | |¡S©N)Z _has_modifier©ÚselfÚname©r úD/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/normalizer_context.pyÚhas_normalizersz NormalizerContext.has_normalizercCs"| |¡std |¡ƒ‚| |¡S)Nz{} normalizer is not provided!)r ÚAssertionErrorÚformatZ _get_modifierrr r r Úget_normalizers ÿz NormalizerContext.get_normalizerN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rr r r r r src@seZdZdZdd„ZdS)Ú UseNormalizera context class to allow setting the current context. Example usage with layer: normalizers = {'norm1': norm1, 'norm2': norm2} with UseNormalizer(normalizers): norm = NormalizerContext.current().get_normalizer('norm1') layer(norm=norm) cCstSr)r)rr r r Ú_context_class$szUseNormalizer._context_classN)rrrrrr r r r rsrN)Z caffe2.pythonrZcaffe2.python.modifier_contextrrZDefaultManagedrrr r r r Ús