Name	Size	Mode	Actions
__pycache__/	-	0755	rm
algebra.py	1296	0644	editdlrm
arg_scope.py	1114	0644	editdlrm
array_helpers.py	648	0644	editdlrm
control_ops.py	625	0644	editdlrm
conv.py	10396	0644	editdlrm
db_input.py	381	0644	editdlrm
dropout.py	412	0644	editdlrm
elementwise_linear.py	1385	0644	editdlrm
fc.py	6405	0644	editdlrm
nonlinearity.py	1145	0644	editdlrm
normalization.py	10891	0644	editdlrm
pooling.py	924	0644	editdlrm
quantization.py	261	0644	editdlrm
tools.py	1089	0644	editdlrm
train.py	2192	0644	editdlrm
__init__.py	0	0644	editdlrm

import contextlib import copy import threading _threadlocal_scope = threading.local() @contextlib.contextmanager def arg_scope(single_helper_or_list, **kwargs): global _threadlocal_scope if not isinstance(single_helper_or_list, list): assert callable(single_helper_or_list), \ "arg_scope is only supporting single or a list of helper functions." single_helper_or_list = [single_helper_or_list] old_scope = copy.deepcopy(get_current_scope()) for helper in single_helper_or_list: assert callable(helper), \ "arg_scope is only supporting a list of callable helper functions." helper_key = helper.__name__ if helper_key not in old_scope: _threadlocal_scope.current_scope[helper_key] = {} _threadlocal_scope.current_scope[helper_key].update(kwargs) yield _threadlocal_scope.current_scope = old_scope def get_current_scope(): global _threadlocal_scope if not hasattr(_threadlocal_scope, "current_scope"): _threadlocal_scope.current_scope = {} return _threadlocal_scope.current_scope