
    %Vji                    N   d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm	Z	 ddl
mZmZmZmZ ddlmZ ddlZddlmZmZmZmZmZmZmZmZmZ g dZ ed	          Z ed
          Z	 	 d0deeef         dededeeef         fdZ ej!        e de"e         fd                        Z#ej!        de"e         fd            Z$ej!        e de%eef         fd                        Z&defdZ'	 d1dee         deeege(f                  de)e         fdZ*dedee         defdZ+ eed          Z, eed          Z- eed          Z.ej!        de/e%ee)e         f         e%eef         f         fd            Z0e de%ee)e         f         fd             Z1e d!             Z2ej!        de"e         fd"            Z3e dede4fd#            Z5d$ Z6 G d% d&          Z7d' Z8d( Z9d) Z:d* Z;ej<        d+             Z= G d, d-e7          Z>ej<        d.             Z?ej<        d/             Z@dS )2aE  
Python implementation of ``__torch_function__``

While most of the torch API and handling for ``__torch_function__`` happens
at the C++ level, some of the torch API is written in Python so we need
python-level handling for ``__torch_function__`` overrides as well. The main
developer-facing functionality in this file are handle_torch_function and
has_torch_function. See torch/functional.py and test/test_overrides.py
for usage examples.

Note
----
heavily inspired by NumPy's ``__array_function__`` (see:
https://github.com/pytorch/pytorch/issues/24015 and
https://www.numpy.org/neps/nep-0018-array-function-protocol.html
)

If changing this file in a way that can affect ``__torch_function__`` overhead,
please report the benchmarks in ``benchmarks/overrides_benchmark``. See the
instructions in the ``README.md`` in that directory.
    N)Iterable)wraps)AnyCallableOptionalTypeVar)	ParamSpec)	_add_docstr_get_function_stack_at_has_torch_function_has_torch_function_unary_has_torch_function_variadic_is_torch_function_mode_enabled_len_torch_function_stack_pop_torch_function_stack_push_on_torch_function_stack)
