
    {Yh                     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     a/var/www/tools.fuzzalab.pt/emblema-extractor/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        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_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@@@ 	@ 	%	@
 	 @ 	 @ 	@ 	@ 	@ 	@ 	
@ 	
@ 	
@ 	@ 	
@  	!@" 	%#@$ 	%@& 	'@( 	)@* 	*+@, 	 -@. 	*/@0 	1@2 	*3@4 	5@6 	7@8 	 9@: 	!;@< 	 =@> 	%?@@ 	A@B 	
C@D 	E@F 	G@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 	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 	(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 	
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 	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 	m@n 	o@p 	q@r 	,s@t 	.u@v 	$w@x 	-y@z 	,{@| 	.}@~ 	4@@ 	7A@B 	1C@D 	#E@F 	'G@H 	 I@J 	+K@L 	*M@P 	$Q@T 	U@V 	W@X 	Y@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 	"y@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 	(Y@Z 	'[@\ 	(]@^ 	%_@` 	!a@b 	2c@d 	*e@f 	;g@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 	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 	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 	M@N 	O@P 	Q@R 	S@T 	U@V 	W@X 	Y@Z 	[@\ 	"]@^ 	_@` 	a@b 	 c@d 	$e@f 	%g@h 	'i@j 	*2k@l 	m@n 	o@p 	q@r 	s@t 	u@v 	w@x 	y@z 	 {@| 	}@~ 	@ @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_gradr1  s    r0   get_default_nowrap_functionsr6  }  s-    $ \F r3   c                     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        d3d`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        d3drt           jg        ddsi t           jh        ddtt           ji        ddut           jj        dv t           jk        d3dw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        d3d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        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        d0dt           j        d1dt           j        d2dt           j        d3dt           jP        j        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 t           j        d t           j        d
d	t           j        dd
t           j        d t           j        d4dt           j        d t           j        d'dt           j        d t           j        d t           j        ddt           j        ddi 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" t           j        d5d#i t           j        d6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        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d0t           j        d7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%        d9 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-        dA t           j.        ddBt           j/        ddCt           j0        ddDt           j1        d8dEi t           j2        ddFt           j3        ddGt           j4        dH t           j5        dI t           j6        ddJt           j7        ddKt           jP        j1        d9dLt           jP        j8        d9dMt           jP        j9        d8dNt           jP        j2        d8dOt           jP        j7        ddPt           j:        dQ t           jP        j:        ddRt           jP        j;        d,dSt           jP        j<        ddTt           j=        dU t           jP        j=        dV i t           j>        d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        db t           jJ        ddct           jK        dddt           jL        ddet           jM        df t           jN        dg i t           jO        dh t           jP        di t           jQ        dj t           jR        dk t           jS        dl t           jT        ddmt           jU        d=dnt           jV        do t           jW        dp t           jX        ddqt           jY        ddrt           jZ        ddst           j[        ddtt           j\        ddut           j]        dv t           j^        dw t           j_        d>dyi t           j`        dz t           ja        d{ t           jb        d| t           jc        d?d}t           jd        d~ t           je        d@dt           jf        d t           jg        ddt           jh        ddt           ji        ddt           jj        ddt           jk        ddt           jl        jm        jn        d t           jl        jm        jo        d t           jl        jm        j        d
dt           jl        jm        jp        d
dt           jl        jm        jq        d
di t           jl        jm        jr        d
dt           jl        jm        js        d
dt           jl        jm        jt        d
dt           jl        jm        ju        ddt           jl        jm        j        dAdt           jl        jm        jv        dBdt           jl        jm        jw        dBdt           jl        jm        j.        dCdt           jl        jm        j7        ddt           jl        jm        jx        dDdt           jl        jm        j8        ddt           jl        jm        jL        ddt           jl        jm        jq        ddt           jl        jm        jy        dEdt           jl        jm        jv        ddt           jl        jm        j        dFdt           jl        jm        jz        dFdi t           jl        jm        j{        dFdt           jl        jm        j|        dFdt           jl        jm        j}        ddt           jl        jm        j        d$dt           jl        jm        j        dGdt           jl        jm        j        dAdt           jl        jm        j~        dHdt           jl        jm        j        dIdt           jl        jm        j        dIdt           jl        jm        j        dIdt           jl        jm        j        dIdt           jl        jm        j        dJdt           jl        jm        j        dKdt           jl        jm        j        dLdt           jl        jm        j        dMdt           jl        jm        j        d?dt           jl        jm        j        dNdi t           jl        jm        j        d/dt           jl        jm        j        dOdt           jl        jm        j        d0dt           jl        jm        j        dPdt           jl        jm        j        dQdt           jl        jm        j        d6dt           jl        jm        j        dRdt           jl        jm        j        d?dt           jl        jm        j        dSdt           jl        jm        j        ddt           jl        jm        j        dTdÄt           jl        jm        j        dUdńt           jl        jm        j        dƄ t           jl        jm        j        d,dǄt           jl        jm        j        d,dȄt           jl        jm        j        d,dɄt           jl        jm        j3        ddʄi t           jl        jm        jA        d;d˄t           jl        jm        jD        d;d̄t           jl        jm        jB        d;d̈́t           jl        jm        j        d;d΄t           jl        jm        jC        d;dτt           jl        jm        j        d;dЄt           jl        jm        j        dVdфt           jl        jm        j        dVd҄t           jl        jm        j        dVdӄt           jl        jm        j        dRdԄt           jl        jm        j        dWdՄt           jl        jm        j        dXdքt           jl        jm        j        dYdׄt           jl        jm        j        dDd؄t           jl        jm        j        dZdلt           jl        jm        j        d[dۄt           jl        jm        j        dLd܄i t           jl        jm        j        d\dބt           jl        jm        j        d]d߄t           jl        jm        j        d^dt           jl        jm        j        d t           jl        jm        j        d
dt           jl        jm        j        d
dt           jl        jm        j        d_dt           jl        jm        j        d`dt           jl        jm        j        d
dt           jl        jm        j        d
dt           jl        jm        j        d
dt           jl        jm        j        dadt           jl        jm        j        dbdt           jl        jm        j        dcdt           jl        jm        j        dYdt           jl        jm        j        dUdt           jl        jm        j        dUdi t           jl        jm        j        dddt           jl        jm        j        d/dt           jl        jm        j        d t           jl        jm        j        d t           jl        jm        j        d
dt           jl        jm        j        dedt           jl        jm        j        ddUdd>ddt           jl        jm        j        dHdt           jl        j        j        dfdt           jl        j        j        dfdt           jl        j        j        d t           jl        j        j        dgd t           j        d
dt           j        dddt           j        d t           j        dhdt           jP        j        didi t           jP        j        djdt           jP        j        	 dkd	t           j        dld
t           j        dhdt           j        d t           j        d t           j        dmdt           j        d]dt           j        d t           j        dndt           j        d"dt           j        dodt           jP        j        dpdt           j        d t           j        d t           j        ddt           j        d i 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        d9d&t           jP        j        dqd(t           j        drd*t           j        drd+t           j        d, i t           j        d- t           j        d. t           j        d/ t           j        d0 t           j        d1 t           j        	 	 dsd5t           j        	 	 dtd8t           j        	 	 dud;t           j        d< t           j        d= t           j        dd>t           j        d&d?t           j        dt           j        ddfd@t           j        d&dAt           j        dB t           j        ddCt           j        ddDi t           jP        j        ddEt           j        dF t           j        dG t           j        ddHt           j        d
dIt           j        ddJt           j        ddKt           j        ddLt           j        dM t           j        d_dNt           j        dO t           j        ddPt           j        dQ t           j        ddRt           j        ddSt           j        dvdUt           j        ddVi t           j        ddWt           j         dX t           j        d`dYt           j        ddZt           j        d4d[t           j        dd\t           j        d] t           j        d^ t           j        d4d_t           j        dd`t           j        dwdbt           j	        dc t           j
        dd t           j        dxdet           j        dx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        do t           jP        j        dp t           j        ddqt           j        ddrt           j        ddst           jP        j        d9dtt           jP        j        d8dut           j        dydddvdwt           j        d3dxt           j        d3dyt           j        ddz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#        dz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 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dt           j9        d{dt           j:        d|dt           jP        j9        d9dt           jP        j;        ddt           j<        d t           j=        d i t           j>        j?        d t           j>        j@        d t           j>        jA        d t           j>        jB        d t           j>        jC        d t           j>        jD        ddt           j>        jE        ddt           j>        jF        ddt           j>        jG        ddt           j>        j        d t           j>        jH        d t           j>        j        d t           j>        j        d t           j>        jI        d t           j>        j        d t           j>        j        d t           j>        jJ        d i t           j>        j        d t           j>        jK        ddt           j>        jL        ddt           j>        jM        d t           j>        jN        ddt           j>        jO        ddt           j>        j        d t           j>        jP        d t           j>        jQ        d t           j>        jR        d t           j>        jS        ddt           j>        jT        ddt           j>        j!        d t           j>        jU        d t           j>        j        ddt           j>        j+        d t           j>        j,        ddi t           j>        jV        d t           j>        jW        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dȄt           j>        j]        dɄ t           j>        j        dʄ t           j>        j^        d˄ t           j>        j_        d̄ t           j>        j`        dd̈́t           j>        ja        dd΄t           j>        jb        ddτt           j>        jc        ddЄt           j>        j        dф i t           j>        j        dd҄t           j>        jd        dӄ t           j>        je        ddԄt           j>        j&        ddՄt           j>        jf        ddքt           jg        dׄ t           jh        d؄ t           ji        ddلt           jj        ddڄt           jk        ddۄt           jP        jl        d"d܄t           jP        jm        dd݄t           jn        d}dބt           jo        d3d߄t           j        d
dt           jp        d t           jq        d=di t           jr        d t           js        d t           jt        ddt           ju        ddt           jv        d~dt           jP        jw        dmdt           jx        ddt           j        dedt           jy        ddt           jz        d t           j{        ddt           j|        d3dt           j}        d t           j~        ddt           j        ddt           j        d t           j        d3di t           j        d3dt           j        d3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 t           j        d t           j        d i 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        dxdt           j        d3dt           j        d3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        d t           j        d t           j        d t           j        d3dt           j        d  t           j        d! t           j        d" | j        d# | 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        ddFdG| j        dH | j        dI | j        j        dJ i | j        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        dW | j        dX | j        dY | j        j        dZ | j        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        d,d{| j        d| | j        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        dId| j        t           j        fdi | j        t           j        fd| j        t           j        fd| j        t           j        fd| j        dTd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 | j        d d| j*        d i | j+        d
d| j,        t           j        fd| j-        t           j        fd| j.        d | j/        d | j0        d | j1        d4ddd| j2        d | j3        d | j4        t           j        fd| j5        t           j        fd| j6        ddd| j        d | j7        t           j        fd| j8        t           j        fd| j9        d | j:        d i | j;        t           j        fd| j<        d | j=        d | j        d | j>        d | j?        d | j@        d | jA        d | jB        dddd| j        d | jC        t           j        fd| jD        d | jE        d | jT        dd| jF        d
d| j        d | jG        d i | jH        dÄ | jI        dĄ | jJ        dń | jK        dƄ | j        dǄ | jL        dȄ | j        dɄ | jM        dʄ | jN        d
d˄| jO        d̄ | jP        dddd̈́| jQ        d΄ | jR        dτ | jS        dЄ | jT        dф | jU        d҄ | jV        dӄ i | jW        d4dԄ| jX        dՄ | jY        dք | jZ        dׄ | j[        d؄ | j\        dل | j]        dڄ | j^        ddۄ| j
        d܄ | j_        d݄ | j`        t           j        fdބ| ja        d߄ | j        dxd| jb        d | jc        d | jd        d
d| je        d i | jf        d | j        dd| jg        d | jh        d | ji        d | jj        d | jk        d | jp        d | jl        ddt           j        fd| jm        dddd| jn        dd| jo        d | jp        d | jq        d | jr        d | j        d | j        dTd| js        d | jt        d | ju        d | jv        d | jw        dd| jx        d t           jP        jy        ddi}t           jz        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 r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  r=  output_sizes     r0   r?  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r9  r;  )inputsrE  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  r=  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r9  r;  r=  otherr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      " r3      c                     dS r9  r;  r=  batch1batch2alphabetar>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r9  r;  r=  tensor1tensor2valuer>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r9  r;  rb  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  r=  mat1mat2r_  r^  r>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  )r=  matvecr_  r^  r>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  )r=  vec1vec2r_  r^  r>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  thetasizealign_cornerss      r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r=  dims     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   h㈵>:0yE>Fc                     dS r9  r;  )r=  rW  trolatol	equal_nans        r0   r?  z'get_testing_overrides.<locals>.<lambda>      VX r3   c                     dS r9  r;  r=  ptraininplaces       r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r=  r{  keepdimr>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  )r=  msgs     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                      dS r9  r;  tensorss    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                      dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                      dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   r   Tc                     dS r9  r;  )r=  kernel_sizer  padding	ceil_modecount_include_pads         r0   r?  z'get_testing_overrides.<locals>.<lambda>      vx r3   c                     dS r9  r;  r[  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      PR r3   c	                     dS r9  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 r9  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 r9  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 r9  r;  )r=  r  r  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  r=  r  r  r  r  r  r  counts           r0   r?  z'get_testing_overrides.<locals>.<lambda>      tv r3   c                     dS r9  r;  r  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>        AC r3   c                     dS r9  r;  r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r9  r;  )r=  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>      Z\ r3   c                     dS r9  r;  )r=  	generatorr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  input1input2r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   r  c                     dS r9  r;  r=  targetr  size_averagereduce	reduction
pos_weights          r0   r?  z'get_testing_overrides.<locals>.<lambda>      rt r3   c                     dS r9  r;  )r=  weights	minlengths      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  probr  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      " r3   c                      dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r=  rk  	out_dtyper>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                      dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  selfrw  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  )r=  
boundaries	out_int32rightr>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      [] r3   c                      dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  r{  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3          @#use_mm_for_euclid_dist_if_necessaryc                     dS r9  r;  )x1x2r  compute_modes       r0   r?  z'get_testing_overrides.<locals>.<lambda>      _a r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3         ?c                     dS r9  r;  r=  r^  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>   r  r3   )r>  c                     dS r9  r;  )r>  matricess     r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  r=  groupss     r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r9  r;  r=  upperr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  check_errorsr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r9  r;  )r  r  r  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS r9  r;  )r=  numeln_binsratio	bit_widths        r0   r?  z'get_testing_overrides.<locals>.<lambda>      WY r3   c                     dS r9  r;  r=  chunksr{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>	  rX  r3   c                     dS r9  r;  r=  minmaxr>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS r9  r;  r+  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  )r=  r,  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r-  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  r>  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  
correctionfweightsaweightss       r0   r?  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>      2 r3      c                     dS r9  r;  )r=  rwith_replacements      r0   r?  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS r9  r;  )realimags     r0   r?  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )absangs     r0   r?  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS r9  r;  )r=  ords     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  padre  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r9  r;  r=  r  r  r  r  dilationr  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>      bd r3   c                     dS r9  r;  rS  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r9  r;  rS  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c	                     dS r9  r;  )	r=  r  r  r  r  rT  
transposedoutput_addingr  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>      uw r3   c                     dS r9  r;  )r=  r  r  rP  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  r=  r  r  r  r  output_paddingr  rT  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>         A r3   c                     dS r9  r;  r^  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>!  r`  r3   c                     dS r9  r;  r^  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>"  r`  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>#  rH  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r3   c                     dS r9  r;  r  r  r  marginr  r  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>%  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>&  rK  r3   c                     dS r9  r;  )r  r  r{  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>'  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r3   c                     dS r9  r;  r=  rW  r{  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>)  rF  r3   r:  c                     dS r9  r;  rl  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>*      2 r3   c                     dS r9  r;  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r0   r?  z'get_testing_overrides.<locals>.<lambda>,  r  r3   c                     dS r9  r;  r=  r{  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>.  r  r3   c                     dS r9  r;  rx  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>/  r  r3   c                     dS r9  r;  r=  r{  r>  r_   s       r0   r?  z'get_testing_overrides.<locals>.<lambda>0  r/  r3   c                     dS r9  r;  r{  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>1  rI  r3   c                     dS r9  r;  yxr{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>2  rF  r3   c                     dS r9  r;  rx  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>3  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>4  rT  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>5       r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>6       r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>7  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS r9  r;  r=  diagonalr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>9  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>:  rF  r3   c                     dS r9  r;  )r=  offsets     r0   r?  z'get_testing_overrides.<locals>.<lambda>;  rB  r3   c                     dS r9  r;  )r=  nr{  prependappendr>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda><      TV r3   c                     dS r9  r;  r=  r  dim1dim2s       r0   r?  z'get_testing_overrides.<locals>.<lambda>=  r/  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>>  rs  r3   c                     dS r9  r;  )r=  srcr  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>?  r8  r3   c                     dS r9  r;  )r  r  rw  r  storage_offsets        r0   r?  z'get_testing_overrides.<locals>.<lambda>@  r&  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>A  rT  r3   c                     dS r9  r;  )r=  rW  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>B  rP  r3   c                     dS r9  r;  r=  rW  rounding_moder>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>C      b r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>D  rs  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>E  rX  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>F  rF  r3   c                     dS r9  r;  r=  rk  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>G  r  r3   c                     dS r9  r;  )rj  rk  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>H      r r3   c                     dS r9  r;  r=  indices_or_sectionss     r0   r?  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS r9  r;  r3  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>J  rB  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>K  rX  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>L  r  r3   Lc                     dS r9  r;  r=  UPLOr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>M  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>N  r  r3   c                     dS r9  r;  )equationoperandss     r0   r?  z'get_testing_overrides.<locals>.<lambda>O  rX  r3   c                     dS r9  r;  r=  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r0   r?  z'get_testing_overrides.<locals>.<lambda>Q      z| r3   c
                     dS r9  r;  )
r=  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r0   r?  z'get_testing_overrides.<locals>.<lambda>T  s      hj r3   c                     dS r9  r;  r=  r_   rm   r^   requires_grads        r0   r?  z'get_testing_overrides.<locals>.<lambda>V      ce r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>W       r3   c                     dS r9  r;  r=  rW  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>X  rC  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>Z  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>[  rP  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>]  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>^  rR  r3   c                     dS r9  r;  )r=  scale
zero_pointaxis	quant_min	quant_maxs         r0   r?  z'get_testing_overrides.<locals>.<lambda>_      mo r3   c                     dS r9  r;  )r=  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>`      fh r3   c                     dS r9  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>b  s      AC r3   c                     dS r9  r;  r=  packed_weightr  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>d  r   r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>e      \^ r3   c                     dS r9  r;  r=  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>f      {} r3   c                     dS r9  r;  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>h      ^` r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>j  rQ  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>k  r  r3   c                     dS r9  r;  )r=  abs      r0   r?  z'get_testing_overrides.<locals>.<lambda>l  r/  r3   c                     dS r9  r;  r=  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>m  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>n  r  r3   c                     dS r9  r;  r=  r  r{  norms       r0   r?  z'get_testing_overrides.<locals>.<lambda>o  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>p  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>q  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>r  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>s  r  r3   r  r:  c                     dS r9  r;  r=  sr{  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>t  ro  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>u  rs  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>v  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>w  rn  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>x  rn  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>y  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>z  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>{  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>|  r?  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>}  ro  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>~  ro  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  )r=  	start_dimend_dims      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  r=  dimss     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  exponentr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  )r=  
fill_valuer>  r_   rm   r^   r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  s      BD r3   c                     dS r9  r;  )r=  r  	dep_tokens      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r{  indexr>  sparse_grads        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  rr  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r9  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  spacingr{  
edge_orders       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   c                     dS r9  r;  r=  gridinterpolation_modepadding_moderx  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      ac r3   c                     dS r9  r;  r?  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      df r3   c                     dS r9  r;  r?  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rE  r3   c                     dS r9  r;  )r=  
num_groupsr  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      km r3   c	                     dS r9  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 r9  r;  r=  rL  w_ihw_hhb_ihb_hhs         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3         ?c                     dS r9  r;  r=  lambds     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  valuesr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  rg  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      xz r3   d   c                     dS r9  r;  )r=  binsr,  r-  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  rg  r,  r-  r  densityr>  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  )r=  rg  r   r  ri  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r&  r3   c                     dS r9  r;  r=  taus     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )rj  rk  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r3  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r=  r{  r/  sources       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  rv  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  indicesra  
accumulates       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  )r=  r{  r/  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  )r=  r{  r/  re  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  )r=  r{  r/  rw  r  include_inputs         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rY   s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )eteassume_uniqueinverts       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c	                     dS r9  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 r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  rW  rtolr  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      UW r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c
                     dS r9  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 r9  r;  r=  r  r  r  r  
