U d@sddlZddlZddlZddlmZddlmZddlTddlTddl Tddl Tddl Tddl Tddl TddlTddlTddlTddlTddlTddlTddlTddlTGdddeZeejeeje<dS)N) basestring) ModelHelper)*c.@seZdZeeeeeee e e e e eeeeeeeeeeeeeeeeeee e!e"e#e$e%e&e'e(e)e*e+e,e-e.e/d-Z0ddZ1ddZ2ddZ3dd Z4d S) HelperWrapper)- arg_scopefc packed_fc fc_decomp fc_sparsefc_prunedropoutmax_pool average_poolmax_pool_with_indexlrnsoftmax instance_norm spatial_bn spatial_gnmoments_with_running_statsreluprelutanhconcat depth_concatsum reduce_sumsubarg_min transposeiteraccuracyconvconv_ndconv_transpose group_convgroup_conv_deprecated image_input video_inputadd_weight_decayelementwise_linear layer_normmat_mul batch_mat_mulcondloopdb_input%fused_8bit_rowwise_quantized_to_float&sparse_lengths_sum_4bit_rowwise_sparsecCs ||_dSN)wrapped)selfr4r66/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/brew.py__init__QszHelperWrapper.__init__cs0jkrtdfdd}|_|S)Nz"Helper function {} not registered.csidkrVt|dkr0t|dtr0|d}nd|krB|d}ntdt|jj}t |\}}}}|dkrfdd|Dt } | i |||S)NrrmodelzjThe first input of helper function should be model. Or you can provide it in kwargs as model=.csi|]}|kr||qSr6r6).0var_nameZ new_kwargsr6r7 kszDHelperWrapper.__getattr__..scope_wrapper..) len isinstancer RuntimeErrorcopydeepcopyr _registryinspect getargspecZget_current_scopeupdateget)argskwargsr9funcZ var_names_varkwZ cur_scope helper_namer5r<r7 scope_wrapper[s(      z0HelperWrapper.__getattr__..scope_wrapper)rCAttributeErrorformat__name__)r5rNrOr6rMr7 __getattr__Ts zHelperWrapper.__getattr__cCs,|j}||jkrtd|||j|<dS)Nz9Helper {} already exists. Please change your helper name.)rRrCrPrQ)r5helpernamer6r6r7Registerxs zHelperWrapper.RegistercCst|tr|n|j}||jkSr3)r?rrRrC)r5Zhelper_or_helper_namerNr6r6r7 has_helpers zHelperWrapper.has_helperN)5rR __module__ __qualname__rrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2rCr8rSrVrWr6r6r6r7r sd0$ r)sysrArDZ past.builtinsrZcaffe2.python.model_helperrZcaffe2.python.helpers.algebraZcaffe2.python.helpers.arg_scopeZ#caffe2.python.helpers.array_helpersZ!caffe2.python.helpers.control_opsZcaffe2.python.helpers.convZcaffe2.python.helpers.db_inputZcaffe2.python.helpers.dropoutZ(caffe2.python.helpers.elementwise_linearZcaffe2.python.helpers.fcZ"caffe2.python.helpers.nonlinearityZ#caffe2.python.helpers.normalizationZcaffe2.python.helpers.poolingZ"caffe2.python.helpers.quantizationZcaffe2.python.helpers.toolsZcaffe2.python.helpers.trainobjectrmodulesrRr6r6r6r7s*  k