get_ignored_functionsget_overridable_functionsget_testing_overrideshandle_torch_functionhas_torch_functionresolve_nameis_tensor_likeis_tensor_method_or_propertywrap_torch_functionenable_reentrant_dispatch_P_R.*is deprecated, please use.*torchfuncregexmodulereturnc                      t                     dt          j        dt          j        dt          f fd            }|S )a  
    Decorator that temporarily disables ``UserWarning``s for the given ``module`` if the warning message matches the
    given ``regex`` pattern.

    Arguments
    ---------
    func : function
        Function to disable the warnings for.
    regex : str
        A regex pattern compilable by ``re.compile``. This is used to match the ``UserWarning`` message.
    module : str
        The python module to which the filtering should be restricted.

    Returns
    -------
    function
        The wrapped function.
    argskwargsr$   c                      t          j                    5  t          j        dt                      | i |cd d d            S # 1 swxY w Y   d S )Nignore)categorymessager#   )warningscatch_warningsfilterwarningsUserWarning)r&   r'   r!   r#   r"   s     I/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperZ   s    $&& 	) 	)#;f    4(((		) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)s   %AAA)r   r   r&   r'   r   )r!   r"   r#   r1   s   ``` r0   _disable_user_warningsr2   B   s_    0 4[[)rw )") ) ) ) ) ) ) ) ) [) N    c                     t           j        } h t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j	        t           j
        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j         t           j!        t           j"        t           j#        t           j$        t           j%        t           j&        t           j'        t           j(        t           j)        t           j*        t           j+        t           j,        t           j-        t           j.        t           j/        t           j0        t           j1        t           j2        t           j3        t           j4        t           j5        t           j6        t           j7        t           j8        t           j9        t           j:        t           j;        t           j<        t           j=        t           j>        t           j?        t           j@        t           jA        t           jB        t           jC        t           jD        t           jE        t           jF        t           jG        t           jH        t           jI        t           jJ        t           jK        t           jL        t           jM        t           jN        jN        t           jN        j        t           jN        jO        t           jN        j        t           jP        t           jQ        jR        t           jQ        jS        t           jT        t           jU        t           jV        t           jW        t           jX        t           jY        t           jZ        t           j[        t           j\        t           j]        t           j^        t           j_        t           j`        t           ja        t           jb        t           jc        t           jd        t           je        t           jf        t           jg        t           jh        t           ji        t           jj        t           jk        t           jl        t           jm        t           jn        t           jo        t           jp        t           jq        t           jr        t           js        t           jt        t           ju        t           jv        t           jw        t           jx        t           jy        jz        t           j{        j|        j}        t           j{        j|        j~        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j|        j        t           j{        j        j        t           j{        j        j        t           j{        j        jb        t           j{        j        j        t           j{        j        jP        t           j{        j        j        t           j{        j        j        t           j{        j        j        t           j{        j        j        t           j{        j        j        t           j{        j        j        t           j{        j        j        t           j        j        t          t          t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j{        j|        j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        j        | j        j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        S )a%  
    Return public functions that cannot be overridden by ``__torch_function__``.

    Returns
    -------
    set[Callable]
        A tuple of functions that are publicly available in the torch API but cannot
        be overridden with ``__torch_function__``. Mostly this is because none of the
        arguments of these functions are tensors or tensor-likes.

    Examples
    --------
    >>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions()
    True
    >>> torch.add in torch.overrides.get_ignored_functions()
    False
    (   r    Tensortypename	is_tensor
is_storageset_default_tensor_typeset_default_deviceget_default_deviceset_rng_stateget_rng_statemanual_seedinitial_seedseedsaveloadset_printoptionsforkget_default_dtypeget_num_interop_threadsget_num_threadsinit_num_threadsimport_ir_moduleimport_ir_module_from_bufferis_anomaly_enabledis_anomaly_check_nan_enabledis_grad_enabledmerge_type_from_type_commentparse_irparse_schemaparse_type_commentset_anomaly_enabledset_flush_denormalset_num_interop_threadsset_num_threadswait	as_tensor
from_numpytensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutalign_tensorsarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptableemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrandrandnrandintrandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   sigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvitals_enabled	set_vitalread_vitalsvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_make_dtensor_make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_countr5   s    r0   r   r   e   s   ( \FAAA 	A 	%	A
 	 A 	 A 	A 	A 	A 	A 	
A 	
A 	
A 	A 	
A  	!A" 	%#A$ 	%A& 	'A( 	)A* 	*+A, 	 -A. 	*/A0 	1A2 	*3A4 	5A6 	7A8 	 9A: 	!;A< 	 =A> 	%?A@ 	AAB 	
CAD 	EAF 	GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	_A` 	aAb 	cAd 	eAf 	gAh 	iAj 	kAl 	mAn 	'oAp 	qAr 	sAt 	uAv 	wAx 	yAz 	{A| 	}A~ 	A@ 	AAB 	)CAD 	EAF 	GAH 	)IAJ 	$KAL 	(MAN 	 OAP 	!QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	'_A` 	aAb 		cAd 		eAf 		gAh 	iAj 	
kAl 	
mAn 	oAp 	qAr 	sAt 	uAv 	wAx 	(yAz 	 {A| 	}A~ 	A@ 	,AAB 	CAD 	EAF 	
GAH 	IAJ 	
KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	&[A\ 	]A^ 	_A` 	aAb 	cAd 	!eAf 	*gAh 	iAj 	kAl 	mAn 	oAp 	qAr 	,sAt 	.uAv 	$wAx 	-yAz 	,{A| 	.}A~ 	4A@ 	7AAB 	1CAD 	#EAF 	'GAH 	 IAJ 	+KAL 	*MAP 	$QAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	$_A` 	#aAb 	%cAd 	$eAf 	 gAh 	iAj 	%kAl 	mAn 	oAp 	"qAr 	!sAt 	 uAv 	 wAx 	"yAz 	&{A| 	%}A~ 	&A@ 	%AAB 	&CAD 	%EAF 	$GAH 	$IAJ 	$KAL 	$MAN 	$OAP 	$QAR 	$SAT 	$UAV 	(WAX 	(YAZ 	'[A\ 	(]A^ 	%_A` 	!aAb 	2cAd 	*eAf 	;gAh 	*iAj 	kAl 	*mAn 	*oAp 	*qAr 	sAt 	$uAv 	wAx 	yAz 	{A| 	}A~ 	
A@ 	
AAB 	CAD 	EAF 	-GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	 UAV 	WAX 	YAZ 	![A\ 	!]A^ 	_A` 	aAb 	cAd 	eAf 	gAh 	
iAj 	kAl 	mAn 	
oAp 	qAr 	sAt 	 uAv 	wAx 	yAz 	{A| 	}A~ 	A@ 	AAB 	CAD 	EAF 	GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	"]A^ 	_A` 	aAb 	 cAd 	$eAf 	gAh 	%iAj 	'kAl 	*2mAn 	oAp 	qAr 	sAt 	uAv 	wAx 	yAz 	{A| 	 }A~ 	A@ 	AA Ar3   c                  `    t           j        } | j        j        | j        j        | j        j        hS )a  
    Return public functions that do not wrap in a subclass when invoked by
    the default ``Tensor.__torch_function__`` that preserves subclasses.  Typically,
    these functions represent field accesses (i.e., retrieving a Tensor that
    is stored somewhere on the Tensor) as opposed to computation.  Users of
    these functions expect object identity to be preserved over multiple accesses
    (e.g., ``a.grad is a.grad``) which cannot be upheld if we're wrapping on
    the fly every time (furthermore, the tensor stored here might already be
    the subclass, in which case wrapping really ought not to happen).

    Not ALL property accessors have this property; for example ``Tensor.T`` actually
    just creates a new transposed tensor on the fly, and so we SHOULD interpose on
    these calls (you need to check the implementation of the function to see if
    this is the case or not).  Additionally, if a property accessor doesn't return a Tensor,
    it doesn't have to be on this list (though it is harmless if it is).
    )r    r5   _baser'  grad_gradr2  s    r0   get_default_nowrap_functionsr7  ~  s-    $ \F r3   c                  N   t           j        } i t           j        ddt           j        ddt           j        d t           j        d t           j        ddt           j        d t           j        ddt           j	        dd	t           j
        dd
t           j        ddt           j        d	dt           j        d
dt           j        d
dt           j        d	dt           j        d	dt           j        d	dt           j        d i t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d  t           j        dd!t           j        dd"t           j        d# t           j         dd$t           j!        dd%t           j"        dd&t           j#        dd'i t           j$        dd(t           j%        dd)t           j&        dd*t           j'        dd+t           j(        dd,t           j)        d- t           j*        d. t           j+        d/ t           j,        dd2t           j-        d	d3t           j.        d4 t           j/        d5 t           j0        d6 t           j1        d7 t           j2        d8 t           j3        d9 t           j4        d: i t           j5        d; t           j6        dd<t           j7        d= t           j8        dd?t           j9        dd@t           j:        ddAt           j;        ddBt           j<        ddCt           j=        ddDt           j>        ddEt           j?        ddFt           j@        ddGt           jA        dH t           jB        ddIt           jC        dJ t           jD        dK t           jE        ddLi t           jF        dM t           jG        ddNt           jH        ddOt           jI        ddPt           jJ        ddSt           jK        ddTt           jL        ddVt           jM        ddWdXt           jN        dY t           jO        ddZt           jP        jO        dd[t           jP        jQ        dd\t           jR        dd]t           jS        dd^t           jT        d_ t           jU        d5d`t           jV        ddai t           jW        ddbt           jX        ddct           jY        dddt           jZ        ddet           j[        ddft           j\        dg t           j]        ddit           j^        dj t           j_        ddkt           j`        dl t           jP        ja        ddmt           jb        ddnt           jc        ddot           jd        ddpt           je        ddqt           jf        d5drt           jg        ddsi t           jh        ddtt           ji        ddut           jj        dv t           jk        d5dwt           jl        ddxt           jm        ddyt           jn        ddzt           jo        d{ t           jp        dd|t           jq        dd}t           jr        dd~t           js        ddt           jt        d t           ju        ddt           jP        ju        ddt           jv        ddt           jw        ddi t           jx        ddt           jy        ddt           jz        ddt           j{        d dt           j|        ddt           j}        ddt           j~        d t           j        d t           jP        j        d t           j        d t           j        ddt           j        ddt           j        d5dt           j        d!dt           j        d"dt           jP        j        d#dt           j        d"di t           j        ddt           j        ddt           j        d$dt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d t           j        ddt           jP        j        ddt           jP        j        ddt           jP        j        d%dt           jP        j        d%dt           j        d t           j        d&di t           j        d'dt           j        d(dt           j        ddt           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d t           j        d)dt           j        d t           j        d t           j        d t           j        d i t           j        d t           j        d t           j        d t           j        d t           j        d t           j        j        d*dt           j        j        d*dt           j        j        d*dt           j        j        d*dÄt           j        j        d*dĄt           j        j        d+dƄt           j        j        d+dǄt           j        j        d*dȄt           j        j        d*dɄt           j        j        ddʄt           j        j        dd˄t           j        j        dd̄i t           j        j        dd̈́t           j        j        d+d΄t           j        j        d+dτt           j        j        d+dЄt           j        j        d+dфt           j        j        dd҄t           j        j        ddӄt           j        j        d*dԄt           j        ddՄt           j        d,dքt           j        dׄ t           j        d؄ t           j        dل t           j        ddڄt           j        ddۄt           j        d܄ t           j        dd݄i t           j        ddބt           j        dd߄t           j        ddt           j        ddt           j        ddfdt           j        d t           j        d-dt           j        d.dt           j        ddt           j        ddt           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d/di t           j        d t           j        d t           j        d t           j        d0dt           j        d t           j        ddt           j        ddt           j        ddt           j        d1dt           j        d2dt           j        ddt           j        d3dt           j        d4dt           j        d5dt           j        d6dt           jP        j        d  t           j        ddi t           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d	 t           j        dd
t           j        ddt           j        d t           j        d7dt           j        d t           j        d)dt           j        d t           j        d t           j        ddi t           j        ddt           j         d t           j        d t           j        ddt           jP        j        ddt           jP        j        ddt           j        d t           j        d t           j        d t           j        d t           j	        d t           j
        d t           j        d t           j        d  t           j        d! t           j        dd"t           j        d# i t           j        d8d$t           j        d9d%t           j        d& t           j        dd't           jP        j        dd(t           jP        j        dd)t           jP        j        dd*t           j        d0d+t           j        dd,t           j        dd-t           j        dd.t           j        dd/t           j        dd0t           j        dd1t           j        d:d2t           j        dd3t           j         dd4i t           j!        dd5t           j"        dd6t           j#        dd7t           j$        dd8t           j%        dd9t           j&        d: t           j'        dd;t           j(        dd<t           j)        dd=t           j*        dd>t           j+        dd?t           j,        dd@t           j-        ddAt           j.        dB t           j/        ddCt           j0        ddDt           j1        ddEi t           j2        d;dFt           j3        ddGt           j4        ddHt           j5        dI t           j6        dJ t           j7        ddKt           j8        ddLt           jP        j2        d<dMt           jP        j9        d<dNt           jP        j:        d;dOt           jP        j3        d;dPt           jP        j8        ddQt           j;        dR t           jP        j;        ddSt           jP        j<        d.dTt           jP        j=        ddUt           j>        dV i t           jP        j>        dW t           j?        ddXt           j@        ddYt           jA        ddZt           jB        d=d[t           jC        d=d\t           jD        d=d]t           jE        d>d^t           jF        dd_t           jG        d?d`t           jH        ddat           jI        ddbt           jJ        dc t           jK        dddt           jL        ddet           jM        ddft           jN        dg i t           jO        dh t           jP        di t           jQ        dj t           jR        dk t           jS        dl t           jT        dm t           jU        ddnt           jV        d@dot           jW        dp t           jX        dq t           jY        ddrt           jZ        ddst           j[        ddtt           j\        ddut           j]        ddvt           j^        dw t           j_        dx i t           j`        dAdzt           ja        d{ t           jb        d| t           jc        d} t           jd        dBd~t           je        dCdt           jf        d t           jg        dDdt           jh        d t           ji        ddt           jj        ddt           jk        ddt           jl        ddt           jm        ddt           jn        jo        jp        d t           jn        jo        jq        d t           jn        jo        j        ddi t           jn        jo        jr        ddt           jn        jo        js        ddt           jn        jo        jt        ddt           jn        jo        ju        ddt           jn        jo        jv        ddt           jn        jo        jw        ddt           jn        jo        j        dEdt           jn        jo        jx        dFdt           jn        jo        jy        dFdt           jn        jo        j.        dGdt           jn        jo        j7        ddt           jn        jo        jz        dHdt           jn        jo        j8        ddt           jn        jo        jL        ddt           jn        jo        jq        ddt           jn        jo        j{        dIdt           jn        jo        jv        ddi t           jn        jo        j        dJdt           jn        jo        j|        dJdt           jn        jo        j}        dJdt           jn        jo        j~        dJdt           jn        jo        j        ddt           jn        jo        j        d&dt           jn        jo        j        dKdt           jn        jo        j        dEdt           jn        jo        j        dLdt           jn        jo        j        dMdt           jn        jo        j        dMdt           jn        jo        j        dMdt           jn        jo        j        dMdt           jn        jo        j        dNdt           jn        jo        j        dOdt           jn        jo        j        dPdt           jn        jo        j        dQdi t           jn        jo        j        dBdt           jn        jo        j        dRdt           jn        jo        j        d1dt           jn        jo        j        dSdt           jn        jo        j        d3dt           jn        jo        j         dTdt           jn        jo        j        dUdt           jn        jo        j        d9dt           jn        jo        j        dVdt           jn        jo        j        dBdt           jn        jo        j        dWdt           jn        jo        j        ddt           jn        jo        j        dXdńt           jn        jo        j         dYdǄt           jn        jo        j        dȄ t           jn        jo        j        d.dɄt           jn        jo        j        d.dʄi t           jn        jo        j        d.d˄t           jn        jo        j4        dd̄t           jn        jo        jB        d>d̈́t           jn        jo        jE        d>d΄t           jn        jo        jC        d>dτt           jn        jo        j        d>dЄt           jn        jo        jD        d>dфt           jn        jo        j        d>d҄t           jn        jo        j        dZdӄt           jn        jo        j        dZdԄt           jn        jo        j        dZdՄt           jn        jo        j        dVdքt           jn        jo        j        d[dׄt           jn        jo        j        d\d؄t           jn        jo        j        d]dلt           jn        jo        j        dHdڄt           jn        jo        j        d^dۄi t           jn        jo        j        d_d݄t           jn        jo        j        dPdބt           jn        jo        j        d`dt           jn        jo        j        dadt           jn        jo        j        dbdt           jn        jo        j        d t           jn        jo        j        ddt           jn        jo        j        ddt           jn        jo        j        dcdt           jn        jo        j        dddt           jn        jo        j        ddt           jn        jo        j        ddt           jn        jo        j        ddt           jn        jo        j        dedt           jn        jo        j        dfdt           jn        jo        j        dgdt           jn        jo        j        d]di t           jn        jo        j        dYdt           jn        jo        j        dYdt           jn        jo        j        dhdt           jn        jo        j        d1dt           jn        jo        j        d t           jn        jo        j        d t           jn        jo        j        ddt           jn        jo        j        didt           jn        jo        j        ddUdd>ddt           jn        jo        j        dLdt           jn        j        j        djdt           jn        j        j        djdt           jn        j        j        d t           jn        j        j        dkdt           j        ddt           j        dddt           j        d i t           j        dldt           jP        j        dmd	t           jP        j        dnd
t           jP        j        	 dodt           j        dpdt           j        dldt           j        d t           j        d t           j        dqdt           j        dadt           j        d t           j        drdt           j        d$dt           j        dsdt           jP        j        dtdt           j        d t           j        d i t           j        ddt           j        d t           j        ddt           j        ddt           j        d t           j        d(dt           j        dd t           j        dd!t           j        dd"t           j        d# t           j        d$ t           j        d% t           j        d& t           j        d' t           j        d<d(t           jP        j        dud*t           j        dvd,i t           j        dvd-t           j        d. t           j        d/ t           j        d0 t           j        d1 t           j        d2 t           j        d3 t           j        	 	 dwd7t           j        	 	 dxd:t           j        	 	 dyd=t           j        d> t           j        d? t           j        dd@t           j        d(dAt           j        dt           j        ddfdBt           j        d(dCt           j        dD i t           j        ddEt           j        ddFt           jP        j        ddGt           j        dH t           j        dI t           j        ddJt           j        ddKt           j        ddLt           j        ddMt           j        ddNt           j        dO t           j        dcdPt           j        dQ t           j        ddRt           j        dS t           j        ddTt           j        ddUi t           j        dzdWt           j         ddXt           j        ddYt           j        dZ t           j        ddd[t           j        dd\t           j        d6d]t           j        d	d^t           j        d_ t           j        d` t           j        d7dat           j	        ddbt           j
        d{ddt           j        de t           j        df t           j        d|dgt           j        d|dhi t           j        ddit           j        ddjt           j        ddkt           j        ddlt           j        ddmt           j        ddnt           j        ddot           j        ddpt           j        dq t           jP        j        dr t           j        ddst           j        ddtt           j        ddut           jP        j        d<dvt           jP        j        d;dwt           j        d}dddxdyt           j        d5dzi t           j        d5d{t           j        dd|t           j        dd}t           j         dd~t           j!        d	dt           j"        ddt           j#        ddt           j$        ddt           j%        d~dt           j&        ddt           j'        ddt           j(        ddt           j)        d t           j*        d t           j+        d t           j,        d t           j-        d i t           j.        d t           j/        d t           j0        d t           j1        d t           j2        d t           j3        d t           j4        d t           j5        d t           j6        d t           j7        d t           j8        d t           j9        d t           j:        ddt           j;        ddt           j<        ddt           jP        j;        d<dt           jP        j=        ddi t           j>        d t           j?        d t           j@        jA        d t           j@        jB        d t           j@        jC        d t           j@        jD        d t           j@        jE        d t           j@        jF        ddt           j@        jG        ddt           j@        jH        ddt           j@        jI        ddt           j@        j        d t           j@        jJ        d t           j@        j        d t           j@        j        d t           j@        jK        d t           j@        j        d i t           j@        j        d t           j@        jL        d t           j@        j        d t           j@        jM        ddt           j@        jN        ddt           j@        jO        d t           j@        jP        ddt           j@        jQ        ddt           j@        j        d t           j@        jR        d t           j@        jS        d t           j@        jT        d t           j@        jU        ddt           j@        jV        ddt           j@        j"        d t           j@        jW        d t           j@        j         ddi t           j@        j,        d t           j@        j-        ddt           j@        jX        dÄ t           j@        jY        dĄ t           j@        jZ        dń t           j@        j[        dƄ t           j@        j\        dǄ t           j@        j]        dȄ t           j@        j^        dɄ t           j@        j        ddʄt           j@        j_        d˄ t           j@        j         d̄ t           j@        j`        d̈́ t           j@        ja        d΄ t           j@        jb        ddτt           j@        jc        ddЄt           j@        jd        ddфi t           j@        je        dd҄t           j@        j        dӄ t           j@        j        ddԄt           j@        jf        dՄ t           j@        jg        ddքt           j@        j'        ddׄt           j@        jh        dd؄t           ji        dل t           jj        dڄ t           jk        ddۄt           jl        dd܄t           jm        dd݄t           jP        jn        d$dބt           jP        jo        dd߄t           jp        ddt           jq        d5dt           j        ddi t           jr        d t           js        d@dt           jt        d t           ju        d t           jv        d dt           jw        d dt           jx        ddt           jP        jy        dqdt           jz        ddt           j        didt           j{        ddt           j|        d t           j}        ddt           j~        d5dt           j        d t           j        ddt           j        ddi t           j        d t           j        d5dt           j        d5dt           j        d5dt           j        ddt           jP        j        ddt           j        ddt           j        ddt           j        d t           j        ddt           j        ddt           j        d t           j        d  t           j        d(dt           j        d t           j        d t           j        d i t           j        d t           j        d t           j        d t           j        ddt           j        d	 t           j        d"d
t           j        dddt           j        d t           j        d t           j        d t           j        d t           j        d t           j        d|dt           j        d5dt           j        d5dt           j        d t           j        d i t           j        d t           j        d t           j        d t           j        d t           j        d t           j        d t           j        d t           j        d t           j        d t           j        d  t           j        d5d!t           j        d" t           j        d# t           j        d$ | j        d% | j        d& | j        d' i | j        d( | j        d) | j        d* | j        d+ | j        d, | j        d- | j        d. | j        d/ | j        d0 | j        d1 | j        d2 | j        d3 | j        d4 | j        d5 | j        d6 | j        d7 | j        d8 i | j        d9 | j        d: | j        d; | j        d< | j        d= | j        d> | j        d? | j        d@ | j        dA | j        dB | j        dC | j        dD | j        dE | j        dF | j        dG | j        ddHdI| j        dJ i | j        dK | j        j        dL | j        j        dM | j        j        dN | j        j        dO | j        j        dP | j        j        dQ | j        j        dR | j        j        dS | j        j        dT | j        j        dU | j        j        dV | j        j        dW | j        j        dX | j        dY | j        dZ | j        d[ i | j        j        d\ | j        j        d] | j        j        d^ | j        j        d_ | j        j        d` | j        j        da | j        j        db | j        j        dc | j        j        dd | j        j        de | j        j        df | j        j        dg | j        j        dh | j        j        di | j        j        dj | j        j        dk | j        j        dl i | j        j        dm | j        j        dn | j        j        do | j        j        dp | j        j        dq | j        j        dr | j         j        ds | j        j        dt | j        j        du | j        j        dv | j        j        dw | j        j        dx | j        j        dy | j        j        dz | j        j        d{ | j        j        d| | j        d.d}i | j	        d~ | j
        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d i | j        dMd| j        t           j        fd| j        t           j        fd| j        t           j        fd| j        t           j        fd| j        dVddd| j         d | j!        d | j"        t           j#        fd| j$        dd| j%        t           j        fd| j&        t           j        fd| j'        t           j        fd| j(        t           j        fd| j)        t           j        fd| j*        d | j+        d i | j        d"d| j,        d | j-        dd| j.        t           j        fd| j/        t           j        fd| j0        d | j1        d | j2        d | j3        d6ddd| j4        d | j5        d | j6        t           j        fd| j7        t           j        fd| j8        ddd| j        d | j9        t           j        fd| j:        t           j        fdi | j;        d | j<        d | j=        t           j        fd| j>        d | j?        d | j	        d | j@        d | jA        d | jB        d | jC        d | jD        dddd| j         d | jE        t           j        fd| jF        d | jG        d | jU        dd| jH        ddi | j        dÄ | jI        dĄ | jJ        dń | jK        dƄ | jL        dǄ | jM        dȄ | j        dɄ | jN        dʄ | j        d˄ | jO        d̄ | jP        dd̈́| jQ        d΄ | jR        ddddτ| jS        dЄ | jT        dф | jU        d҄ | jV        dӄ i | jW        dԄ | jX        dՄ | jY        d7dք| jZ        dׄ | j[        d؄ | j\        dل | j]        dڄ | j^        dۄ | j_        d܄ | j`        dd݄| j        dބ | ja        d߄ | jb        t           j        fd| jc        d | j        d|d| jd        d | je        d i | jf        dd| jg        d | jh        d | j!        d	d| ji        d | jj        d | jk        d | jl        d | jm        d | jr        d | jn        ddt           j        fd| jo        dddd| jp        dd| jq        d | jr        d | js        d | jt        d | j        d | j        dVd| ju        d | jv        d | jw        d | jx        d | jy        dd| jz        d t           jP        j{        ddi	}t           j|        j}        j~        }t          | |          r8d.d |t          | |          <   d |t          | d|           j        <   i }t                      }|                                D ]\  }}|j        |j        dz   d|j        z   dz   d|j        z   dz   d|j        z   dz   g}|j                            d          rP|j        t          d          d         }|                    d|z   dz   d|z   dz   d|z   dz   g           |D ]1}	t          | |	d          }
t          |
          r|
|vr	|
|vr|||
<   2|                    |           |S (  ar  Return a dict containing dummy overrides for all overridable functions

    Returns
    -------
    Dict[Callable, Callable]
        A dictionary that maps overridable functions in the PyTorch API to
        lambda functions that have the same signature as the real function
        and unconditionally return -1. These lambda functions are useful
        for testing API coverage for a type that defines ``__torch_function__``.

    Examples
    --------
    >>> import inspect
    >>> my_add = torch.overrides.get_testing_overrides()[torch.add]
    >>> inspect.signature(my_add)
    <Signature (input, other, out=None)>
    Nc                     dS N inputouts     r0   <lambda>z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  r>  output_sizes     r0   r@  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r:  r<  )inputsrF  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  r>  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r:  r<  r>  otherr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      " r3      c                     dS r:  r<  r>  batch1batch2alphabetar?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r:  r<  r>  tensor1tensor2valuer?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r:  r<  rc  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  r>  mat1mat2r`  r_  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  )r>  matvecr`  r_  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  )r>  vec1vec2r`  r_  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  thetasizealign_cornerss      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r>  dims     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   h㈵>:0yE>Fc                     dS r:  r<  )r>  rX  trolatol	equal_nans        r0   r@  z'get_testing_overrides.<locals>.<lambda>      VX r3   c                     dS r:  r<  r>  ptraininplaces       r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r>  r|  keepdimr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  )r>  msgs     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                      dS r:  r<  tensorss    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   r   Tc                     dS r:  r<  )r>  kernel_sizer  padding	ceil_modecount_include_pads         r0   r@  z'get_testing_overrides.<locals>.<lambda>      vx r3   c                     dS r:  r<  r\  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      PR r3   c	                     dS r:  r<  )	r>  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r0   r@  z'get_testing_overrides.<locals>.<lambda>      y{ r3   c                     dS r:  r<  )grad_outr>  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r>  r  r  r  input_gweight_gbias_gs           r0   r@  z'get_testing_overrides.<locals>.<lambda>      su r3   c                     dS r:  r<  )r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  r>  r  r  r  r  r  r  counts           r0   r@  z'get_testing_overrides.<locals>.<lambda>      tv r3   c                     dS r:  r<  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>        AC r3   c                     dS r:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r:  r<  )r>  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>      Z\ r3   c                     dS r:  r<  )r>  	generatorr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  input1input2r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   r  c                     dS r:  r<  r>  targetr  size_averagereduce	reduction