log_targets         r0   r?  z'get_testing_overrides.<locals>.<lambda>  s    pr r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS r9  r;  )r=  kr{  r  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   c                     dS r9  r;  )r=  	hermitianr  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  )r=  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  )LDpivotsBr  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>      QS r3   c                     dS r9  r;  )r=  normalized_shaper  r  espr  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  endr  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  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 r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r=  r{  r_   s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  )r=  namesr  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c	                     dS r9  r;  )	databatch_sizesrL  rM  rN  rO  rP  r  rQ  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r9  r;  rU  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )Apivot	get_infosr>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS r9  r;  )r  r*  r+  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  rf  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  )r=  maskre  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  rw  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )LUr  r  leftadjointr>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      Y[ r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS r9  r;  )r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r=  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  )r=  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS r9  r;  r3  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>	  r  r3   c                     dS r9  r;  r=  r  r  r  rT  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>
      jl r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  r  r  rT  return_indicesr  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  )r=  r{  r  r_   r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                      dS r9  r;  )r  r'   s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r  r  r  exponential_average_factorepsilons           r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c	                     dS r9  r;  	r=  r  r  r  r  rT  r  	benchmarkdeterministics	            r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c	                     dS r9  r;  )	r=  r  zr^  r  r  r  rT  r  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rS  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r9  r;  )