pos_weights          r0   r@  z'get_testing_overrides.<locals>.<lambda>      rt r3   c                     dS r:  r<  )r>  weights	minlengths      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  probr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      " r3   c                      dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r>  rl  	out_dtyper?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                      dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  selfrx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  )r>  
boundaries	out_int32rightr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>      [] r3   c                      dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  r|  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r   s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r:  r<  r   s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3          @#use_mm_for_euclid_dist_if_necessaryc                     dS r:  r<  )x1x2r  compute_modes       r0   r@  z'get_testing_overrides.<locals>.<lambda>      _a r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>   rL  r3         ?c                     dS r:  r<  r>  r_  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   )r?  c                     dS r:  r<  )r?  matricess     r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  r>  groupss     r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r:  r<  r>  upperr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  check_errorsr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r:  r<  )r>  numeln_binsratio	bit_widths        r0   r@  z'get_testing_overrides.<locals>.<lambda>	      WY r3   c                     dS r:  r<  r>  chunksr|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>
  rY  r3   c                     dS r:  r<  r>  minmaxr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r,  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  )r>  r-  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r.  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  r?  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  
correctionfweightsaweightss       r0   r@  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>      2 r3      c                     dS r:  r<  )r>  rwith_replacements      r0   r@  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r:  r<  )realimags     r0   r@  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )absangs     r0   r@  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r:  r<  )r>  ords     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  padrf  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r:  r<  r>  r  r  r  r  dilationr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>      bd r3   c                     dS r:  r<  rT  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c                     dS r:  r<  rT  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c	                     dS r:  r<  )	r>  r  r  r  r  rU  
transposedoutput_addingr  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>      uw r3   c                     dS r:  r<  )r>  r  r  rQ  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>   rR  r3   c                     dS r:  r<  r>  r  r  r  r  output_paddingr  rU  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>!        A r3   c                     dS r:  r<  r_  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>"  ra  r3   c                     dS r:  r<  r_  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>#  ra  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>$  rI  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>%  rA  r3   c                     dS r:  r<  r  r  r  marginr  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>&  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>'  rL  r3   c                     dS r:  r<  )r  r  r|  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>(  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rA  r3   c                     dS r:  r<  r>  rX  r|  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>*  rG  r3   r;  c                     dS r:  r<  rm  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>+      2 r3   c                     dS r:  r<  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r0   r@  z'get_testing_overrides.<locals>.<lambda>-  r  r3   c                     dS r:  r<  r>  r|  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>/  r  r3   c                     dS r:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>0  r  r3   c                     dS r:  r<  r>  r|  r?  r_   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>1  r0  r3   c                     dS r:  r<  r|  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rJ  r3   c                     dS r:  r<  yxr|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rG  r3   c                     dS r:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>4  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rU  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>6       r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>7       r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>9  r  r3   c                     dS r:  r<  r>  diagonalr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>:  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rG  r3   c                     dS r:  r<  )r>  offsets     r0   r@  z'get_testing_overrides.<locals>.<lambda><  rC  r3   c                     dS r:  r<  )r>  nr|  prependappendr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>=      TV r3   c                     dS r:  r<  r>  r  dim1dim2s       r0   r@  z'get_testing_overrides.<locals>.<lambda>>  r0  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rt  r3   c                     dS r:  r<  )r>  srcr  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>@  r9  r3   c                     dS r:  r<  )r  r  rx  r  storage_offsets        r0   r@  z'get_testing_overrides.<locals>.<lambda>A  r'  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rU  r3   c                     dS r:  r<  )r>  rX  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>C  rQ  r3   c                     dS r:  r<  r>  rX  rounding_moder?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>D      b r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rt  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rY  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rG  r3   c                     dS r:  r<  r>  rl  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>H  r  r3   c                     dS r:  r<  )rk  rl  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>I      r r3   c                     dS r:  r<  r>  indices_or_sectionss     r0   r@  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS r:  r<  r4  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>K  rC  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>L  rY  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>M  r  r3   Lc                     dS r:  r<  r>  UPLOr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>N  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>O  r  r3   c                     dS r:  r<  )equationoperandss     r0   r@  z'get_testing_overrides.<locals>.<lambda>P  rY  r3   c                     dS r:  r<  r>  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r0   r@  z'get_testing_overrides.<locals>.<lambda>R      z| r3   c
                     dS r:  r<  )
r>  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>U  s      hj r3   c                     dS r:  r<  r>  r_   rm   r^   requires_grads        r0   r@  z'get_testing_overrides.<locals>.<lambda>W      ce r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>X       r3   c                     dS r:  r<  r>  rX  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rD  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rQ  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rS  r3   c                     dS r:  r<  )r>  scale
zero_pointaxis	quant_min	quant_maxs         r0   r@  z'get_testing_overrides.<locals>.<lambda>`      mo r3   c                     dS r:  r<  )r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>a      fh r3   c                     dS r:  r<  )r  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r0   r@  z'get_testing_overrides.<locals>.<lambda>c  s      AC r3   c                     dS r:  r<  r>  packed_weightr  outputs       r0   r@  z'get_testing_overrides.<locals>.<lambda>e  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>f      df r3   c                     dS r:  r<  r>  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>g      {} r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>i      ^` r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>k  rR  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>l  r  r3   c                     dS r:  r<  )r>  abs      r0   r@  z'get_testing_overrides.<locals>.<lambda>m  r0  r3   c                     dS r:  r<  r>  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>n  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>o  r  r3   c                     dS r:  r<  r>  r  r|  norms       r0   r@  z'get_testing_overrides.<locals>.<lambda>p  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>q  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>r  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>s  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>t  r  r3   r  r;  c                     dS r:  r<  r>  sr|  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rp  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>v  rt  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>w  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>x  ro  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>y  ro  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>z  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>{  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>|  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>}  r@  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rp  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  )r>  	start_dimend_dims      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  r>  dimss     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  exponentr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  )r>  
fill_valuer?  r_   rm   r^   r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  s      BD r3   c                     dS r:  r<  )r>  r  	dep_tokens      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r|  indexr?  sparse_grads        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  rs  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r;  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  spacingr|  
edge_orders       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r3   c                     dS r:  r<  r>  gridinterpolation_modepadding_modery  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>      ac r3   c                     dS r:  r<  rA  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rA  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  
num_groupsr  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      km r3   c	                     dS r:  r<  	r>  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r0   r@  z'get_testing_overrides.<locals>.<lambda>      qs r3   c                     dS r:  r<  r>  rM  w_ihw_hhb_ihb_hhs         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3         ?c                     dS r:  r<  r>  lambds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r|  r  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  valuesr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  rh  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      xz r3   d   c                     dS r:  r<  )r>  binsr-  r.  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  ri  r-  r.  r  densityr?  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  )r>  ri  r   r  rk  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r'  r3   c                     dS r:  r<  r>  taus     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )rk  rl  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r4  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r>  r|  r1  sources       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r:  r<  rx  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  indicesrc  
accumulates       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  )r>  r|  r1  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  )r>  r|  r1  rf  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r:  r<  )r>  r|  r1  ry  r  include_inputs         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rY   s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )eteassume_uniqueinverts       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c	                     dS r:  r<  )	r>  r  r  r  r  use_input_statsr  r  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  rX  rtolr  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>      UW r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r3   c
                     dS r:  r<  )
r>  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r0   r@  z'get_testing_overrides.<locals>.<lambda>        bd r3   c                     dS r:  r<  r>  r  r  r  r  
log_targets         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    pr r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r:  r<  )r>  kr|  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r3   c                     dS r:  r<  )r>  	hermitianr  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rE  r3   c                     dS r:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  )LDpivotsBr  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>      QS r3   c                     dS r:  r<  )r>  normalized_shaper  r  espr  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  endr  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  )r>  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r0   r@  z'get_testing_overrides.<locals>.<lambda>  s      IK r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r>  r|  r_   s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  )r>  namesr  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c	                     dS r:  r<  )	databatch_sizesrM  rN  rO  rP  rQ  r  rR  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r:  r<  rV  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )Apivot	get_infosr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r:  r<  )r  r,  r-  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  rg  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  )r>  maskrf  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS r:  r<  )LUr  r  leftadjointr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      Y[ r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  )r>  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r:  r<  r4  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rA  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  r  r  rU  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      jl r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  r  r  rU  return_indicesr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  )r>  r|  r  r_   r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                      dS r:  r<  )r  r'   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r  r  r  exponential_average_factorepsilons           r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c	                     dS r:  r<  	r>  r  r  r  r  rU  r  	benchmarkdeterministics	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c	                     dS r:  r<  )	r>  r  zr_  r  r  r  rU  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rT  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r:  r<  )
r>  r  r  r  r`  r  rU  r  r  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>!  r\  r3   c	                     dS r:  r<  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>$      eg r3   c                     dS r:  r<  )r>  r  weight_stride0rM  cxr  hidden_sizerP  rS  rQ  r  rR  r  dropout_states                 r0   r@  z'get_testing_overrides.<locals>.<lambda>'  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>)  r0  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>*  ro  r3   c                     dS r:  r<  r>  ry  destinations      r0   r@  z'get_testing_overrides.<locals>.<lambda>+  r  r3   c                     dS r:  r<  r+  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rR  r3   c                     dS r:  r<  )r>  
descendingr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>-  rJ  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rY  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>/  r  r3   c                     dS r:  r<  )r>  num_samplesreplacementr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>0      SU r3   c                     dS r:  r<  )r>  ro  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rU  r3   c                     dS r:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>2  r  r3   c                     dS r:  r<  )r>  r|  startr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>3  r  r3           c                     dS r:  r<  )r>  nanposinfneginfr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>4  r  r3   c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rT  r3   c                     dS r:  r<  )r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>6      ]_ r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>7  r  r3   c                     dS r:  r<  r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS r:  r<  r>  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>9  r'  r3   c                     dS r:  r<  )r>  r  r  NCHxWgroupr  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>:  r  r3   c                     dS r:  r<  )r>  r  r|  r  r_   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>;  r5  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda><  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>=  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rC  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS r:  r<  rE  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>B  r!  r3   c                     dS r:  r<  rE  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>C  r!  r3   c                     dS r:  r<  r>  rF  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>D  rV  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>E      oq r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rV  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rY  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rV  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>I  rY  r3   c                     dS r:  r<  rv  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS r:  r<  r>  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>K  r
  r3   c                     dS r:  r<  r>  r  r  r  r  r  divisor_overrides          r0   r@  z'get_testing_overrides.<locals>.<lambda>M        @B r3   c                     dS r:  r<  rb  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>P  rd  r3   皙?c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>S  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>U  r  r3   c                     dS r:  r<  r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>W  rE  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rg  r3   c                     dS r:  r<  rg  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>^      gi r3   c                     dS r:  r<  )r>  r  r  r  ignore_indexr  r  label_smoothings           r0   r@  z'get_testing_overrides.<locals>.<lambda>a        JL r3   c                     dS r:  r<  rq  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>d  r  r3   c                     dS r:  r<  r`  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>f      XZ r3   c                     dS r:  r<  r`  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>g  r  r3   c                     dS r:  r<  r`  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>h  r  r3   c                     dS r:  r<  r`  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>i  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rt  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>l  r  r3   c                     dS r:  r<  )r>  r  r  r  r  r  r  r   r  include_last_offsetr  s              r0   r@  z'get_testing_overrides.<locals>.<lambda>o  s      HJ r3   c                     dS r:  r<  r`  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>q  rn  r3   c                     dS r:  r<  )r>  rF  r  rU  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rJ  r3   c                     dS r:  r<  r>  r  rF  output_ratior  _random_sampless         r0   r@  z'get_testing_overrides.<locals>.<lambda>t  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rf  r3   ư>c                     dS r:  r<  )r>  r  varr   r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   nonec                     dS r:  r<  )r>  approximates     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   bilinearr   c                     dS r:  r<  )r>  rB  r  rD  ry  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r:  r<  )r>  rI  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r"  r3   绽|=c                     dS r:  r<  )logitsro  hardr  r|  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rE  r3   c                     dS r:  r<  r_  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3         c                     dS r:  r<  )r>  min_valmax_valr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  re  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>      `b r3   c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  s      GI r3   nearestc                     dS r:  r<  )r>  rx  scale_factorr  ry  recompute_scale_factor	antialiass          r0   r@  z'get_testing_overrides.<locals>.<lambda>  s      KM r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rE  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   {Gz?c                     dS r:  r<  )r>  negative_sloper  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   -C6?      ?c                     dS r:  r<  )r>  rx  r_  r`  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r"  r3      c                     dS r:  r<  r>  r|  _stacklevelr_   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    \^ r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  rg  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS r:  r<  r>  r  r  r  rU  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r|  r  r  r  rF  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )querykeyrf  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r0   r@  z'get_testing_overrides.<locals>.<lambda>  s      ]_ r3   c                     dS r:  r<  )r>  r  r  rh  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rj  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rE  r3   c                     dS r:  r<  )r>  r  r  r  rq  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   -q=c                     dS r:  r<  )r>  r  r|  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )rY   num_classess     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   r   c                     dS r:  r<  )r>  rQ  r  rf  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r
  r3   c                     dS r:  r<  )r>  r  	log_inputr   r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  rG  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rB  r3         ?UUUUUU?c                     dS r:  r<  r>  lowerr  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    wy r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  )r  r  rf  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r  r  r`  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  )r>  r  r  deltar  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    hj r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rv  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rv  r3      c                     dS r:  r<  )r>  r`  	thresholds      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  r_  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  r   rf  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r:  r<  
anchorpositivenegativerh  r  r  swapr  r  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rs  r3   )distance_functionrh  r  r  c                    dS r:  r<  )r  r	  r
  r  rh  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  rU  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )rY   r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  )rY   r  stdr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )rY   vals     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   fan_in
leaky_reluc                     dS r:  r<  )rY   r  r  nonlinearityr  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  as_tuples     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   )r(  c                    dS r:  r<  )r>  rx  r(  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   froc                     dS r:  r<  r>  r  r|  r  r?  r_   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  rK  r|  r  r?  r_   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c                     dS r:  r<  r   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s     13 r3   c                     dS r:  r<  )vpowr|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r3   c                     dS r:  r<  rn  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  input3r  	transposes        r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  r|  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  qr  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r8  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   V瞯<c                     dS r:  r<  )r>  rconds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r4  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  )r>  upscale_factors     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r:  r<  )r>  downscale_factors     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  )r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r   r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>   r'  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c                     dS r:  r<  r"  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r_   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  )r>  r1  ry  r}  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rY  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  )r>  somer?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   reducedc                     dS r:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   linearc                     dS r:  r<  r>  r0  r|  r  interpolationr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS r:  r<  )r>  scaleszero_pointsr  r_   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r_   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r_   reduce_ranges      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r  r  r  output_scaleoutput_zero_points           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r:  r<  r>  rM  rW  rX  rY  rZ  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r0   r@  z'get_testing_overrides.<locals>.<lambda>        _a r3   c                     dS r:  r<  r[  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>  rd  r3   r<  r   rZ  c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s
     " r3   r   r   rZ  rZ  c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>!  s     !# r3   r   r   r   rZ  rZ  rZ  c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>'  s     !#	 r3   c                     dS r:  r<  r[  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rd  r3   c                     dS r:  r<  r[  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rd  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rU  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rV  r3   c                     dS r:  r<  )r>  highr_   rm   r^   r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>5  r  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>6  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>7  r;  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>8  rL  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>9  r  r3   c                     dS r:  r<  rm  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>:  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda><  rS  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rY  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>?  r  r3   c                     dS r:  r<  )r>  r  r|  maxnormr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rJ  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS r:  r<  )r>  shapes     r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rL  r3   c                     dS r:  r<  rG  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>C  ra  r3   c	                     dS r:  r<  rL  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>D  r  r3   c                     dS r:  r<  rV  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>E  r  r3   c	                     dS r:  r<  rL  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>F  r  r3   c                     dS r:  r<  rV  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>G  r  r3   c                     dS r:  r<  )r>  shiftsr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>H  r  r3   r   rZ  c                     dS r:  r<  )r>  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>J  rS  r3   c                     dS r:  r<  r4  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>K  r  r3   c                     dS r:  r<  )r  r  compressed_indices_dtypes      r0   r@  z'get_testing_overrides.<locals>.<lambda>L  r  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>M  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>N  rS  r3   c                     dS r:  r<  )r>  rX  r_  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rY  r3   c                     dS r:  r<  rj  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>P  r  r3   c                     dS r:  r<  r>  r|  r1  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>Q  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>R  r  r3   c                     dS r:  r<  )r>  r|  r1  r  r  include_selfs         r0   r@  z'get_testing_overrides.<locals>.<lambda>S  rv  r3   c                     dS r:  r<  )sorted_sequencer>  r  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>T  r  r3   r.  c                     dS r:  r<  )r  r  lengthsr|  r  r  unsafes          r0   r@  z'get_testing_overrides.<locals>.<lambda>U  r  r3   c                     dS r:  r<  )r>  r|  r1  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rC  r3   c                     dS r:  r<  )r>  r  r|  r1  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS r:  r<  r>  r  r|  r:  r  steps         r0   r@  z'get_testing_overrides.<locals>.<lambda>X  r  r3   c                     dS r:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rU  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>`  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rL  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>b  rO  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>c  rL  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>d  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>e  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>f  r  r3   c                     dS r:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rG  r3   c                     dS r:  r<  )r  r  r  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>h  r  r3   )stabler?  c                    dS r:  r<  )r>  r|  r/  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>i  r'  r3   c                     dS r:  r<  rY   split_size_or_sectionsr|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>j  ro  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>k  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rL  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>m  rQ  r3   c                     dS r:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>n  r  r3   c                     dS r:  r<  rj  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>o  r9  r3   c                     dS r:  r<  r   s      r0   r@  z'get_testing_overrides.<locals>.<lambda>p  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>q  rA  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rC  r3   reflectc                     dS r:  r<  )r>  r  r  r  r  r  pad_moder  r  r  align_to_windows              r0   r@  z'get_testing_overrides.<locals>.<lambda>t  s      ~@ r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>v  rY  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>w  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>x  rA  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>y  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rO  r3   c                     dS r:  r<  r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>{  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>|  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rO  r3   c                     dS r:  r<  )r  r  cs      r0   r@  z'get_testing_overrides.<locals>.<lambda>~  r  r3   c                     dS r:  r<  )r&   s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  )r>  rI  