r=  r  r  r  r_  r  rT  r  r  r  s
             r0   r?  z'get_testing_overrides.<locals>.<lambda>  r[  r3   c	                     dS r9  r;  r  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>"      eg r3   c                     dS r9  r;  )r=  r  weight_stride0rL  cxr  hidden_sizerO  rR  rP  r  rQ  r  dropout_states                 r0   r?  z'get_testing_overrides.<locals>.<lambda>%  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>'  r/  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>(  rn  r3   c                     dS r9  r;  r=  rw  destinations      r0   r?  z'get_testing_overrides.<locals>.<lambda>)  r  r3   c                     dS r9  r;  r)  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>*  rQ  r3   c                     dS r9  r;  )r=  
descendingr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>+  rI  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>,  rX  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>-  r  r3   c                     dS r9  r;  )r=  num_samplesreplacementr>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>.      SU r3   c                     dS r9  r;  )r=  rn  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>/  rT  r3   c                     dS r9  r;  r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>0  r  r3   c                     dS r9  r;  )r=  r{  startr  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>1  r  r3           c                     dS r9  r;  )r=  nanposinfneginfr>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>2  r  r3   c                     dS r9  r;  )r=  r  r  r  r  r  r  r  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>3  rS  r3   c                     dS r9  r;  )r=  r  r  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>4      ]_ r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>5  r  r3   c                     dS r9  r;  r=  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>6  rE  r3   c                     dS r9  r;  )r=  r  r  NCHxWgroupr  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>7  r  r3   c                     dS r9  r;  )r=  r  r{  r  r_   s        r0   r?  z'get_testing_overrides.<locals>.<lambda>8  r3  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>9  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>:  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>;  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>=  rB  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r9  r;  rD  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>?  r   r3   c                     dS r9  r;  rD  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>@  r   r3   c                     dS r9  r;  r=  rE  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>A  rU  r3   c                     dS r9  r;  rS  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>B      oq r3   c                     dS r9  r;  rS  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>C  rU  r3   c                     dS r9  r;  rS  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>D  rU  r3   c                     dS r9  r;  rS  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>E  rU  r3   c                     dS r9  r;  rS  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>F  rU  r3   c                     dS r9  r;  ru  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>G  r  r3   c                     dS r9  r;  r=  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>H  r	  r3   c                     dS r9  r;  r=  r  r  r  r  r  divisor_overrides          r0   r?  z'get_testing_overrides.<locals>.<lambda>J        @B r3   c                     dS r9  r;  r^  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>M  r`  r3   皙?c                     dS r9  r;  )r=  r  r  r  r  r  r  r  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>P  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>R  r  r3   c                     dS r9  r;  r=  r  r  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>T  rC  r3   c                     dS r9  r;  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>Y  rf  r3   c                     dS r9  r;  rf  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>[      gi r3   c                     dS r9  r;  )r=  r  r  r  ignore_indexr  r  label_smoothings           r0   r?  z'get_testing_overrides.<locals>.<lambda>^        JL r3   c                     dS r9  r;  rp  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>a  r  r3   c                     dS r9  r;  r\  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>c      XZ r3   c                     dS r9  r;  r\  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>d  r  r3   c                     dS r9  r;  r\  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>e  r  r3   c                     dS r9  r;  r\  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>f  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>g  rs  r3   c                     dS r9  r;  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>i  r  r3   c                     dS r9  r;  )r=  r  r  r  r  r  r  r   r  include_last_offsetr  s              r0   r?  z'get_testing_overrides.<locals>.<lambda>l  s      HJ r3   c                     dS r9  r;  r\  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>n  rj  r3   c                     dS r9  r;  )r=  rE  r  rT  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>o  rI  r3   c                     dS r9  r;  r=  r  rE  output_ratior  _random_sampless         r0   r?  z'get_testing_overrides.<locals>.<lambda>q  rd  r3   c                     dS r9  r;  r}  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>t  rd  r3   c                     dS r9  r;  r}  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>w  rd  r3   c                     dS r9  r;  r}  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>z  rd  r3   ư>c                     dS r9  r;  )r=  r  varr   r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>|  r  r3   nonec                     dS r9  r;  )r=  approximates     r0   r?  z'get_testing_overrides.<locals>.<lambda>}  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>~  r  r3   bilinearr   c                     dS r9  r;  )r=  r@  r  rB  rx  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS r9  r;  )r=  rH  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   绽|=c                     dS r9  r;  )logitsrm  hardr  r{  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r^  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3         c                     dS r9  r;  )r=  min_valmax_valr  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rc  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>      `b r3   c                     dS r9  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 r9  r;  )r=  rw  scale_factorr  rx  recompute_scale_factor	antialiass          r0   r?  z'get_testing_overrides.<locals>.<lambda>  s      KM r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  r=  r  r  r  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rC  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   {Gz?c                     dS r9  r;  )r=  negative_sloper  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   -C6?      ?c                     dS r9  r;  )r=  rw  r^  r_  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3      c                     dS r9  r;  r=  r{  _stacklevelr_   s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  rf  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  rj  r3   c                     dS r9  r;  r=  r  r  r  rT  r  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r
  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r
  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r
  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r
  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  rz  r  r  r  rE  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )querykeyre  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 r9  r;  )r=  r  r  rg  r  r  r  r  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rf  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  )r=  r  r  r  rm  r  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   -q=c                     dS r9  r;  )r=  r  r{  r  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )rY   num_classess     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   r   c                     dS r9  r;  )r=  rP  r  re  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r	  r3   c                     dS r9  r;  )r=  r  	log_inputr   r  r  r  r  s           r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  r=  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3         ?UUUUUU?c                     dS r9  r;  r=  lowerr  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  s    wy r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  )r  r  re  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r  r  r_  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  )r=  r  r  deltar  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  s    hj r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rr  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rr  r3      c                     dS r9  r;  )r=  r_  	thresholds      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  r^  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  r  re  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r9  r;  