compute_uvr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3      c                     dS r:  r<  )r>  r0  r  Ms       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  )r>  full_matricesr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r>  dim0r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  )r>  axis0axis1s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r8  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  rW  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  rX  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    r r3   c                     dS r:  r<  )r>  r1  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r|  r|  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  )r  inds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r|  r/  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  rN  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r+  unitriangulars        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r  r  r  r4  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r
  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r:  r<  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  )r>  r|  sizesr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  sortedreturn_inversereturn_countsr|  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  r@  rA  r|  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r"  r3   c                     dS r:  r<  )r|  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r)  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  rI  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r{  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r4  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  )	conditionr  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r>  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  levels     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )primaltangentrY  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  )r  rx  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   )implicitc                    dS r:  r<  )r  rx  re  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r|  r:  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  rx  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  )r  r|  r1  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r|  r:  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r3   c                     dS r:  r<  )r  
split_sizer|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  split_sizesr|  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r.  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>	  r;  r3   c                     dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS r:  r<  r.  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r.  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  r_   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r  	dimensionrx  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  rX  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>   rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>!  rU  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>"  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>#  rC  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>$  rL  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>%  rA  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>&  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>'  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>(  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>*  rI  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>+  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rO  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>-  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rL  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>/  rS  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>0  rS  r3   c                     dS r:  r<  )r  arrays     r0   r@  z'get_testing_overrides.<locals>.<lambda>1  r  r3   c                     dS r:  r<  )r  idxs     r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rQ  r3   c                     dS r:  r<  )r  memos     r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rC  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rO  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rI  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>6  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>7  rO  r3   c                     dS r:  r<  )r  format_specs     r0   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS r:  r<  )r  protos     r0   r@  z'get_testing_overrides.<locals>.<lambda>9  rY  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>:  rD  r3   )tensor_contentsc                    dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rJ  r3   c                     dS r:  r<  )r  r  r$  s      r0   r@  z'get_testing_overrides.<locals>.<lambda><  rU  r3   c                     dS r:  r<  )r  ds     r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>?  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>@  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>B  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>C  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>D  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>I  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>J  rQ  r3   c                     dS r:  r<  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r0   r@  z'get_testing_overrides.<locals>.<lambda>K  s    np r3   c                     dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>L  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>M  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>N  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>P  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Q  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>R  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>S  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>T  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>U  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>W  rY  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>X  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rY  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>`  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>b  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>c  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>d  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>e  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>f  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>h  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>i  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>k  rQ  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>m  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>n  rG  r3   c                     dS r:  r<  )r  r_   non_blockingr'   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>o  r!  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>p  r;  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>q  r;  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rI  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>s  rI  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>t      " r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>v  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>x  r  r3   c                     dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>y  rR  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rO  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>{  rO  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>|  rS  r3   c                     dS r:  r<  )r  orderellipsis_idxs      r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rR  r3   c                     dS r:  r<  )r  callables     r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rQ  r3   c                     dS r:  r<  rj  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rj  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  gradientretain_graphcreate_graphrI  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    ik r3   c                     dS r:  r<  r  rf   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   )r  c                    dS r:  r<  )r  mediansigmar  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  )r  	coalesceds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  ambiguity_checks     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                    dS r:  r<  )r  r`  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  rf  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r"  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                    dS r:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  rY   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                    dS r:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r.  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  )r  rY   r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  rl  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  rX  assigns      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS r:  r<  )r  r  r:  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r3   c                     dS r:  r<  r.  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r|  rY   r}  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                    dS r:  r<  )r  from_tor  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r3   c                     dS r:  r<  )r  streams     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  hooks     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rO  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  names     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  ry  r  rx  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r|  r1  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  r|  r:  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  )r  r  accumulate_matchess      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  size1size2	dense_dims       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  rg  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r3   c                     dS r:  r<  )r  rk  rl  r`  r_  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  )r  repss     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r_   r  copyrf   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  s    ln r3   )masked_gradc                    dS r:  r<  r  r_   rv  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r:  r<  rx  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r:  r<  )r  rI  rJ  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r3   c                     dS r:  r<  )r  rL  max_version	dl_deviceru  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r:  r<  )r  r  r   drivers       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  )r  r^   r  r'   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r:  r<  r^  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  r  r3   is______i__rbitwise_N)rZ  rZ  N)rZ  N)r}  r~  F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)r  r  )r  F)FN)NNN)rZ  NN)r<  F)NrZ  r   rZ  rZ  )NrZ  r   r   rZ  rZ  )r   NNr  )rZ  r~  )r;  N)r   r  Fr:  )rZ  r;  NNN)r   r   rZ  )r   r  r;  )r<  )r  N)NNr  FF)Nr<  Fr  FNN)NNNF)FF)Nr;  N)Nr  N)r   r;  )TT)NF)NNrZ  )NNr}  T)r]  )NFr   N)r  NNr  )rg  r   r   N)rg  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rZ  F)Nr   rZ  FF)NFNN)r;  FN)r;  NNN)NNr}  )Nr}  )r<  NFN)r]  FF)Nr   FTN)NNFrf  r}  )NNNr  )NNro  Nr  r;  )r]  TF)	NNr<  Fr  FNFN)rZ  r   rZ  )NNFN)Fr  r  )r  )r;  )r  r   N)rZ  Fr  r;  )r  r  F)NNNNTrf  r}  )NNr  NNF)NNr  N)r  F)r  r  r  )Nr  N)Nr   N)TNTNFNNNNNNF)rZ  r  NNNr  )NNr  )NNro  Nr  )r<  rZ  r  N)r   r   )r  r  F)TFNr~  Nr  )Nr  )r  r  FF)Nr;  )NNr  r  )r  r  N)rZ  r  )r  r<  r  FNNr  )r;  r  N)r   r  r  N)r  NFNN)NNFNN)r<  NFNN)r  r  FNN)r<  r   )TF)NTr<  )r2  )r2  F)rJ  N)NFrL  N)r<  rf  rg  F)r<  ri  rj  F)r<  rl  rm  F)rZ  r  )r.  NNNr   F)r   NNrZ  )r;  F)	NNNTr  FTNN)TTN)r  r<  N)r<  N)TFF)TFFN)rZ  r<  )Nr   NN)NNNN(  r    r5   rG  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshaddaddbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullir   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffr  diagonal_scatteras_strided_scatterdigammadistdivdividedotrQ  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdge
get_devicegreater_equalgeqrfi0inneroutergerr   grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrinkhash_tensor	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacrC  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr.  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr	  	nanmedianmeshgridr-  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_fused_rms_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr
  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_lossr  rL  local_response_norm
logsigmoid	lp_pool1d	lp_pool2d	lp_pool3dmax_pool2d_with_indicesmax_pool3d_with_indicesmax_unpool1dmax_unpool2dmax_unpool3dmse_lossmulti_head_attention_forwardmulti_margin_lossmultilabel_margin_lossmultilabel_soft_margin_lossnll_loss	normalizeone_hotrQ  pairwise_distancepoisson_nll_lossprelurelurelu6rms_normrreluselusilumishscaled_dot_product_attentionsmooth_l1_loss
huber_losssoft_margin_losssoftmaxsoftminsoftplus
softshrinksoftsign
tanhshrinkr   triplet_margin_loss!triplet_margin_with_distance_lossunfoldr   uniform_normal_	constant_kaiming_uniform_nonzerononzero_staticargwherer  vector_normmatrix_normnorm_except_dimnuclear_normr#  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar	  	ones_liker%  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2deg	rand_likerandint_like
randn_likeravelrB  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_itesym_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer+  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr'  HmTmH_backward_hooks_post_accumulate_grad_hooksr4  _cdatar5  r6  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  r^   r_   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizerm   rS  r  nbytesndim	output_nrr  r  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_update_names_valuesalign_asalign_toapply_rp   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrrj  r|  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesr|  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rg   random_record_streamrefine_namesregister_hook"register_post_accumulate_grad_hookrenamerepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_shortrx  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizerJ  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asrc  viewview_aszero_
__dlpack____dlpack_device__r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrr   items__name__
startswithlenextendr  update)r5   retprivateuse1_backend_nameret2ignoredr  r$  r  subnamerS  r!   s              r0   r   r     s_   6 \F|%	---|%222|% 	!#@#@|% 	!#A#A	|%
 	
...|% 	''|% 	000|% 	///|% 	111|% 		444|% 	QQQ|% 	LLL|% 	LLL|% 	LLL|% 	JJJ|%  	
KKK!|%" 	#%J%J#|% |%$ 		---%|%& 	XXX'|%( 	FFF)|%* 	
...+|%, 	
...-|%. 	JJJ/|%0 	///1|%2 		FFF3|%4 	&&5|%6 	&&7|%8 	1119|%: 	
...;|%< 	22=|%> 	000?|%@ 	///A|%B 	111C|%D 	
...E|% |% |%F 	000G|%H 	666I|%J 	888K|%L 	///M|%N 	111O|%P 	--Q|%R 	--S|%T 	--U|%V 	xxxW|%X 	RRRY|%Z 	{{[|%\ 	'){){]|%^ 	(*u*u_|%` 	 Q Qa|%b 	%'v'vc|%d 	1  4C  4Ce|%f 	 5 5g|% |% |%h 	%'\'\i|%j 	CCCk|%l 	??m|%n 	.tttq|%t 	CCCu|%v 	>>>w|%x 	<<<y|%z 	555{|%| 	;;;}|%~ 	<<<|%@ 	 "C"C"CA|%B 	!#D#D#DC|%D 	--E|%F 		CCCG|%H 	!4!4I|%J 	11K|%L 	]]]M|% |% |%N 	11O|%P 		666Q|%R 	999S|%T 	>>>U|%V 	aaaW|%X 	
...Y|%Z 	
>>>[|%\ 	$::::]|%^ 	77_|%` 	???a|%b 	999c|%d 	 "P"P"Pe|%f 	 G G Gg|%h 	NNNi|%j 	&(Y(Yk|%l 	444m|%n 	CCCo|% |% |%p 	
BBBq|%r 	888s|%t 	888u|%v 	888w|%x 		OOOy|%z 	%%{|%| 	III}|%~ 	,,|%@ 	999A|%B 	((C|%D 	555E|%F 	
...G|%H 	777I|%J 	666K|%L 	555M|%N 	===O|%P 	dddQ|% |% |%R 	dddS|%T 	dddU|%V 	wwW|%X 	===Y|%Z 	  !A  !A  !A[|%\ 	  !A  !A  !A]|%^ 	  !A  !A  !A_|%` 	((a|%b 		---c|%d 	#  &C  &C  &Ce|%f 	
...g|%h 	!C!C!Ci|%j 	--k|%l 	@@@m|%n 	EEEo|%p 	xxxs|%v 	555w|% |% |%x 	555y|%z 	BBB{|%| 	AAA}|%~ 	"$@$@$@|%@ 	;;;A|%B 	111C|%D 	**E|%F 		##G|%H 	**I|%J 	&&K|%L 	
:::M|%N 	@@@O|%P 	222Q|%R 	
VVVS|%T 	BBBU|%V 	KKKW|%X 	 O O OY|% |% |%Z 	 "Y"Y"Y[|%\ 	111]|%^ 	
000_|%` 		HHHa|%b 	KKKc|%d 		444e|%f 	@@@g|%h 	
:::i|%j 	
))k|%l 	;;m|%n 	222o|%p 	444q|%r 	888s|%t 	???u|%v 	CCCw|%x 	44y|%z 	|||}|% |% |%@ 	 j  j  jC|%F 	eeeG|%H 	333I|%J 	,,K|%L 		---M|%N 	
...O|%P 	000Q|%R 		---S|%T 	
...U|%V 	///W|%X 	.0o0oY|%Z 	-/h/h[|%\ 	- C  C  C_|%b 	')V)Vc|%d 	79f9fe|%f 	')})}g|%h 	7``k|% |% |%n 	+-=-=o|%p 	*,<,<q|%r 	*,B,Bs|%t 	#%?%?u|%v 	99w|%x 		CCCy|%z 		CCC{|%| 		DDD}|%~ 		CCC|%@ 		DDDA|%B 		JJJC|%D 		KKKE|%F 		DDDG|%H 		EEEI|%J 		EEEK|%L 		FFFM|%N 		FFFO|% |% |%P 		GGGQ|%R 		IIIS|%T 		JJJU|%V 		JJJW|%X 		KKKY|%Z 		666[|%\ 		777]|%^ 		BBB_|%` 		---a|%b 	@@@c|%d 	
**e|%f 	&&g|%h 	&&i|%j 	QQQk|%l 	///m|%n 	33o|%p 	???q|% |% |%r 	
555s|%t 	
...u|%v 	///w|%x 	t4PUP]fjz  D  D  Dy|%z 	&(H(H{|%| 	\\\}|%~ 	OOO|%@ 		444A|%B 	333C|%D 	**E|%F 	>>>G|%H 	///I|%J 	,,,K|%L 	666M|%N 	555O|%P 		333Q|%R 	NNNS|% |% |%T 	ccU|%V 	ffW|%X 	ffY|%Z 	mmm[|%\ 		ss]|%^ 	NNN_|%` 	333a|%b 	888c|%d 	555e|%f 	VVVg|%h 	;;;i|%j 	"$z$z$zk|%l 	GGGm|%n 	mmmo|%p 	YYYq|%r 	(*?*?s|%t 	444u|% |% |%v 	;;w|%x 	222y|%z 	666{|%| 	777}|%~ 	888|%@ 	
...A|%B 	==C|%D 	>>E|%F 	LLLG|%H 	BBBI|%J 	==K|%L 	\\\M|%N 	))O|%P 	
GGGQ|%R 	&&S|%T 	''U|%V 	222W|% |% |%X 	222Y|%Z 	tt]|%` 	((a|%b 	111c|%d 	444e|%f 	KKKg|%h 	**i|%j 	''k|%l 	&&m|%n 	..o|%p 	,,q|%r 	!1!1s|%t 	**u|%v 	33w|%x 	))y|%z 	WWW{|%| 	%%}|% |% |%~ 	 d  d  dA	|%D	 	rrrE	|%F	 	
++G	|%H	 	NNNI	|%J	 	"$c$c$cK	|%L	 	!L!L!LM	|%N	 	 S S SO	|%P	 	sssQ	|%R	 		444S	|%T	 	666U	|%V	 	333W	|%X	 	;;;Y	|%Z	 	
;;;[	|%\	 	000]	|%^	 	  K  K  K_	|%`	 		---a	|%b	 	<<<c	|% |% |%d	 	///e	|%f	 	///g	|%h	 	
...i	|%j	 	:::k	|%l	 	;;;m	|%n	 	&&o	|%p	 	...q	|%r	 	<<<s	|%t	 	555u	|%v	 	;;;w	|%x	 	<<<y	|%z	 	///{	|%|	 	III}	|%~	 	
ss	|%@
 	OOOA
|%B
 	333C
|%D
 	
555E
|% |% |%F
 	EEEG
|%H
 	BBBI
|%J
 	!  $A  $A  $AK
|%L
 	88M
|%N
 	<<O
|%P
 	===Q
|%R
 	777S
|%T
 	???U
|%V
 	 F F FW
|%X
 	!#]#]#]Y