anchorpositivenegativerg  r  r  swapr  r  r  s
             r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   )distance_functionrg  r  r  c                    dS r9  r;  )r  r  r  r	  rg  r  r  s          r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  rT  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )rY   r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  )rY   r  stdr  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )rY   vals     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   fan_in
leaky_reluc                     dS r9  r;  )rY   r  r  nonlinearityr  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  as_tuples     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   )r&  c                    dS r9  r;  )r=  rw  r&  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   froc                     dS r9  r;  r=  r  r{  r  r>  r_   s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  rJ  r{  r  r>  r_   s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  s     13 r3   c                     dS r9  r;  )vpowr{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c                     dS r9  r;  rl  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  input3r  	transposes        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  r{  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  qr  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r6  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   V瞯<c                     dS r9  r;  )r=  rconds     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r1  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  )r=  upscale_factors     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  )r=  downscale_factors     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  )r=  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r   r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r&  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rU  r3   c                     dS r9  r;  r   s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r_   s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  )r=  r/  rw  r{  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>	  rC  r3   c                     dS r9  r;  )r=  somer>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>
  r  r3   reducedc                     dS r9  r;  )r=  r  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   linearc                     dS r9  r;  r=  r-  r{  r  interpolationr>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rE  r3   c                     dS r9  r;  rK  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  rj  r3   c                     dS r9  r;  )r=  scaleszero_pointsr  r_   s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r_   s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r_   reduce_ranges      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r  r  r  output_scaleoutput_zero_points           r0   r?  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS r9  r;  r=  rL  rV  rW  rX  rY  	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 r9  r;  rX  s                 r0   r?  z'get_testing_overrides.<locals>.<lambda>  ra  r3   r;  r   rY  c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  s
     " r3   r   r   rY  rY  c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  s     !# r3   r   r   r   rY  rY  rY  c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>$  s     !#	 r3   c                     dS r9  r;  rX  s                 r0   r?  z'get_testing_overrides.<locals>.<lambda>+  ra  r3   c                     dS r9  r;  rX  s                 r0   r?  z'get_testing_overrides.<locals>.<lambda>.  ra  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>0  rT  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>1  rU  r3   c                     dS r9  r;  )r=  highr_   rm   r^   r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>2  r  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>3  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>4  r:  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>5  rK  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>6  r  r3   c                     dS r9  r;  rl  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>7  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>8  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>9  rR  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>:  rX  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>;  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda><  r  r3   c                     dS r9  r;  )r=  r  r{  maxnormr>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>=  rI  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r9  r;  )r=  shapes     r0   r?  z'get_testing_overrides.<locals>.<lambda>?  rK  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>@  r`  r3   c	                     dS r9  r;  rK  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS r9  r;  rU  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>B  r  r3   c	                     dS r9  r;  rK  s	            r0   r?  z'get_testing_overrides.<locals>.<lambda>C  r  r3   c                     dS r9  r;  rU  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>D  r  r3   c                     dS r9  r;  )r=  shiftsr  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>E  r  r3   r   rY  c                     dS r9  r;  )r=  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>F  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>G  rR  r3   c                     dS r9  r;  r3  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>H  r  r3   c                     dS r9  r;  )r  r  compressed_indices_dtypes      r0   r?  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>K  rR  r3   c                     dS r9  r;  )r=  rW  r^  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>L  rX  r3   c                     dS r9  r;  ri  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>M  r  r3   c                     dS r9  r;  r=  r{  r/  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>N  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>O  r  r3   c                     dS r9  r;  )r=  r{  r/  r  r  include_selfs         r0   r?  z'get_testing_overrides.<locals>.<lambda>P  rr  r3   c                     dS r9  r;  )sorted_sequencer=  r  r  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>Q  r  r3   r-  c                     dS r9  r;  )r  r  lengthsrz  r  r  unsafes          r0   r?  z'get_testing_overrides.<locals>.<lambda>R  r  r3   c                     dS r9  r;  )r=  r{  r/  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>S  rB  r3   c                     dS r9  r;  )r=  r  r{  r/  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>T  r  r3   c                     dS r9  r;  r=  r  r{  r8  r  steps         r0   r?  z'get_testing_overrides.<locals>.<lambda>U  r  r3   c                     dS r9  r;  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>V  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>X  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>Y  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>Z  rT  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>]  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>^  rK  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>_  rN  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>`  rK  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>a  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>b  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>c  r  r3   c                     dS r9  r;  )r  r  r  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>d  rF  r3   c                     dS r9  r;  )r  r  r  r  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>e  r  r3   )stabler>  c                    dS r9  r;  )r=  r{  r-  r  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>f  r&  r3   c                     dS r9  r;  rY   split_size_or_sectionsr{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>g  rn  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>h  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>i  rK  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>j  rP  r3   c                     dS r9  r;  rx  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>k  r  r3   c                     dS r9  r;  ri  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>l  r8  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>m  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>o  rB  r3   reflectc                     dS r9  r;  )r=  r  r  r  r  r  pad_moder  r  r  align_to_windows              r0   r?  z'get_testing_overrides.<locals>.<lambda>q  s      ~@ r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>s  rX  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>t  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>v  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>w  rN  r3   c                     dS r9  r;  r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>x  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>y  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>z  rN  r3   c                     dS r9  r;  )r  r  cs      r0   r?  z'get_testing_overrides.<locals>.<lambda>{  r  r3   c                     dS r9  r;  )r&   s    r0   r?  z'get_testing_overrides.<locals>.<lambda>|  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>}  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>~  rH  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  )r=  rF  
compute_uvr>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3      c                     dS r9  r;  )r=  r-  r  Ms       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  )r=  full_matricesr>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r=  dim0r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  )r=  axis0axis1s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r6  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  rV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rW  r>  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  s    r r3   c                     dS r9  r;  )r=  r/  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  rz  r{  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  )r  inds     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  r>  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r{  r-  r>  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  rM  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r~  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r~  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r(  unitriangulars        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r  r  r  r1  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r	  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS r9  r;  r  s
             r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r<  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  )r=  r{  sizesr  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  sortedreturn_inversereturn_countsr{  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r=  r=  r>  r{  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   c                     dS r9  r;  )rz  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r(  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rx  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rE  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  rz  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r3  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  )	conditionr  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  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 r9  r;  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  levels     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )primaltangentrV  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  r  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  )r  rw  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   )implicitc                    dS r9  r;  )r  rw  rb  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r{  r8  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  rw  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  )r  r{  r/  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r{  r8  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r   r3   c                     dS r9  r;  )r  
split_sizer{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  split_sizesr{  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r+  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c                     dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r+  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>	  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r+  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  r_   s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r  	dimensionrw  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  rW  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>   rB  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>!  rK  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>#  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>$  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>%  r  r3   c                     dS r9  r;  r}  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>&  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>'  rH  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>(  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>)  rN  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>*  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>+  rK  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>,  rR  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>-  rR  r3   c                     dS r9  r;  )r  arrays     r0   r?  z'get_testing_overrides.<locals>.<lambda>.  r  r3   c                     dS r9  r;  )r  idxs     r0   r?  z'get_testing_overrides.<locals>.<lambda>/  rP  r3   c                     dS r9  r;  )r  memos     r0   r?  z'get_testing_overrides.<locals>.<lambda>0  rB  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>1  rN  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>2  rH  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>3  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>4  rN  r3   c                     dS r9  r;  )r  format_specs     r0   r?  z'get_testing_overrides.<locals>.<lambda>5  r  r3   c                     dS r9  r;  )r  protos     r0   r?  z'get_testing_overrides.<locals>.<lambda>6  rX  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>7  rC  r3   )tensor_contentsc                    dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>8  rI  r3   c                     dS r9  r;  )r  r  r!  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>9  rT  r3   c                     dS r9  r;  )r  ds     r0   r?  z'get_testing_overrides.<locals>.<lambda>:  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>;  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda><  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>=  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>?  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>@  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>B  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>C  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>E  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>F  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>G  rP  r3   c                     dS r9  r;  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r0   r?  z'get_testing_overrides.<locals>.<lambda>H  s    np r3   c                     dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>K  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>L  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>N  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>O  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>P  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>Q  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>R  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>S  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>T  rX  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>U  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>V  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>W  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>X  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>Y  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>Z  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>[  rX  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>\  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>]  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>^  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>_  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>`  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>a  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>b  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>c  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>d  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>e  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>f  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>h  rP  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>i  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>j  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>k  rF  r3   c                     dS r9  r;  )r  r_   non_blockingr'   s       r0   r?  z'get_testing_overrides.<locals>.<lambda>l  r   r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>m  r:  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>n  r:  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>o  rH  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>p  rH  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>q      " r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>r  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>s  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>t  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>u  r  r3   c                     dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>v  rQ  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>w  rN  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>x  rN  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>y  rR  r3   c                     dS r9  r;  )r  orderellipsis_idxs      r0   r?  z'get_testing_overrides.<locals>.<lambda>z  rQ  r3   c                     dS r9  r;  )r  callables     r0   r?  z'get_testing_overrides.<locals>.<lambda>{  rP  r3   c                     dS r9  r;  rg  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>|  r  r3   c                     dS r9  r;  rg  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>}  r  r3   c                     dS r9  r;  )r  gradientretain_graphcreate_graphrH  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>~  s    ik r3   c                     dS r9  r;  r  rf   s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   )r  c                    dS r9  r;  )r  mediansigmar  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  )r  	coalesceds     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  r  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  ambiguity_checks     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                    dS r9  r;  )r  r_  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  re  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rs  r3   c                    dS r9  r;  )r  r  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rY   s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                    dS r9  r;  )r  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r+  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  )r  rY   r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rk  r  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rW  assigns      r0   r?  z'get_testing_overrides.<locals>.<lambda>  rF  r3   c                     dS r9  r;  )r  r  r8  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rH  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c                     dS r9  r;  r+  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rz  rY   r{  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                    dS r9  r;  )r  from_tor  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS r9  r;  r  streams     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  hooks     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  rM  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  names     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  rw  r  rw  r  s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r{  r/  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  r{  r8  r  r  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  )r  r  accumulate_matchess      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  size1size2	dense_dims       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  re  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS r9  r;  )r  rj  rk  r_  r^  r>  s         r0   r?  z'get_testing_overrides.<locals>.<lambda>  r8  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rC  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  )r  repss     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r_   r  copyrf   s        r0   r?  z'get_testing_overrides.<locals>.<lambda>  s    ln r3   )masked_gradc                    dS r9  r;  r  r_   rt  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r?  r3   c                     dS r9  r;  rv  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r  s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  rQ  r3   c                     dS r9  r;  )r  rF  rG  s      r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  rK  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r:  r3   c                     dS r9  r;  rI  s     r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS r9  r;  )r  r  r   drivers       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  )r  r^   r  r'   s       r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS r9  r;  r[  s    r0   r?  z'get_testing_overrides.<locals>.<lambda>  r  r3   is______i__rbitwise_N)rY  rY  N)rY  N)r|  r}  F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)r  r  )r  F)FN)NNN)rY  NN)r;  F)NrY  r   rY  rY  )NrY  r   r   rY  rY  )r   NNr  )rY  r}  )r:  N)r   r  Fr9  )rY  r:  NNN)r   r   rY  )r   r  r:  )r;  )r  N)NNr  FF)Nr;  Fr  FNN)NNNF)FF)Nr:  N)Nr  N)r   r:  )TT)NF)NNrY  )NNr|  T)r\  )r  NNr  )re  r   r   N)re  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rY  F)Nr   rY  FF)NFNN)r:  FN)r9  NNN)NNr|  )r;  NFN)r\  FF)Nr   FTN)NNFrb  r|  )NNNr  )NNrk  Nr  r9  )r\  TF)	NNr;  Fr  FNFN)rY  r   rY  )NNFN)Fr  r  )r  )r:  )r  r   N)rY  Fr  r:  )r  r  F)NNNNTrb  r|  )NNr  NNF)NNr  N)r  F)r  r  r  )Nr  N)Nr   N)TNTNFNNNNNNF)rY  r  NNNr  )NNr  )NNrk  Nr  )r;  rY  r  N)r   r   )r  r  F)TFNr}  Nr  )Nr  )r  r  FF)Nr9  )NNr  r  )r  r  N)rY  r  )r  r;  r  FNNr  )r9  r  N)r   r  r  N)r  NFNN)NNFNN)r;  NFNN)r  r  FNN)r;  r   )TF)NTr;  )r/  )r/  F)rG  N)NFrI  N)r;  rc  rd  F)r;  rf  rg  F)r;  ri  rj  F)rY  r  )r-  NNNr   F)r   NNrY  )r:  F)	NNNTr  FTNN)TTN)r  r;  N)r;  N)TFF)TFFN)rY  r;  )Nr   NN(  r    r5   rF  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dotrP  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	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacrB  	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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  rI  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_hotrP  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ravelrA  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_hooksr3  _cdatar4  r5  _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   rQ  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_ptrrh  r{  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesrz  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_shortrw  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizerG  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asra  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  subnamerQ  r!   s              r0   r   r     s_   6 \Fz%	---z%222z% 	!#@#@z% 	!#A#A	z%
 	
...z% 	''z% 	000z% 	///z% 	111z% 		444z% 	QQQz% 	LLLz% 	LLLz% 	LLLz% 	JJJz%  	
KKK!z%" 	#%J%J#z% z%$ 		---%z%& 	XXX'z%( 	FFF)z%* 	
...+z%, 	
...-z%. 	JJJ/z%0 	///1z%2 		FFF3z%4 	&&5z%6 	&&7z%8 	1119z%: 	
...;z%< 	22=z%> 	000?z%@ 	///Az%B 	111Cz%D 	
...Ez% z% z%F 	000Gz%H 	666Iz%J 	888Kz%L 	///Mz%N 	111Oz%P 	--Qz%R 	--Sz%T 	--Uz%V 	xxxWz%X 	RRRYz%Z 	{{[z%\ 	'){){]z%^ 	(*u*u_z%` 	 Q Qaz%b 	%'v'vcz%d 	1  4C  4Cez%f 	 5 5gz% z% z%h 	%'\'\iz%j 	CCCkz%l 	??mz%n 	.tttqz%t 	CCCuz%v 	>>>wz%x 	<<<yz%z 	555{z%| 	;;;}z%~ 	<<<z%@ 	 "C"C"CAz%B 	!#D#D#DCz%D 	--Ez%F 		CCCGz%H 	!4!4Iz%J 	11Kz%L 	]]]Mz% z% z%N 	11Oz%P 		666Qz%R 	999Sz%T 	>>>Uz%V 	aaaWz%X 	
...Yz%Z 	
>>>[z%\ 	$::::]z%^ 	77_z%` 	???az%b 	999cz%d 	 "P"P"Pez%f 	 G G Ggz%h 	NNNiz%j 	&(Y(Ykz%l 	444mz%n 	CCCoz% z% z%p 	
BBBqz%r 	888sz%t 	888uz%v 	888wz%x 		OOOyz%z 	%%{z%| 	III}z%~ 	,,z%@ 	999Az%B 	((Cz%D 	555Ez%F 	
...Gz%H 	777Iz%J 	666Kz%L 	555Mz%N 	===Oz%P 	dddQz% z% z%R 	dddSz%T 	dddUz%V 	wwWz%X 	===Yz%Z 	  !A  !A  !A[z%\ 	  !A  !A  !A]z%^ 	  !A  !A  !A_z%` 	((az%b 		---cz%d 	#  &C  &C  &Cez%f 	
...gz%h 	!C!C!Ciz%j 	--kz%l 	@@@mz%n 	EEEoz%p 	xxxsz%v 	555wz% z% z%x 	555yz%z 	BBB{z%| 	AAA}z%~ 	"$@$@$@z%@ 	;;;Az%B 	111Cz%D 	**Ez%F 		##Gz%H 	**Iz%J 	&&Kz%L 	
:::Mz%N 	@@@Oz%P 	222Qz%R 	
VVVSz%T 	BBBUz%V 	KKKWz%X 	 O O OYz% z% z%Z 	 "Y"Y"Y[z%\ 	111]z%^ 	
000_z%` 		HHHaz%b 	KKKcz%d 		444ez%f 	@@@gz%h 	
:::iz%j 	
))kz%l 	;;mz%n 	222oz%p 	444qz%r 	888sz%t 	???uz%v 	CCCwz%x 	44yz%z 	|||}z% z% z%@ 	 j  j  jCz%F 	eeeGz%H 	333Iz%J 	,,Kz%L 		---Mz%N 	
...Oz%P 	000Qz%R 		---Sz%T 	
...Uz%V 	///Wz%X 	.0o0oYz%Z 	-/h/h[z%\ 	- C  C  C_z%b 	')N)Ncz%d 	79^9^ez%f 	')})}gz%h 	7``kz% z% z%n 	+-=-=oz%p 	*,<,<qz%r 	*,B,Bsz%t 	#%?%?uz%v 	99wz%x 		CCCyz%z 		CCC{z%| 		DDD}z%~ 		CCCz%@ 		DDDAz%B 		JJJCz%D 		KKKEz%F 		DDDGz%H 		EEEIz%J 		EEEKz%L 		FFFMz%N 		FFFOz% z% z%P 		GGGQz%R 		IIISz%T 		JJJUz%V 		JJJWz%X 		KKKYz%Z 		666[z%\ 		777]z%^ 		BBB_z%` 		---az%b 	@@@cz%d 	
**ez%f 	&&gz%h 	&&iz%j 	QQQkz%l 	///mz%n 	33oz%p 	???qz% z% z%r 	
555sz%t 	
...uz%v 	///wz%x 	t4PUP]fjz  D  D  Dyz%z 	&(H(H{z%| 	\\\}z%~ 	OOOz%@ 		444Az%B 	333Cz%D 	**Ez%F 	>>>Gz%H 	///Iz%J 	,,,Kz%L 	666Mz%N 	555Oz%P 		333Qz%R 	NNNSz% z% z%T 	ccUz%V 	ffWz%X 	ffYz%Z 	mmm[z%\ 		ss]z%^ 	NNN_z%` 	333az%b 	888cz%d 	555ez%f 	;;;gz%h 	"$z$z$ziz%j 	GGGkz%l 	mmmmz%n 	YYYoz%p 	(*?*?qz%r 	444sz%t 	;;uz% z% z%v 	222wz%x 	666yz%z 	777{z%| 	888}z%~ 	
...z%@ 	==Az%B 	>>Cz%D 	LLLEz%F 	BBBGz%H 	==Iz%J 	\\\Kz%L 	))Mz%N 	
GGGOz%P 	&&Qz%R 	''Sz%T 	222Uz%V 	222Wz% z% z%X 	tt[z%^ 	((_z%` 	111az%b 	444cz%d 	KKKez%f 	**gz%h 	''iz%j 	&&kz%l 	..mz%n 	,,oz%p 	!1!1qz%r 	**sz%t 	33uz%v 	))wz%x 	WWWyz%z 	%%{z%| 	 d  d  dz% z% z%B	 	rrrC	z%D	 	
++E	z%F	 	NNNG	z%H	 	"$c$c$cI	z%J	 	!L!L!LK	z%L	 	 S S SM	z%N	 	sssO	z%P	 		444Q	z%R	 	666S	z%T	 	333U	z%V	 	;;;W	z%X	 	
;;;Y	z%Z	 	000[	z%\	 	  K  K  K]	z%^	 		---_	z%`	 	<<<a	z%b	 	///c	z% z% z%d	 	///e	z%f	 	
...g	z%h	 	:::i	z%j	 	;;;k	z%l	 	&&m	z%n	 	...o	z%p	 	<<<q	z%r	 	555s	z%t	 	;;;u	z%v	 	<<<w	z%x	 	///y	z%z	 	III{	z%|	 	
ss}	z%~	 	OOO	z%@
 	333A
z%B
 	
555C
z%D
 	EEEE
z% z% z%F
 	BBBG
z%H
 	!  $A  $A  $AI
z%J
 	88K
z%L
 	<<M
z%N
 	===O
z%P
 	777Q
z%R
 	???S
z%T
 	 F F FU
z%V
 	!#]#]#]W
z%X
 	[[[Y
z%Z
 	>>>[
z%\
 	//]
z%^
 	!#@#@#@_
z%`
 	 "M"M"Ma
z%b
 	 < < <c
z%d
 	**e
z%f
 	!1!1g
z% z% z%h
 		---i
z%j
 	888k
z%l
 	
555m
z%n
 	lllo
z%p
 	lllq
z%r
 	llls
z%t
 	%tttw
z%z
 	
...{
z%|
 	VVV}
z%~
 	000
z%@ 	333Az%B 	55Cz%D 		---Ez%F 	888Gz%H 	
555Iz%J 	ttMz%P 	 "}"}Qz% z% z%R 	)+v+vSz%T 	%'h'hUz%V 	*wwYz%\ 	*gg_z%b 	 d  dez%h 	BBBiz%j 	
EEEkz%l 	<<mz%n 	==oz%p 	AAAqz%r 		444sz%t 	999uz%v 	UUUwz%x 	111yz%z 	++{z%| 	::}z%~ 	WWWz% z% z%@ 	!s!sAz%B 	&(_(_Cz%D 	88Ez%F 	!f!f!fGz%H 	!V!VIz%J 	UUUKz%L 	$&>&>Mz%N 	333Oz%P 	:::Qz%R 		---Sz%T 	222Uz%V 	:::Wz%X 	/1N1NYz%Z 	/1N1N[z%\ 	/1d1d1d]z%^ 	<>q>q>q_z%` 	/1d1d1daz% z% z%b 	<>q>q>qcz%d 	/1d1d1dez%f 	<>q>q>qgz%h 	')S)S)Siz%j 	)+a+a+akz%l 	& B  B  Boz%r 	& B  B  Buz%x 	&xxx{z%~ 	$&R&R&Rz%@ 	0cccCz%F 	<tttIz%L 	 "L"L"LMz%N 	1iiiQz%T 	) L  L  LWz%Z 	$xxx]z%` 	#%Z%Z%Zaz%b 	%'\'\'\cz% z% z%d 	%'\'\'\ez%f 	%'\'\'\gz%h 	!K!K!Kiz%j 	%|||mz%p 	) J  J  Jsz%v 	13i3i3iwz%x 	 "m"m"myz%z 	1zzz}z%@ 	>zzzCz%F 	1zzzIz%L 	>zzzOz%R 	-/u/u/uSz%T 	 "F"F"FUz%V 	!9!9!9Wz%X 	')z)z)zYz%Z 	&(g(g(g[z%\ 	*,c,c,c]z% z% z%^ 	&(C(C(C_z%` 	$&`&`&`az%b 	0bbbez%h 	) I  I  Ikz%n 	' M  M  Mqz%t 	"  %A  %A  %Auz%v 	#%|%|%|wz%x 	&(m(m(myz%z 	&(\(\(\{z%| 	"$G$G$G}z%~ 	/1g1g1gz%@ 	')^)^)^Az%B 	&(8(8Cz%D 	%'m'm'mEz%F 	%'m'm'mGz%H 	%'m'm'mIz%J 	/iiiMz% z% z%P 	&tttSz%V 	3tttYz%\ 	&ttt_z%b 	3tttez%h 	&tttkz%n 	3tttqz%t 	(*z*z*zuz%v 	(*z*z*zwz%x 	(*z*z*zyz%z 	$&}&}&}{z%| 	8 _  _  _z%B 	-tttEz%H 	2VVVKz%N 	7cccQz%T 	$vvvWz%Z 	%'X'X'X[z%\ 	#%F%F%F]z% z% z%^ 	!P!P!P_z%` 	-/a/a/aaz%b 	,}}}ez%h 	!#;#;iz%j 	 "A"A"Akz%l 	!#B#B#Bmz%n 	$&_&_&_oz%p 	!#y#y#yqz%r 	 "A"A"Asz%t 	 "A"A"Auz%v 	 "A"A"Awz%x 	8:u:u:uyz%z 	*  -A  -A  -A{z%| 	&(j(j(j}z%~ 	,.x.x.xz%@ 	#%Z%Z%ZAz%B 	#%Z%Z%ZCz% z% z%D 	$&L&L&LEz%F 	&(C(C(CGz%H 	$&6&6Iz%J 	&(8(8Kz%L 	%'X'X'XMz%N 	/ L  L  LQz%T 	=DHQT[`lrvvvvWz%Z 	"$b$b$b[z%\ 	 O O O]z%^ 	SSS_z%` 	!7!7az%b 	&(x(x(xcz%d 	777ez%f 	FFFFgz%h 	((iz%j 	
\\\kz%l 	dddmz% z% z%n 	 "h"h"hoz%p 	  +/#3 #3 #3qz%x 	999yz%z 	ddd{z%| 	%%}z%~ 	**z%@ 	QQQAz%B 	!S!S!SCz%D 	++Ez%F 	IIIGz%H 	***Iz%J 	555Kz%L 	IIIMz%N 	==Oz%P 	AAQz%R 	777Sz%T 	 Y YUz% z% z%V 	666Wz%X 	222Yz%Z 	--[z%\ 	ddd]z%^ 		777_z%` 	
000az%b 		DDDcz%d 	 "2"2ez%f 	"$4$4gz%h 	')9)9iz%j 	''kz%l 	,,mz%n 	777oz%p 	CCCqz%r 	fffsz%t 	iiiuz%v 	"$V$Vwz% z% z%x 	!#M#Myz%z 	)+P+P{z%| 	"$s$s}z%~ 	  a  aAz%D 	! a  aGz%J 	"J# # #Mz%T 	"L # # #Wz%` 	"O 	# # #cz%n 	% a  aqz%t 	% a  awz%z 	111{z%| 	ddd}z%~ 	d5=Y]mrvvvz%@ 	eeeAz%B 	%%Cz%D 	
...Ez%F 	
555Gz% z% z%H 	FFFIz%J 	,,Kz%L 	//Mz%N 	444Oz%P 	
333Qz%R 	:::Sz%T 	AAAUz%V 	!;!;!;Wz%X 	..Yz%Z 	QQQ[z%\ 	xx]z%^ 	SSS_z%` 	xxaz%b 	SSScz%d 	
777ez%f 	777gz%h 	///iz% z% z%j 	555kz%l 	PPmz%n 	bbboz%p 	///qz%r 	
444sz%t 	MMMuz%v 	88wz%x 	<<yz%z 	ZZZ{z%| 	eee}z%~ 	|||z%@ 	22Az%B 	??Cz%D 	WWWEz%F 	WWWGz%H 	
333Iz%J 	111Kz% z% z%L 	
...Mz%N 	111Oz%P 		---Qz%R 		---Sz%T 	
...Uz%V 	
...Wz%X 	''Yz%Z 	..[z%\ 		999]z%^ 	
:::_z%` 	888az%b 	@@@cz%d 	WWWez%f 	
YeQUYYYYgz%h 	EEEiz%j 	 P P Pkz%l 	
...mz% z% z%n 	000oz%p 	;;;qz%r 	OOOsz%t 	888uz%v 		---wz%x 	222yz%z 	
 @  @  @}z%@ 		444Az%B 	999Cz%D 		---Ez%F 	))Gz%H 	''Iz%J 	Kz%L 	Mz%N 	''Oz%P 	))Qz%R 	Sz% z% z%T 	))Uz%V 	((Wz%X 	))Yz%Z 	(([z%\ 	))]z%^ 	((_z%` 	))az%b 	))cz%d 	))ez%f 	))gz%h 	000iz%j 		IIIkz%l 	AAAmz%n 	HHHoz%p 	888qz%r 	44sz%t 	66uz% z% z%v 	//wz%x 	!1!1yz%z 	!1!1{z%| 	!1!1}z%~ 	!1!1z%@ 	,.K.K.KAz%B 	,.K.K.KCz%D 	,.K.K.KEz%F 	,.K.K.KGz%H 	//Iz%J 	,,Kz%L 	++Mz%N 	,,Oz%P 	--Qz%R 	..Sz%T 	,,Uz%V 	--Wz% z% z%X 	--Yz%Z 	 A A A[z%\ 	!B!B!B]z%^ 	//_z%` 	*,I,I,Iaz%b 	+-J-J-Jcz%d 	**ez%f 	++gz%h 	**iz%j 	++kz%l 	+-J-J-Jmz%n 	+-J-J-Joz%p 	--qz%r 	 0 0sz%t 	!#D#D#Duz%v 	--wz%x 	!O!O!Oyz% z% z%z 	(*:*:{z%| 	(*:*:}z%~ 	(*:*:z%@ 	(*:*:Az%B 	"$7$7Cz%D 	,,Ez%F 	--Gz%H 	!>!>!>Iz%J 	++Kz%L 	--Mz%N 	/1A1AOz%P 	/1A1AQz%R 	46S6S6SSz%T 	46S6S6SUz%V 	46S6S6SWz%X 	46S6S6SYz%Z 	,,[z% z% z%\ 	@@@]z%^ 	)+;+;_z%` 	@@@az%b 	>>>cz%d 	<<<ez%f 	!!gz%h 	
++iz%j 	KKKkz%l 		---mz%n 	
...oz%p 	 3 3 3qz%r 	 "<"<"<sz%t 	:::uz%v 	HHHwz%x 	JJJyz%z 	
**{z%| 	
KKK}z% z% z%~ 	%%z%@ 	55Az%B 	111Cz%D 	555Ez%F 	 e e eGz%H 	%'a'a'aIz%J 	
:::Kz%L 	! L  L  LOz%R 	
:::Sz%T 	22Uz%V 	///Wz%X 	---Yz%Z 	<<[z%\ 	hhh]z%^ 	 "g"g"g_z%` 	66az%b 	;;;cz% z% z%d 	LLLez%f 	%'W'W'Wgz%h 	888iz%j 	111kz%l 		---mz%n 	222oz%p 	;;qz%r 	222sz%t 	999uz%v 	%'`'`wz%x 	1oo{z%~ 	eeez%@ 	55Az%B 	@@Cz%D 	Ez%F 	"OOGz%H 	//Iz% z% z%J 	ooKz%L 	QQQMz%N 	')>)>Oz%P 	FFFQz%R 	%CCCCSz%T 	>>Uz%V 	11Wz%X 	!#@#@Yz%Z 	66[z%\ 	??]z%^ 	NNN_z%` 	<<<az%b 	#%H%H%Hcz%d 	00ez%f 	oogz%h 	99iz%j 	22kz% z% z%l 	__mz%n 	OOoz%p 	OOqz%r 	??sz%t 	uz%v 	wz%x 	yz%z 	{z%| 	111}z%~ 	//z%@ 	AAAz%B 	//Cz%D 	33Ez%F 	44Gz%H 	44Iz%J 	22Kz%L 	33Mz% z% z%N 	33Oz%P 	11Qz%R 	22Sz%T 	22Uz%V 	11Wz%X 	22Yz%Z 	22[z%\ 	..]z%^ 	--_z%` 	..az%b 	//cz%d 	OOez%f 	00gz%h 	iz%j 	33kz%l 	mz%n 	??oz% z% z%p 	..qz%r 	//sz%t 	//uz%v 	55wz%x 	00yz%z 	22{z%| 	}z%~ 	z%@ 	//Az%B 	Cz%D 	77Ez%F 	44Gz%H 	__Iz%J 	AAAAKz%L 	11Mz%N 	//Oz%P 	//Qz% z% z%R 	//Sz%T 		??Uz%V 		??Wz%X 	&Yz%Z 	*2OO[z%\ 	oo]z%^ 	_z%` 	__az%b 	oocz%d 	ez%f 	gz%h 	iz%j 	-/p/pkz%l 	*,V,Vmz%n 	2OOoz%p 	__qz%r 	sz% z% z%t 	oouz%v 	wz%x 	yz%z 	{z%| 	}z%~ 	z%@ 	Az%B 	#__Cz%D 	Ez%F 	Gz%H 	Iz%J 	 //Kz%L 	Mz%N 	 //Oz%P 	#__Qz%R 	 //Sz%T 	$ooUz% z% z%V 	 //Wz%X 	Yz%Z 	[z%\ 	__]z%^ 	oo_z%` 	az%b 	__cz%d 	 //ez%f 	$oogz%h 	ooiz%j 	kz%l 	__mz%n 	__oz%p 	'/qz%r 	NNNsz%t 	oouz%v 	oowz% z% z%x 	yz%z 	{z%| 	__}z%~ 	__z%@ 	OOAz%B 	__Cz%D 	OOEz%F 	==Gz%H 	Iz%J 	Kz%L 	//Mz%N 	==Oz%P 	00Qz%R 	88Sz%T 	99Uz%V 	kkkWz%X 	E4IMMMYz% z% z%Z 	0EIII[z%\ 	0EIII]z%^ 	0EIII_z%` 	MTMMMMaz%b 	cz%d 	66ez%f 	e6MQQQgz%h 	>>>iz%j 	
u/DHHHkz%l 	0EIIImz%n 	0EIIIoz%p 	
u/DHHHqz%r 	
u/DHHHsz%t 	uz%v 	//wz%x 	!O!O!Oyz%z 	
OO{z% z% z%| 	@@@}z%~ 	%2GKKKz%@ 	53HLLLAz%B 	__Cz%D 	,,Ez%F 	00Gz%H 	HHHHHIz%J 	,,Kz%L 	55Mz%N 	1FJJJOz%P 	%2GKKKQz%R 	@@@@Sz%T 	??Uz%V 	0EIIIWz%X 	1FJJJYz%Z 	//[z%\ 	]z% z% z%^ 	
u/DHHH_z%` 	__az%b 	oocz%d 	__ez%f 	//gz%h 	11iz%j 	//kz%l 	__mz%n 	MTMMMMoz%p 	00qz%r 	0EIIIsz%t 	66uz%v 	55wz%x 		888yz%z 	@@@{z%| 	EE}z%~ 	??z% z% z%@  	A z%B  	"OOC z%D  	-E z%F  	%G z%H  	I z%J  	ooK z%L  	,,M z%N  	??O z%P  	GGGQ z%R  	S z%T  	LDLLLLU z%V  	55W z%X  	33Y z%Z  	33[ z%\  	13H3H] z%^  	,,_ z%`  	--a z% z% z%b  	BBBc z%d  	11e z%f  	--g z%h  	--i z%j  	00k z%l  	 "8"8m z%n  	OOo z%p  	[[[q z%r  	??s z%t  	oou z%v  	1FJJJw z%x  	__y z%z  	WWW{ z%|  	??} z%~  	11 z%@! 	&(W(W(WA!z%B! 	GGC!z% z% z%D! 	')Q)QE!z%F! 	OOOG!z%H! 	I!z%J! 	K!z%L! 	M!z%N! 	__O!z%P! 	11Q!z%R! 	++S!z%T! 		EUZUjnnnU!z%V! 	IIIIIW!z%X! 	GGGY!z%Z! 	//[!z%\! 	]!z%^! 	//_!z%`! 	..a!z%b! 	==c!z%d! 	777e!z% z%f! 	++..oo777 //FFFs!z% z%Cz! 	(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>.wrapped/  s]    &J777M!-00 V,WmUdUUUfUUU4((((r3   )	functoolsr   )r!   r
  r
  s   `@r0   r^  z"wrap_torch_function.<locals>.inner.  sD    				) 	) 	) 	) 	) 	) 
		) r3   r;  )r
  r^  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   )rc
  r    _C_is_torch_function_enabledsetr
  r   _disabled_torch_function_implr  r
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer/  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;  )rc
  ).0r
  s     r0   
<listcomp>z)handle_torch_function.<locals>.<listcomp>  s    EEEcS		EEEr3   z nor in mode )r
  tuplemaprc
  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_funcsr/  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   j  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.
    rY  )r  r    _ops
OpOverloadOpOverloadPacketstrr  get)fs    r0   r   r   w  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_methodsr*    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&  )r*  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   )rc
  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__`` overrideable 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!)
    r^  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   r6  r<  classmethodr@  r;  r3   r0   r/  r/    s          <    " " " "       [  r3   r/  c                  R    t                      } | dk    rt          | dz
            nd S )Nr   rY  )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   rG  s    r0    _get_current_function_mode_stackrK    s)    )++I@@uY/?/?@@@@r3   c                 $    t          |            d S r  )r   )r  s    r0   r5  r5    s    !$'''''r3   c                  "    t                      } | S r  )r   olds    r0   r8  r8    s    
#
%
%CJr3   c               #   v   K   t                      } 	 | V  t          |            d S # t          |            w xY wr  )r8  r5  rN  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;  r3  s        r0   r   z(BaseTorchFunctionMode.__torch_function__(  s!    >FtT$V$$$r3   rA  )r
  r  rB  r   r;  r3   r0   rR  rR  '  s(        % % % % % %r3   rR  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_functionrZ  .  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   8  s       
	+	+	-	-  	EEEDDDDD	                 s   ;(;*;??)r   r    r  )ArC  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   r6  dictr   r   rc
  r  r
  r   r   r   r   r
  r  r   r   r*  rv
  r   r   r/  r  rK  r5  r8  contextmanagerr
  rR  rZ  r   r;  r3   r0   <module>re     sr   ,                   $ $ $ $ $ $       3 3 3 3 3 3 3 3 3 3 3 3 ' ' ' ' ' ' 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
   Yt__WT]]
 1   
2r6
     b"f	       F Ss8} S S S  Sl c(m    2 ztHh$67 z z z  zz##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   