|%Z
 	[[[[
|%\
 	>>>]
|%^
 	//_
|%`
 	!#@#@#@a
|%b
 	 "M"M"Mc
|%d
 	 < < <e
|%f
 	**g
|% |% |%h
 	!1!1i
|%j
 		---k
|%l
 	888m
|%n
 	
555o
|%p
 	lllq
|%r
 	llls
|%t
 	lllu
|%v
 	%ttty
|%|
 	
...}
|%~
 	VVV
|%@ 	000A|%B 	333C|%D 	55E|%F 		---G|%H 	888I|%J 	
555K|%L 	ttO|% |% |%R 	 "}"}S|%T 	)+v+vU|%V 	%'h'hW|%X 	*ww[|%^ 	*gga|%d 	 d  dg|%j 	BBBk|%l 	
EEEm|%n 	<<o|%p 	==q|%r 	AAAs|%t 		444u|%v 	999w|%x 	UUUy|%z 	111{|%| 	++}|%~ 	::|% |% |%@ 	WWWA|%B 	!s!sC|%D 	&(_(_E|%F 	88G|%H 	!f!f!fI|%J 	YYYK|%L 	!V!VM|%N 	UUUO|%P 	$&>&>Q|%R 	333S|%T 	:::U|%V 		---W|%X 	222Y|%Z 	:::[|%\ 	/1N1N]|%^ 	/1N1N_|%` 	/1d1d1da|% |% |%b 	<>q>q>qc|%d 	/1d1d1de|%f 	<>q>q>qg|%h 	/1d1d1di|%j 	<>q>q>qk|%l 	')S)S)Sm|%n 	)+a+a+ao|%p 	& B  B  Bs|%v 	& B  B  By|%| 	&xxx|%B 	$&R&R&RC|%D 	0cccG|%J 	<tttM|%P 	 "L"L"LQ|%R 	1iiiU|%X 	) L  L  L[|%^ 	$xxxa|% |% |%d 	#%Z%Z%Ze|%f 	%'\'\'\g|%h 	%'\'\'\i|%j 	%'\'\'\k|%l 	!K!K!Km|%n 	%|||q|%t 	) J  J  Jw|%z 	13i3i3i{|%| 	 "m"m"m}|%~ 	1zzzA|%D 	>zzzG|%J 	1zzzM|%P 	>zzzS|%V 	-/u/u/uW|%X 	 "F"F"FY|%Z 	!9!9!9[|%\ 	')z)z)z]|% |% |%^ 	&(g(g(g_|%` 	*,c,c,ca|%b 	&(C(C(Cc|%d 	$&`&`&`e|%f 	0bbbi|%l 	) I  I  Io|%r 	' M  M  Mu|%x 	"  %A  %A  %Ay|%z 	#%|%|%|{|%| 	&(m(m(m}|%~ 	&(\(\(\|%@ 	"$G$G$GA|%B 	/1g1g1gC|%D 	')^)^)^E|%F 	&(8(8G|%H 	%'m'm'mI|%J 	%'m'm'mK|% |% |%L 	%'m'm'mM|%N 	/iiiQ|%T 	&tttW|%Z 	3ttt]|%` 	&tttc|%f 	3ttti|%l 	&ttto|%r 	3tttu|%x 	(*z*z*zy|%z 	(*z*z*z{|%| 	(*z*z*z}|%~ 	$&}&}&}|%@ 	8 _  _  _C|%F 	-tttI|%L 	2VVVO|%R 	7cccU|%X 	$vvv[|% |% |%^ 	%'X'X'X_|%` 	#%F%F%Fa|%b 	!P!P!Pc|%d 	-/a/a/ae|%f 	,}}}i|%l 	!#;#;m|%n 	 "A"A"Ao|%p 	!#B#B#Bq|%r 	$&_&_&_s|%t 	!#y#y#yu|%v 	 "A"A"Aw|%x 	 "A"A"Ay|%z 	 "A"A"A{|%| 	8:u:u:u}|%~ 	*  -A  -A  -A|%@ 	&(j(j(jA|%B 	,.x.x.xC|% |% |%D 	#%Z%Z%ZE|%F 	#%Z%Z%ZG|%H 	$&L&L&LI|%J 	&(C(C(CK|%L 	$&6&6M|%N 	&(8(8O|%P 	%'X'X'XQ|%R 	/ L  L  LU|%X 	=DHQT[`lrvvvv[|%^ 	"$b$b$b_|%` 	 O O Oa|%b 	SSSc|%d 	!7!7e|%f 	&(x(x(xg|%h 	777i|%j 	FFFFk|%l 	((m|% |% |%n 	
\\\o|%p 	dddq|%r 	 "h"h"hs|%t 	  +/#3 #3 #3u|%| 	999}|%~ 	ddd|%@ 	%%A|%B 	**C|%D 	QQQE|%F 	!S!S!SG|%H 	++I|%J 	IIIK|%L 	***M|%N 	555O|%P 	IIIQ|%R 	==S|%T 	AAU|% |% |%V 	777W|%X 	 Y YY|%Z 	666[|%\ 	222]|%^ 	--_|%` 	ddda|%b 		777c|%d 	
000e|%f 		DDDg|%h 	 "2"2i|%j 	"$4$4k|%l 	')9)9m|%n 	''o|%p 	,,q|%r 	777s|%t 	CCCu|%v 	fffw|% |% |%x 	iiiy|%z 	"$V$V{|%| 	!#M#M}|%~ 	)+P+P|%@ 	"$s$sA|%B 	  a  aE|%H 	! a  aK|%N 	"J# # #Q|%X 	"L # # #[|%d 	"O 	# # #g|%r 	% a  au|%x 	% a  a{|%~ 	111|%@ 	dddA|%B 	d5=Y]mrvvvC|%D 	eeeE|%F 	%%G|% |% |%H 	
...I|%J 	
555K|%L 	FFFM|%N 	,,O|%P 	//Q|%R 	444S|%T 	
333U|%V 	:::W|%X 	AAAY|%Z 	!;!;!;[|%\ 	..]|%^ 	QQQ_|%` 	xxa|%b 	SSSc|%d 	xxe|%f 	SSSg|%h 	
777i|% |% |%j 	777k|%l 	///m|%n 	555o|%p 	PPq|%r 	bbbs|%t 	///u|%v 	
444w|%x 	MMMy|%z 	88{|%| 	<<}|%~ 	ZZZ|%@ 	eeeA|%B 	|||C|%D 	22E|%F 	??G|%H 	WWWI|%J 	WWWK|% |% |%L 	
333M|%N 	111O|%P 	
...Q|%R 	111S|%T 		---U|%V 		---W|%X 	
...Y|%Z 	
...[|%\ 	'']|%^ 	.._|%` 		999a|%b 	
:::c|%d 	888e|%f 	@@@g|%h 	WWWi|%j 	
YeQUYYYYk|%l 	EEEm|% |% |%n 	 P P Po|%p 	
...q|%r 	000s|%t 	;;;u|%v 	OOOw|%x 	888y|%z 		---{|%| 	222}|%~ 	
 @  @  @A|%D 		444E|%F 	999G|%H 		---I|%J 	))K|%L 	''M|%N 	O|%P 	Q|%R 	''S|% |% |%T 	))U|%V 	W|%X 	))Y|%Z 	(([|%\ 	))]|%^ 	((_|%` 	))a|%b 	((c|%d 	))e|%f 	))g|%h 	))i|%j 	))k|%l 	000m|%n 		IIIo|%p 	AAAq|%r 	HHHs|%t 	888u|% |% |%v 	44w|%x 	66y|%z 	//{|%| 	!1!1}|%~ 	!1!1|%@ 	!1!1A|%B 	!1!1C|%D 	,.K.K.KE|%F 	,.K.K.KG|%H 	,.K.K.KI|%J 	,.K.K.KK|%L 	//M|%N 	,,O|%P 	++Q|%R 	,,S|%T 	--U|%V 	..W|% |% |%X 	,,Y|%Z 	--[|%\ 	--]|%^ 	 A A A_|%` 	!B!B!Ba|%b 	//c|%d 	*,I,I,Ie|%f 	+-J-J-Jg|%h 	**i|%j 	++k|%l 	**m|%n 	++o|%p 	+-J-J-Jq|%r 	+-J-J-Js|%t 	--u|%v 	 0 0w|%x 	!#D#D#Dy|% |% |%z 	--{|%| 	!O!O!O}|%~ 	(*:*:|%@ 	(*:*:A|%B 	(*:*:C|%D 	(*:*:E|%F 	"$7$7G|%H 	,,I|%J 	--K|%L 	!>!>!>M|%N 	++O|%P 	--Q|%R 	/1A1AS|%T 	/1A1AU|%V 	46S6S6SW|%X 	46S6S6SY|%Z 	46S6S6S[|% |% |%\ 	46S6S6S]|%^ 	,,_|%` 	@@@a|%b 	)+;+;c|%d 	@@@e|%f 	>>>g|%h 	<<<i|%j 	!!k|%l 	
++m|%n 	KKKo|%p 		---q|%r 	
...s|%t 	 3 3 3u|%v 	 "<"<"<w|%x 	:::y|%z 	HHH{|%| 	JJJ}|% |% |%~ 	
**|%@ 	
KKKA|%B 	%%C|%D 	55E|%F 	111G|%H 	555I|%J 	 e e eK|%L 	%'a'a'aM|%N 	
:::O|%P 	! L  L  LS|%V 	
:::W|%X 	22Y|%Z 	///[|%\ 	---]|%^ 	<<_|%` 	hhha|%b 	 "g"g"gc|% |% |%d 	66e|%f 	;;;g|%h 	LLLi|%j 	%'W'W'Wk|%l 	888m|%n 	111o|%p 		---q|%r 	222s|%t 	;;u|%v 	222w|%x 	999y|%z 	%'`'`{|%| 	1oo|%B 	eeeC|%D 	55E|%F 	@@G|%H 	I|% |% |%J 	"OOK|%L 	//M|%N 	ooO|%P 	QQQQ|%R 	')>)>S|%T 	FFFU|%V 	%CCCCW|%X 	>>Y|%Z 	11[|%\ 	!#@#@]|%^ 	66_|%` 	??a|%b 	NNNc|%d 	<<<e|%f 	#%H%H%Hg|%h 	00i|%j 	ook|% |% |%l 	99m|%n 	22o|%p 	__q|%r 	OOs|%t 	OOu|%v 	??w|%x 	y|%z 	{|%| 	}|%~ 	|%@ 	111A|%B 	//C|%D 	AAE|%F 	//G|%H 	33I|%J 	44K|%L 	44M|% |% |%N 	22O|%P 	33Q|%R 	33S|%T 	11U|%V 	22W|%X 	22Y|%Z 	11[|%\ 	22]|%^ 	22_|%` 	..a|%b 	--c|%d 	..e|%f 	//g|%h 	OOi|%j 	00k|%l 	m|%n 	33o|% |% |%p 	q|%r 	??s|%t 	..u|%v 	//w|%x 	//y|%z 	55{|%| 	00}|%~ 	22|%@ 	A|%B 	C|%D 	//E|%F 	G|%H 	77I|%J 	44K|%L 	__M|%N 	AAAAO|%P 	11Q|% |% |%R 	//S|%T 	//U|%V 	//W|%X 		??Y|%Z 		??[|%\ 	&]|%^ 	*2OO_|%` 	ooa|%b 	c|%d 	__e|%f 	oog|%h 	i|%j 	k|%l 	m|%n 	-/p/po|%p 	*,V,Vq|%r 	2OOs|% |% |%t 	__u|%v 	w|%x 	ooy|%z 	{|%| 	}|%~ 	|%@ 	A|%B 	C|%D 	E|%F 	#__G|%H 	I|%J 	K|%L 	M|%N 	 //O|%P 	Q|%R 	 //S|%T 	#__U|% |% |%V 	 //W|%X 	$ooY|%Z 	 //[|%\ 	]|%^ 	_|%` 	__a|%b 	ooc|%d 	e|%f 	__g|%h 	 //i|%j 	$ook|%l 	oom|%n 	o|%p 	__q|%r 	__s|%t 	'/u|%v 	NNNw|% |% |%x 	ooy|%z 	oo{|%| 	}|%~ 	|%@ 	__A|%B 	__C|%D 	OOE|%F 	__G|%H 	OOI|%J 	==K|%L 	M|%N 	O|%P 	//Q|%R 	==S|%T 	00U|%V 	88W|%X 	99Y|% |% |%Z 	kkk[|%\ 	E4IMMM]|%^ 	0EIII_|%` 	0EIIIa|%b 	0EIIIc|%d 	MTMMMMe|%f 	g|%h 	66i|%j 	e6MQQQk|%l 	>>>m|%n 	
u/DHHHo|%p 	0EIIIq|%r 	0EIIIs|%t 	
u/DHHHu|%v 	
u/DHHHw|%x 	y|%z 	//{|% |% |%| 	!O!O!O}|%~ 	
OO|%@ 	@@@A|%B 	%2GKKKC|%D 	53HLLLE|%F 	__G|%H 	,,I|%J 	00K|%L 	HHHHHM|%N 	,,O|%P 	55Q|%R 	1FJJJS|%T 	%2GKKKU|%V 	@@@@W|%X 	??Y|%Z 	0EIII[|%\ 	1FJJJ]|% |% |%^ 	//_|%` 	a|%b 	
u/DHHHc|%d 	__e|%f 	oog|%h 	__i|%j 	//k|%l 	11m|%n 	//o|%p 	__q|%r 	MTMMMMs|%t 	00u|%v 	0EIIIw|%x 	66y|%z 	55{|%| 		888}|%~ 	@@@|% |% |%@  	EEA |%B  	??C |%D  	E |%F  	"OOG |%H  	-I |%J  	%K |%L  	M |%N  	ooO |%P  	,,Q |%R  	??S |%T  	GGGU |%V  	W |%X  	LDLLLLY |%Z  	55[ |%\  	33] |%^  	33_ |%`  	13H3Ha |% |% |%b  	,,c |%d  	--e |%f  	BBBg |%h  	11i |%j  	--k |%l  	--m |%n  	00o |%p  	 "8"8q |%r  	OOs |%t  	[[[u |%v  	??w |%x  	ooy |%z  	1FJJJ{ |%|  	__} |%~  	WWW |%@! 	??A!|%B! 	11C!|% |% |%D! 	&(W(W(WE!|%F! 	GGG!|%H! 	')Q)QI!|%J! 	OOOK!|%L! 	M!|%N! 	O!|%P! 	Q!|%R! 	__S!|%T! 	11U!|%V! 	++W!|%X! 		EUZUjnnnY!|%Z! 	IIIII[!|%\! 	GGG]!|%^! 	//_!|%`! 	a!|%b! 	//c!|%d! 	..e!|% |%f! 	==777++..ooddd //FFFw!|% |%C~! 	(B  v/00 YFFF 	GF4556 JYGF<":<<==EFD#%%G		  1 JJ1:$AJ%AJ%
 :  ,, 	 jZ!2!23GLL$&$(>RV@VW    	 	D64..D~~ $c//d'6I6IT
	
 JJtJr3   
dispatcherc                       fd}|S )a  Wraps a given function with ``__torch_function__`` -related functionality.

    Parameters
    ----------
    dispatcher: Callable
        A callable that returns an iterable of Tensor-likes passed into the function.

    Note
    ----
    This decorator may reduce the performance of your code. Generally, it's enough to express
    your code as a series of functions that, themselves, support __torch_function__. If you
    find yourself in the rare situation where this is not the case, e.g. if you're wrapping a
    low-level library and you also need it to work for Tensor-likes, then this function is available.

    Examples
    --------
    >>> def dispatcher(a):  # Must have the same signature as func
    ...     return (a,)
    >>> @torch.overrides.wrap_torch_function(dispatcher)
    >>> def func(a):  # This will make func dispatchable by __torch_function__
    ...     return a + 0
    c                 N     t          j                    fd            S )Nc                  d     | i |}t          |          rt          |g| R i |S  | i |S r  )r   r   )r&   r'   relevant_argsr
  r!   wrappeds      r0   r
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped2  s]    &J777M!-00 V,WmUdUUUfUUU4((((r3   )	functoolsr   )r!   r
  r
  s   `@r0   rb  z"wrap_torch_function.<locals>.inner1  sD    				) 	) 	) 	) 	) 	) 
		) r3   r<  )r
  rb  s   ` r0   r   r     s#    0	 	 	 	 	 Lr3   r
  get_type_fnc                    |t           }t          j                                        sg S t	                      }g }| D ]} ||          }||vrt          |d          r|j        t          j        j        k    ru|rm|                    |           t          |          }t          |          D ]"\  }}t          | ||                    r|} n#|                    ||           |h}|g}|S )a
  Returns a list of arguments on which to call __torch_function__.

    Checks arguments in relevant_args for __torch_function__ implementations,
    storing references to the arguments and their types in overloaded_args and
    overloaded_types in order of calling precedence. Only distinct types are
    considered. If a type is a subclass of another type it will have higher
    precedence, otherwise the precedence order is the same as the order of
    arguments in relevant_args, that is, from left-to-right in the argument list.

    The precedence-determining algorithm implemented in this function is
    described in `NEP-0018`_.

    See torch::append_overloaded_arg for the equivalent function in the C++
    implementation.

    Parameters
    ----------
    relevant_args : iterable of array-like
        Iterable of array-like arguments to check for __torch_function__
        methods.

    get_type_fn : callable, optional
        Function to call on each argument in relevant_args to get its type.

    Returns
    -------
    overloaded_args : list
        Arguments from relevant_args on which to call __torch_function__
        methods, in the order in which they should be called.

    .. _NEP-0018:
       https://numpy.org/neps/nep-0018-array-function-protocol.html
    Nr   )ri
  r    _C_is_torch_function_enabledsetr
  r   _disabled_torch_function_implr  r
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer1  iold_args	            r0   _get_overloaded_argsr
  ?  s2   J  8..00 	"%%%!#O ( (;s## ,,,"677 -+ux/UUU   ( $$X... O,,"+O"<"<  JAw!(KK,@,@AA  !  &&uc2222$,: #&%r3   
public_apic                    t          |          }t          t          t          |                    }t	                      rIt                      5 }|                    | |||          }ddd           n# 1 swxY w Y   |t          ur|S |D ]j}|j        }	t          |	d          r6|	j	        |u r-|	t          j        j        urt          j        dt                      |	| |||          }|t          ur|c S k| j         d| j         }
d|
 dd |D              }t	                      r|dt%                       z  }t'          |          )	a=  Implement a function with checks for ``__torch_function__`` overrides.

    See torch::autograd::handle_torch_function for the equivalent of this
    function in the C++ implementation.

    Arguments
    ---------
    public_api : function
        Function exposed by the public torch API originally called like
        ``public_api(*args, **kwargs)`` on which arguments are now being
        checked.
    relevant_args : iterable
        Iterable of arguments to check for __torch_function__ methods.
    args : tuple
        Arbitrary positional arguments originally passed into ``public_api``.
    kwargs : tuple
        Arbitrary keyword arguments originally passed into ``public_api``.

    Returns
    -------
    object
        Result from calling ``implementation`` or an ``__torch_function__``
        method, as appropriate.

    Raises
    ------
    TypeError : if no implementation is found.

    Example
    -------
    >>> def func(a):
    ...     if has_torch_function_unary(a):
    ...         return handle_torch_function(func, (a,), a)
    ...     return a + 0
    N__self__zDefining your `__torch_function__ as a plain method is deprecated and will be an error in future, please define it as a classmethod..zno implementation found for 'z.' on types that implement __torch_function__: c                 ,    g | ]}t          |          S r<  )ri
  ).0r
  s     r0   
<listcomp>z)handle_torch_function.<locals>.<listcomp>  s    EEEcS		EEEr3   z nor in mode )r
  tuplemapri
  r   _pop_mode_temporarilyr   NotImplementedr
  r
  r    r
  r
  r,   warnDeprecationWarning
__module__r
  _get_current_function_mode	TypeError)r
  r
  r&   r'   r
  typesr  resultoverloaded_argtorch_func_method	func_namer  s               r0   r   r     s   T +=99O#dO,,--E '((  #$$ 	N,,ZfMMF	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N''M *   +=%z22		!*n<<!)OOOMQ"   #":udFCC''MMM ( (@@:+>@@I	H	 	H 	HEE_EEE	H 	H  '(( >=9;;===
C..s   A33A7:A7a  Check for __torch_function__ implementations in the elements of an iterable
    or if a __torch_function__ mode is enabled.  Considers exact ``Tensor`` s
    and ``Parameter`` s non-dispatchable.  Use this to guard a call to
    :func:`handle_torch_function`; don't use it to test if something
    is Tensor-like, use :func:`is_tensor_like` instead.
    Arguments
    ---------
    relevant_args : iterable
        Iterable or arguments to check for __torch_function__ methods.
    Returns
    -------
    bool
        True if any of the elements of relevant_args have __torch_function__
        implementations, False otherwise.
    See Also
    ________
    torch.is_tensor_like
        Checks if something is a Tensor-like, including an exact ``Tensor``.
    zSpecial case of `has_torch_function` for single inputs.
    Instead of:
      `has_torch_function((t,))`
    call:
      `has_torch_function_unary(t)`
    which skips unnecessary packing and unpacking work.
    a'  Special case of `has_torch_function` that skips tuple creation.

    This uses the METH_FASTCALL protocol introduced in Python 3.7

    Instead of:
      `has_torch_function((a, b))`
    call:
      `has_torch_function_variadic(a, b)`
    which skips unnecessary packing and unpacking work.
    c                  V   t          j        t                    } i }dt          t          j        fdt          j        t          j        j        fdt          j        j        t          t          j        j                  fdt          j        j        t          t          j        j                  fdt          j	        t          t          j	                  fdt          j
        t          t          j
                  fdt          j        t          t          j                  fdt          j        t          t          j                  fg}|D ]c\  }}}|D ]X}d	}|t          j	        urk|                    d
          r)|                    d          rd}no|                    d          rd}nW|d                                         sd}n:|dk    r}n2t!          ||          }t!          t"          |d           |k    r|dk    rt!          ||          }|t          j	        u rt!          t"          |d           |k    rt%          |t&          j                  rt%          |t*          j                  r!t/          |          st1          |d          r| d| d||j        <   | d| d||j        <   |rd|j        t7                      v r<d}	|j        t9                      vs#J |	                    ||j                              | |                             |j                   t/          |          s| d| ||<   |r|t7                      v r7d}	|t9                      vs#J |	                    ||j                              =| |                             |           Ze| |fS )Nr    ztorch.functionalztorch.nn.functionalztorch.nn.initztorch.Tensorztorch.linalgz	torch.fftztorch.specialFr  r  Tr   
unique_dim__weakref__r'  r
  z.__get__z.__set__zk{}.{} is in the tuple returned by torch._overrides.get_ignored_functions but still has an explicit override) collectionsdefaultdictlistr    __all__r   r   dirr   r5   r  r   r	  r
  endswithislowerr
  object
isinstancer
  
ModuleType
__future___Featurer  r
  r'  __set__r   r   formatr
  r  )
overridable_funcsr1  tested_namespacesnamespace_str	namespacens_funcsr  r)   r!   r  s
             r0   _get_overridable_functionsr%    s    $/55E	%'	U-u/?/GH	 3S9L5M5MN	%(-UX]););<	s5<'8'89	s5<'8'89	eiUY0	%-U]););<	 /@ C6 C6*y(! B	6 B	6IF,,''-- 	))#.. !FF'',, !FF"1--// !FF,.. / y)4469d33t;;--9i00DEL((WVY-M-MQU-U-U$ 011 $
 344 D>> gdI&>&> )6&L&L&L&L&Ldl#)6&L&L&L&L&Ldl# <#8#:#:::=   </D/F/FFFF

!4=I IFFF %d+224<@@@D>> *88Y88E$K  ,....9  #8#:#::::CJJt}= =::: i(//5555EB	6F e##r3   c                  *    t                      d         S )a  List functions that are overridable via __torch_function__

    Returns
    -------
    Dict[Any, List[Callable]]
        A dictionary that maps namespaces that contain overridable functions
        to functions in that namespace that can be overridden.
    r   )r%  r<  r3   r0   r   r   m  s     &''**r3   c                     t          | t          j        j        t          j        j        f          rt          |           S t                      d                             |           S )a&  Get a human readable string name for a function passed to
    __torch_function__

    Arguments
    ---------
    f : Callable
        Function to resolve the name of.

    Returns
    -------
    str
        Name of the function; if eval'ed it should give back the input
        function.
    rZ  )r  r    _ops
OpOverloadOpOverloadPacketstrr%  get)fs    r0   r   r   z  sO      !ej+UZ-HIJJ 1vv%''*..q111r3   c                  `    t                      } t          | t          j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r
  r    r5   )r   methodss     r0   _get_tensor_methodsr0    s*     233#EL122GNr3   c                 8    | t                      v p
| j        dk    S )aw  
    Returns True if the function passed in is a handler for a
    method or property belonging to ``torch.Tensor``, as passed
    into ``__torch_function__``.

    .. note::
       For properties, their ``__get__`` method must be passed in.

    This may be needed, in particular, for the following reasons:

    1. Methods/properties sometimes don't contain a `__module__` slot.
    2. They require that the first passed-in argument is an instance
       of ``torch.Tensor``.

    Examples
    --------
    >>> is_tensor_method_or_property(torch.Tensor.add)
    True
    >>> is_tensor_method_or_property(torch.add)
    False
    r'  )r0  r
  )r!   s    r0   r   r     s"    . &(((FDMY,FFr3   c                 X    t          |           t          j        u pt          | d          S )a9  
    Returns ``True`` if the passed-in input is a Tensor-like.

    Currently, this occurs whenever there's a ``__torch_function__``
    attribute on the type of the input.

    Examples
    --------
    A subclass of tensor is generally a Tensor-like.

    >>> class SubTensor(torch.Tensor): ...
    >>> is_tensor_like(SubTensor([0]))
    True

    Built-in or user types aren't usually Tensor-like.

    >>> is_tensor_like(6)
    False
    >>> is_tensor_like(None)
    False
    >>> class NotATensor: ...
    >>> is_tensor_like(NotATensor())
    False

    But, they can be made Tensor-like by implementing __torch_function__.

    >>> class TensorLike:
    ...     @classmethod
    ...     def __torch_function__(cls, func, types, args, kwargs):
    ...         return -1
    >>> is_tensor_like(TensorLike())
    True
    r   )ri
  r    r5   r
  )inps    r0   r   r     s(    D 99$J5I(J(JJr3   c                   P    e Zd ZU dZd ed<   ddZddZd Zd	 Ze	d
             Z
dS )TorchFunctionModea  
    A ``TorchFunctionMode`` allows you to override the meaning of all
    ``__torch_function__`` overridable functions within a dynamic scope,
    without having to actually create a tensor subclass or manually
    monkey-patch functions in the PyTorch API.  Some common situations
    where you should use a mode:

        * You want to override the meaning of factory functions, or other
          functions that do not otherwise take a tensor as an argument
          (these cannot be overridden with tensor subclasses).

        * You want to override the behavior of all functions without needing
          to wrap your inputs in tensor subclasses; e.g., if you are just
          interested in logging intermediate computations.

        * You want to control the order of execution of various tensor
          subclasses explicitly, rather than implicitly via the return of
          ``NotImplemented``.

    Independent subclasses of :class:`TorchFunctionMode` are compositional:
    modes can be pushed onto a stack using ``with MyMode():``.
    When you call functions in the PyTorch API inside your
    ``__torch_function__`` implementation, by default, they will forward on to
    the next mode on the mode stack.  If you want recursively call back into
    your current ``__torch_function__`` implementation, either explicitly
    invoke ``self.__torch_function__(...)``, or use the context manager
    ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
    API self-referential (beware of infinite loops, in this case!)
    rb  r$   Nc                     d S r  r<  r^  s    r0   r   zTorchFunctionMode.__init__  s    r3   r<  c                     t           r  )NotImplementedErrorr  r!   r
  r&   r'   s        r0   r   z$TorchFunctionMode.__torch_function__  s    !!r3   c                 $    t          |            | S r  )
_push_moder^  s    r0   	__enter__zTorchFunctionMode.__enter__  s    4r3   c                 "    t                       d S r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r0   __exit__zTorchFunctionMode.__exit__  s    r3   c                 >    t          j        d            | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`)r,   r  )clsr&   r'   instances       r0   pushzTorchFunctionMode.push  s4    ^	
 	
 	
 3'''r3   )r$   Nr<  N)r
  r  __qualname____doc____annotations__r   r   r<  rB  classmethodrF  r<  r3   r0   r5  r5    s          <    " " " "       [  r3   r5  c                  R    t                      } | dk    rt          | dz
            nd S )Nr   rZ  )r   r   	stack_lens    r0   r  r    s-    )++I4=MM!)a-000tKr3   c                  P    t                      } d t          |           D             S )Nc                 ,    g | ]}t          |          S r<  )r   )r
  r
  s     r0   r   z4_get_current_function_mode_stack.<locals>.<listcomp>  s!    @@@!"1%%@@@r3   )r   r   rM  s    r0    _get_current_function_mode_stackrQ    s)    )++I@@uY/?/?@@@@r3   c                 $    t          |            d S r  )r   )r  s    r0   r;  r;    s    !$'''''r3   c                  "    t                      } | S r  )r   olds    r0   r>  r>    s    
#
%
%CJr3   c               #   v   K   t                      } 	 | V  t          |            d S # t          |            w xY wr  )r>  r;  rT  s    r0   r  r  !  s<      
++C			3
3s   ' 8c                       e Zd ZddZdS )BaseTorchFunctionModer<  Nc                     |i } ||i |S r  r<  r9  s        r0   r   z(BaseTorchFunctionMode.__torch_function__+  s!    >FtT$V$$$r3   rG  )r
  r  rH  r   r<  r3   r0   rX  rX  *  s(        % % % % % %r3   rX  c               #   <  K   t           j                                        } 	 t           j                            t           j        j        j                   d V  t           j                            |            d S # t           j                            |            w xY wr  )r    r
  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r0   _enable_torch_functionr`  1  s}      2244I6**58+G+OPPP**955555**95555s   7A: :!Bc               #      K   t           j                                        5  	 d V  n# w xY w	 d d d            d S # 1 swxY w Y   d S r  )r    r
  _RestorePythonTLSSnapshotr<  r3   r0   r   r   ;  s       
	+	+	-	-  	EEEDDDDD	                 s   ;(;*;??)r   r    r  )ArI  r  r  
contextlibr
  r
  r,   collections.abcr   r   typingr   r   r   r   typing_extensionsr	   r    torch._Cr
   r   r   r   r   r   r   r   r   r  r   r   r+  r2   cacher
  r   r7  dictr   r   ri
  r  r
  r   r   r   r   r  r%  r   r   r0  r|
  r   r   r5  r  rQ  r;  r>  contextmanagerr  rX  r`  r   r<  r3   r0   <module>rk     sr   ,                   $ $ $ $ $ $       3 3 3 3 3 3 3 3 3 3 3 3 ' ' ' ' ' ' 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
   Yt__WT]]
 1   
2r6
     b"f	       F Ts8} T T T  Tn c(m    2 |tHh$67 | | |  |~##H # # # #P 48K KC=K(C5$;/0K 
#YK K K K\UUC=U
 	U U U Up ![  . ';	 	  *k 	   S$Ed8n	tHcM22% S$ S$ S$ S$l 	+4T(^(;#< 	+ 	+ 	+ 	+ 2 2 2( S]     Gx GD G G G G2"K "K "KJ5 5 5 5 5 5 5 5pL L L
A A A
( ( (  
   % % % % %- % % % 6 6 6     r3   