
    קg1             	       (   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	m
Z
mZmZmZmZmZ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	 	 d+d	e
d
edede
fdZ ej        d          edee
         fd                        Z  ej        d          dee
         fd            Z! ej        d          edee
e
f         fd                        Z"de
fdZ#	 d,dee	         d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          deee	ee
         f         ee
ef         f         fd            Z)edee	ee
         f         fd            Z*ed             Z+ ej        d          dee
         fd            Z,ed	e
de-fd            Z.d  Z/ G d! d"          Z0d# Z1d$ Z2d% Z3d& Z4ej5        d'             Z6 G d( d)e0          Z7ej5        d*             Z8dS )-aE  
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wraps)AnyCallableDictIterableListSetTupleType)	_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.*is deprecated, please use.*torchfuncregexmodulereturnc                 D     t                      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.
    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)argskwargsr"   r$   r#   s     K/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperU   s    $&& 	) 	)#;f    4(((		) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)s   %AAAr   )r"   r#   r$   r2   s   ``` r1   _disable_user_warningsr3   =   sA    0 4[[) ) ) ) ) ) [) 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        t           jO        jO        t           jO        j        t           jO        jP        t           jO        j        t           jQ        t           jR        jS        t           jR        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        jc        t           j{        j        j        t           j{        j        jQ        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
get_device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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r6   s    r1   r   r   `   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 	 {@| 	}@~ 	@ @r4   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!   r6   _baser'  grad_gradr2  s    r1   get_default_nowrap_functionsr7  x  s-    $ \F r4   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        d2d`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        d2drt           jg        ddsi t           jh        ddtt           ji        ddut           jj        dv t           jk        d2dw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        d2d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 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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 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        d0dt           jP        j        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 t           j        d t           j        ddt           j        dd	t           j        d
 t           j        d1dt           j        d t           j        d$dt           j        d t           j        d t           j        ddt           j        ddt           j        d i 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        d2d"t           j        d3d#i 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d/t           j        d4d0t           j        dd1t           j        dd2t           j        dd3t           j         dd4i t           j!        dd5t           j"        dd6t           j#        dd7t           j$        d8 t           j%        d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@ t           j-        d
dAt           j.        ddBt           j/        ddCt           j0        d5dDt           j1        ddEi t           j2        ddFt           j3        dG t           j4        dH t           j5        ddIt           j6        ddJt           jP        j0        d6dKt           jP        j7        d6dLt           jP        j8        d5dMt           jP        j1        d5dNt           jP        j6        ddOt           j9        dP t           jP        j9        ddQt           jP        j:        d)dRt           jP        j;        ddSt           j<        dT t           jP        j<        dU t           j=        ddVi t           j>        ddWt           j?        ddXt           j@        d7dYt           jA        d7dZt           jB        d7d[t           jC        d8d\t           jD        dd]t           jE        d9d^t           jF        dd_t           jG        dd`t           jH        da t           jI        ddbt           jJ        ddct           jK        dddt           jL        de t           jM        df t           jN        dg i t           jO        dh t           jP        di t           jQ        dj t           jR        dk t           jS        ddlt           jT        d:dmt           jU        dn t           jV        do t           jW        ddpt           jX        ddqt           jY        ddrt           jZ        ddst           j[        ddtt           j\        du t           j]        dv t           j^        d;dxt           j_        dy i t           j`        dz t           ja        d{ t           jb        d<d|t           jc        d} t           jd        d=d~t           je        d t           jf        ddt           jg        ddt           jh        ddt           ji        ddt           jj        ddt           jk        jl        jm        d t           jk        jl        jn        d t           jk        jl        j        ddt           jk        jl        jo        ddt           jk        jl        jp        ddt           jk        jl        jq        ddi t           jk        jl        jr        ddt           jk        jl        js        ddt           jk        jl        jt        ddt           jk        jl        j        d>dt           jk        jl        ju        d?dt           jk        jl        jv        d?dt           jk        jl        j.        d@dt           jk        jl        j7        ddt           jk        jl        jw        dAdt           jk        jl        j8        ddt           jk        jl        jL        ddt           jk        jl        jq        ddt           jk        jl        jx        dBdt           jk        jl        jv        ddt           jk        jl        j        dCdt           jk        jl        jy        dCdt           jk        jl        jz        dCdi t           jk        jl        j{        dCdt           jk        jl        j|        ddt           jk        jl        j        d!dt           jk        jl        j        dDdt           jk        jl        j        d>dt           jk        jl        j}        dEdt           jk        jl        j~        dFdt           jk        jl        j        dFdt           jk        jl        j        dFdt           jk        jl        j        dFdt           jk        jl        j        dGdt           jk        jl        j        dHdt           jk        jl        j        dIdt           jk        jl        j        dJdt           jk        jl        j        d<dt           jk        jl        j        dKdt           jk        jl        j        d,di t           jk        jl        j        dLdt           jk        jl        j        d-dt           jk        jl        j        dMdt           jk        jl        j        dNdt           jk        jl        j        d3dt           jk        jl        j        dOdt           jk        jl        j        d<dt           jk        jl        j        dPdt           jk        jl        j        ddt           jk        jl        j        dQdt           jk        jl        j        dRdĄt           jk        jl        j        dń t           jk        jl        j        d)dƄt           jk        jl        j        d)dǄt           jk        jl        j        d)dȄt           jk        jl        j2        ddɄt           jk        jl        j@        d8dʄi t           jk        jl        jC        d8d˄t           jk        jl        jA        d8d̄t           jk        jl        j        d8d̈́t           jk        jl        jB        d8d΄t           jk        jl        j        d8dτt           jk        jl        j        dSdЄt           jk        jl        j        dSdфt           jk        jl        j        dSd҄t           jk        jl        j        dOdӄt           jk        jl        j        dTdԄt           jk        jl        j        dUdՄt           jk        jl        j        dOdքt           jk        jl        j        dAdׄt           jk        jl        j        dVd؄t           jk        jl        j        dWdڄt           jk        jl        j        dIdۄt           jk        jl        j        dXd݄i t           jk        jl        j        dYdބt           jk        jl        j        dZd߄t           jk        jl        j        d t           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        d[dt           jk        jl        j        d\dt           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        d]dt           jk        jl        j        d^dt           jk        jl        j        d_dt           jk        jl        j        dOdt           jk        jl        j        dRdt           jk        jl        j        dRdt           jk        jl        j        d`di t           jk        jl        j        d,dt           jk        jl        j        d t           jk        jl        j        d t           jk        jl        j        ddt           jk        jl        j        dadt           jk        jl        j        ddUdd>ddt           jk        jl        j        dEdt           jk        j        j        dbdt           jk        j        j        dbdt           jk        j        j        d t           jk        j        j        dcdt           j        dd t           j        dddt           j        d t           j        dddt           jP        j        dedt           jP        j        dfdi t           jP        j        	 dgdt           j        dhd	t           j        ddd
t           j        d t           j        d t           j        didt           j        dYdt           j        d t           j        djdt           j        ddt           j        dkdt           jP        j        dldt           j        d t           j        d t           j        ddt           j        d t           j        ddi 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        d6d%t           jP        j        dmd't           j        dnd)t           j        dnd*t           j        d+ t           j        d, i t           j        d- t           j        d. t           j        d/ t           j        d0 t           j        	 	 dod4t           j        	 	 dpd7t           j        	 	 dq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#d@t           j        dA t           j        ddBt           j        ddCt           jP        j        ddDi t           j        dE t           j        dF t           j        ddGt           j        ddHt           j        ddIt           j        ddJt           j        ddKt           j        dL t           j        d[dMt           j        dN t           j        d
dOt           j        dP t           j        d
dQt           j        ddRt           j        drdTt           j        ddUt           j        ddVi t           j        dW t           j        d\dXt           j         ddYt           j        d3dZt           j        dd[t           j        d\ t           j        d] t           j        d1d^t           j        dd_t           j        dsdat           j        db t           j	        dc t           j
        dtddt           j        dtdet           j        d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        dn t           jP        j        do t           j        dp t           j        dq t           j        ddrt           jP        j        d6dst           jP        j        d5dtt           j        dudddudvt           j        d2dwt           j        d2dxt           j        d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"        dv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 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dt           j7        dwdt           j8        dxdt           jP        j7        d6dt           jP        j9        ddt           j:        d t           j;        d t           j<        j=        d t           j<        j>        d i t           j<        j?        d t           j<        j@        d t           j<        jA        d t           j<        jB        ddt           j<        jC        ddt           j<        jD        ddt           j<        jE        ddt           j<        j        d t           j<        jF        d t           j<        j        d t           j<        j        d t           j<        jG        d t           j<        j        d t           j<        j        d t           j<        jH        d t           j<        j        d t           j<        jI        ddi t           j<        jJ        ddt           j<        jK        d t           j<        jL        ddt           j<        jM        ddt           j<        j        d t           j<        jN        d t           j<        jO        d t           j<        jP        d t           j<        jQ        ddt           j<        jR        ddt           j<        j         d t           j<        jS        d t           j<        j        ddt           j<        j*        d t           j<        j+        ddt           j<        jT        d t           j<        jU        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dƄt           j<        j[        dǄ t           j<        j        dȄ t           j<        j\        dɄ t           j<        j]        dʄ t           j<        j^        dd˄t           j<        j_        dd̄t           j<        j`        dd̈́t           j<        ja        dd΄t           j<        j        dτ t           j<        j        ddЄt           j<        jb        dф i t           j<        jc        dd҄t           j<        j%        ddӄt           j<        jd        ddԄt           je        dՄ t           jf        dք t           jg        d
dׄt           jh        dd؄t           ji        ddلt           jP        jj        ddڄt           jP        jk        ddۄt           jl        dyd܄t           jm        d2d݄t           j        ddބt           jn        d߄ t           jo        d:dt           jp        d t           jq        d i t           jr        ddt           js        ddt           jt        dzdt           jP        ju        didt           jv        ddt           j        dadt           jw        ddt           jx        d t           jy        ddt           jz        d2dt           j{        d t           j|        d{dt           j}        ddt           j~        d t           j        d2dt           j        d2dt           j        d2di 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 t           j        d t           j        ddi 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        dtdt           j        d2dt           j        d2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           j        d t           j        d t           j        d t           j        d t           j        d2dt           j        d t           j        d t           j        d  | j        d! | 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        ddDdE| j        dF | j        dG | j        j        dH | j        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        dU | j        dV | j        j        dW | j        j        dX | j        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        d)dx| j        dy | j        dz | j        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        dFd| j        t           j        fd| j        t           j        fd| j        t           j        fd| j        t           j        fdi | j        dSd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 | j(        d | j)        t           j        fd| j*        t           j        fdi | j+        d | j,        d | j-        d | j.        d3ddd| j/        d | j0        d | j1        t           j        fd| j2        t           j        fd| j3        ddd| j4        d | j5        t           j        fd| j6        t           j        fd| j7        d | j8        d | j9        t           j        fd| j:        d | j;        d i | j        d | j<        d | j=        d | j>        d | j?        d | j@        d|ddd| j        d | jA        t           j        fd| jB        d | jC        d | jS        d | jD        dd| j        d | jE        d | jF        d | jG        d | jH        d i | jI        dÄ | j        dĄ | jJ        dń | j        dƄ | jK        dǄ | jL        ddȄ| jM        dɄ | jN        ddddʄ| jO        d˄ | jP        d̄ | jQ        d̈́ | jR        d΄ | jS        dτ | jT        dЄ | jU        d1dф| jV        d҄ | jW        dӄ i | jX        dԄ | jY        dՄ | jZ        dք | j[        dׄ | j\        d}d؄| j	        dل | j]        dڄ | j^        t           j        fdۄ| j_        d܄ | j        dtd݄| j`        dބ | ja        d߄ | jb        dd| jc        d | jd        d | j        dd| je        d i | jf        d | jg        d | jh        d | ji        d | jn        d | jj        ddt           j        fd| jk        dddd| jl        d
d| jm        d | jn        d | jo        d | jp        d | j        d | j        dSd| jq        d | jr        d | js        d | jt        d | ju        dd| jv        d t           jP        jw        d
di}t           jx        jy        jz        }t          | |          r8d)d|t          | |          <   d |t          | d|           j        <   i }t                      }|~                                D ]\  }}|j        |j        dz   d|j        z   dz   d |j        z   dz   d|j        z   dz   g}|j                            d          rP|j        t          d          d         }|                    d|z   dz   d |z   dz   d|z   dz   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     r1   <lambda>z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  r>  output_sizes     r1   r@  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r:  r<  )inputsrF  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  r>  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r:  r<  r>  otherr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      " r4      c                     dS r:  r<  r>  batch1batch2alphabetar?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r:  r<  r>  tensor1tensor2valuer?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r:  r<  rc  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  r>  mat1mat2r`  r_  r?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  )r>  matvecr`  r_  r?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  )r>  vec1vec2r`  r_  r?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  thetasizealign_cornerss      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  r>  dims     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   h㈵>:0yE>Fc                     dS r:  r<  )r>  rX  trolatol	equal_nans        r1   r@  z'get_testing_overrides.<locals>.<lambda>      VX r4   c                     dS r:  r<  r>  ptraininplaces       r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r>  r|  keepdimr?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  )r>  msgs     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                      dS r:  r<  tensorss    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                      dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                      dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   r   Tc                     dS r:  r<  )r>  kernel_sizer  padding	ceil_modecount_include_pads         r1   r@  z'get_testing_overrides.<locals>.<lambda>      vx r4   c                     dS r:  r<  r\  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      PR r4   c	                     dS r:  r<  )	r>  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r1   r@  z'get_testing_overrides.<locals>.<lambda>      y{ r4   c                     dS r:  r<  )grad_outr>  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r>  r  r  r  input_gweight_gbias_gs           r1   r@  z'get_testing_overrides.<locals>.<lambda>      su r4   c                     dS r:  r<  )r>  r  r  r  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r>  r  r  r  r  r  r  counts           r1   r@  z'get_testing_overrides.<locals>.<lambda>      tv r4   c                     dS r:  r<  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>        AC r4   c                     dS r:  r<  r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r:  r<  )r>  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>      Z\ r4   c                     dS r:  r<  )r>  	generatorr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  input1input2r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>      R r4   r  c                     dS r:  r<  r>  targetr  size_averagereduce	reduction
pos_weights          r1   r@  z'get_testing_overrides.<locals>.<lambda>      rt r4   c                     dS r:  r<  )r>  weights	minlengths      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  probr  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      " r4   c                      dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r>  rl  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                      dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  selfrx  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  )r>  
boundaries	out_int32rightr?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      [] r4   c                      dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  r|  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r   s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r:  r<  r   s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4          @#use_mm_for_euclid_dist_if_necessaryc                     dS r:  r<  )x1x2r  compute_modes       r1   r@  z'get_testing_overrides.<locals>.<lambda>      _a r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4         ?c                     dS r:  r<  r>  r_  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   )r?  c                     dS r:  r<  )r?  matricess     r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  r>  groupss     r1   r@  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r:  r<  r>  upperr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  check_errorsr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>       b r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r:  r<  )r  r  r  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r:  r<  )r>  numeln_binsratio	bit_widths        r1   r@  z'get_testing_overrides.<locals>.<lambda>      WY r4   c                     dS r:  r<  r>  chunksr|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  r>  minmaxr?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r,  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  )r>  r-  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r.  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  r?  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>	  r  r4   c                     dS r:  r<  )r>  
correctionfweightsaweightss       r1   r@  z'get_testing_overrides.<locals>.<lambda>
      R r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>      2 r4      c                     dS r:  r<  )r>  rwith_replacements      r1   r@  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r:  r<  )realimags     r1   r@  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )absangs     r1   r@  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r:  r<  )r>  ords     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  padrf  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r:  r<  r>  r  r  r  r  dilationr  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>      bd r4   c                     dS r:  r<  rT  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r4   c                     dS r:  r<  rT  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r4   c	                     dS r:  r<  )	r>  r  r  r  r  rU  
transposedoutput_addingr  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>      uw r4   c                     dS r:  r<  )r>  r  r  rQ  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  r>  r  r  r  r  output_paddingr  rU  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>        A r4   c                     dS r:  r<  r_  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r_  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  r  r  marginr  r  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>   r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>!  rL  r4   c                     dS r:  r<  )r  r  r|  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>"  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>#  rA  r4   c                     dS r:  r<  r>  rX  r|  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>$  rG  r4   r;  c                     dS r:  r<  rm  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>%      2 r4   c                     dS r:  r<  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r1   r@  z'get_testing_overrides.<locals>.<lambda>'  r  r4   c                     dS r:  r<  r>  r|  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>)  r  r4   c                     dS r:  r<  ry  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>*  r  r4   c                     dS r:  r<  r>  r|  r?  ra   s       r1   r@  z'get_testing_overrides.<locals>.<lambda>+  r0  r4   c                     dS r:  r<  r|  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>,  rJ  r4   c                     dS r:  r<  yxr|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>-  rG  r4   c                     dS r:  r<  ry  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>.  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>/  rU  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>0       r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>1       r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>2  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>3  r  r4   c                     dS r:  r<  r>  diagonalr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>4  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>5  rG  r4   c                     dS r:  r<  )r>  offsets     r1   r@  z'get_testing_overrides.<locals>.<lambda>6  rC  r4   c                     dS r:  r<  )r>  nr|  prependappendr?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>7      TV r4   c                     dS r:  r<  r>  r  dim1dim2s       r1   r@  z'get_testing_overrides.<locals>.<lambda>8  r0  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>9  rt  r4   c                     dS r:  r<  )r>  srcr  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>:  r9  r4   c                     dS r:  r<  )r  r  rx  r  storage_offsets        r1   r@  z'get_testing_overrides.<locals>.<lambda>;  r'  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda><  rU  r4   c                     dS r:  r<  )r>  rX  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>=  rQ  r4   c                     dS r:  r<  r>  rX  rounding_moder?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>>      b r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>?  rt  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>@  rY  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>A  rG  r4   c                     dS r:  r<  r>  rl  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>B  r  r4   c                     dS r:  r<  )rk  rl  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>C      r r4   c                     dS r:  r<  r>  indices_or_sectionss     r1   r@  z'get_testing_overrides.<locals>.<lambda>D  r  r4   c                     dS r:  r<  r4  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>E  rC  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>F  rY  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>G  r  r4   Lc                     dS r:  r<  r>  UPLOr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>H  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>I  r  r4   c                     dS r:  r<  )equationoperandss     r1   r@  z'get_testing_overrides.<locals>.<lambda>J  rY  r4   c                     dS r:  r<  r>  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r1   r@  z'get_testing_overrides.<locals>.<lambda>L      z| r4   c
                     dS r:  r<  )
r>  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r1   r@  z'get_testing_overrides.<locals>.<lambda>O        hj r4   c                     dS r:  r<  r>  ra   ro   r`   requires_grads        r1   r@  z'get_testing_overrides.<locals>.<lambda>Q      ce r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>R  r  r4   c                     dS r:  r<  r>  rX  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>S  rD  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>T  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>U  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>V  rQ  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>W  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>X  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>Y  rS  r4   c                     dS r:  r<  )r>  scale
zero_pointaxis	quant_min	quant_maxs         r1   r@  z'get_testing_overrides.<locals>.<lambda>Z      mo r4   c                     dS r:  r<  )r>  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>[      fh r4   c                     dS r:  r<  )r  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r1   r@  z'get_testing_overrides.<locals>.<lambda>]  s      AC r4   c                     dS r:  r<  r>  packed_weightr  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>_  r!  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>`      \^ r4   c                     dS r:  r<  r>  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>a      {} r4   c                     dS r:  r<  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>c      ^` r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>e  rR  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>f  r  r4   c                     dS r:  r<  )r>  abs      r1   r@  z'get_testing_overrides.<locals>.<lambda>g  r0  r4   c                     dS r:  r<  r>  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>h  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>i  r  r4   c                     dS r:  r<  r>  r  r|  norms       r1   r@  z'get_testing_overrides.<locals>.<lambda>j  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>k  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>l  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>m  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>n  r  r4   r  r;  c                     dS r:  r<  r>  sr|  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>o  rp  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>p  rt  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>q  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>r  ro  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>s  ro  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>t  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>u  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>v  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>w  r@  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>x  rp  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>y  rp  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>z  rt  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>{  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>|  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>}  r0  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>~  rA  r4   c                     dS r:  r<  )r>  	start_dimend_dims      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  r>  dimss     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  exponentr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  )r>  
fill_valuer?  ra   ro   r`   r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  s      BD r4   c                     dS r:  r<  )r>  r  	dep_tokens      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r|  indexr?  sparse_grads        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  rs  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r9  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  spacingr|  
edge_orders       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r4   c                     dS r:  r<  r>  gridinterpolation_modepadding_modery  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      ac r4   c                     dS r:  r<  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      df r4   c                     dS r:  r<  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rE  r4   c                     dS r:  r<  )r>  
num_groupsr  r  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      km r4   c	                     dS r:  r<  	r>  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r1   r@  z'get_testing_overrides.<locals>.<lambda>      qs r4   c                     dS r:  r<  r>  rL  w_ihw_hhb_ihb_hhs         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4         ?c                     dS r:  r<  r>  lambds     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  valuesr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  rh  r  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      xz r4   d   c                     dS r:  r<  )r>  binsr-  r.  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  rg  r-  r.  r  densityr?  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  )r>  rg  r   r  ri  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r'  r4   c                     dS r:  r<  r>  taus     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )rk  rl  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r4  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r>  r|  r0  sources       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  rv  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  indicesra  
accumulates       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  )r>  r|  r0  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r4   c                     dS r:  r<  )r>  r|  r0  rf  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  )r>  r|  r0  rw  r  include_inputs         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r[   s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )eteassume_uniqueinverts       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c	                     dS r:  r<  )	r>  r  r  r  r  use_input_statsr  r  r  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  rX  rtolr  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      UW r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c
                     dS r:  r<  )
r>  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r1   r@  z'get_testing_overrides.<locals>.<lambda>        bd r4   c                     dS r:  r<  r>  r  r  r  r  
log_targets         r1   r@  z'get_testing_overrides.<locals>.<lambda>  s    pr r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r:  r<  )r>  kr|  r  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r4   c                     dS r:  r<  )r>  	hermitianr  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  )r>  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  )LDpivotsBr  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      QS r4   c                     dS r:  r<  )r>  normalized_shaper  r  espr  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  endr  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  )r>  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r1   r@  z'get_testing_overrides.<locals>.<lambda>  s      IK r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r>  r|  ra   s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  )r>  namesr  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c	                     dS r:  r<  )	databatch_sizesrL  rM  rN  rO  rP  r  rQ  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rU  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )Apivot	get_infosr?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r4   c                     dS r:  r<  )r  r+  r,  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r4   c                     dS r:  r<  rg  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  )r>  maskrf  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  rw  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )LUr  r  leftadjointr?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      Y[ r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r>  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  )r>  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r:  r<  r4  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>   rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  r  r  rU  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      jl r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  r  r  rU  return_indicesr  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>
  rL  r4   c                     dS r:  r<  )r>  r|  r  ra   r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                      dS r:  r<  )r  r0   s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r  r  r  exponential_average_factorepsilons           r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c	                     dS r:  r<  	r>  r  r  r  r  rU  r  	benchmarkdeterministics	            r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c	                     dS r:  r<  )	r>  r  zr_  r  r  r  rU  r  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rT  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c
                     dS r:  r<  )
r>  r  r  r  r`  r  rU  r  r  r  s
             r1   r@  z'get_testing_overrides.<locals>.<lambda>  r\  r4   c	                     dS r:  r<  r  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>      eg r4   c                     dS r:  r<  )r>  r  weight_stride0rL  cxr  hidden_sizerO  rR  rP  r  rQ  r  dropout_states                 r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>!  rC  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>"  ro  r4   c                     dS r:  r<  r>  rw  destinations      r1   r@  z'get_testing_overrides.<locals>.<lambda>#  r  r4   c                     dS r:  r<  r)  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>$  rR  r4   c                     dS r:  r<  )r>  
descendingr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>%  rJ  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>&  rY  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>'  r  r4   c                     dS r:  r<  )r>  num_samplesreplacementr?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>(      SU r4   c                     dS r:  r<  )r>  ro  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>)  rU  r4   c                     dS r:  r<  r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>*  r  r4   c                     dS r:  r<  )r>  r|  startr  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>+  r  r4           c                     dS r:  r<  )r>  nanposinfneginfr?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>,  r  r4   c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>-  rS  r4   c                     dS r:  r<  )r>  r  r  r  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>.      ]_ r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>/  r  r4   c                     dS r:  r<  r>  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>0  rE  r4   c                     dS r:  r<  )r>  r  r  NCHxWgroupr  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>1  r  r4   c                     dS r:  r<  )r>  r  r|  r  ra   s        r1   r@  z'get_testing_overrides.<locals>.<lambda>2  r3  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>3  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>4  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>5  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>6  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>7  rC  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r4   c                     dS r:  r<  rE  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>9  r!  r4   c                     dS r:  r<  rE  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>:  r!  r4   c                     dS r:  r<  r>  rF  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>;  rV  r4   c                     dS r:  r<  rS  s      r1   r@  z'get_testing_overrides.<locals>.<lambda><      oq r4   c                     dS r:  r<  rS  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>=  rV  r4   c                     dS r:  r<  rS  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>>  rU  r4   c                     dS r:  r<  rS  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>?  rV  r4   c                     dS r:  r<  rS  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>@  rU  r4   c                     dS r:  r<  rv  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>A  r  r4   c                     dS r:  r<  r>  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>B  r
  r4   c                     dS r:  r<  r>  r  r  r  r  r  divisor_overrides          r1   r@  z'get_testing_overrides.<locals>.<lambda>D        @B r4   c                     dS r:  r<  r^  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>G  r`  r4   皙?c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>J  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>L  r  r4   c                     dS r:  r<  r>  r  r  r  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>N  rC  r4   c                     dS r:  r<  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>Q  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>S  rg  r4   c                     dS r:  r<  rg  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>U      gi r4   c                     dS r:  r<  )r>  r  r  r  ignore_indexr  r  label_smoothings           r1   r@  z'get_testing_overrides.<locals>.<lambda>X        JL r4   c                     dS r:  r<  rq  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>[  r  r4   c                     dS r:  r<  r\  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>]      XZ r4   c                     dS r:  r<  r\  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>^  r  r4   c                     dS r:  r<  r\  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>_  r  r4   c                     dS r:  r<  r\  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>`  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>a  rt  r4   c                     dS r:  r<  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>c  r  r4   c                     dS r:  r<  )r>  r  r  r  r  r  r  r   r  include_last_offsetr  s              r1   r@  z'get_testing_overrides.<locals>.<lambda>f  s      HJ r4   c                     dS r:  r<  r\  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>h  rj  r4   c                     dS r:  r<  )r>  rF  r  rU  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>i  rI  r4   c                     dS r:  r<  r>  r  rF  output_ratior  _random_sampless         r1   r@  z'get_testing_overrides.<locals>.<lambda>k  rd  r4   c                     dS r:  r<  r}  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>n  rd  r4   c                     dS r:  r<  r}  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>q  rd  r4   c                     dS r:  r<  r}  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>t  rd  r4   ư>c                     dS r:  r<  )r>  r  varr   r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>v  r  r4   nonec                     dS r:  r<  )r>  approximates     r1   r@  z'get_testing_overrides.<locals>.<lambda>w  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>x  r  r4   bilinearr   c                     dS r:  r<  )r>  r@  r  rB  ry  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>y  rd  r4   c                     dS r:  r<  )r>  rH  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>z  r   r4   绽|=c                     dS r:  r<  )logitsrm  hardr  r|  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>{  rC  r4   c                     dS r:  r<  r^  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>|  r  r4         c                     dS r:  r<  )r>  min_valmax_valr  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>}  r  r4   c                     dS r:  r<  rc  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>      `b r4   c                     dS r:  r<  )r>  r  r  r  r  r  r  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>  s      GI r4   nearestc                     dS r:  r<  )r>  rx  scale_factorr  ry  recompute_scale_factor	antialiass          r1   r@  z'get_testing_overrides.<locals>.<lambda>  s      KM r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r>  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rC  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   {Gz?c                     dS r:  r<  )r>  negative_sloper  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   -C6?      ?c                     dS r:  r<  )r>  rx  r_  r`  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r   r4      c                     dS r:  r<  r>  r|  _stacklevelra   s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  rg  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  rj  r4   c                     dS r:  r<  r>  r  r  r  rU  r  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r
  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r
  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r
  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r
  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  rz  r  r  r  rF  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rd  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rd  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rd  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>      np r4   c                     dS r:  r<  )querykeyrf  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r1   r@  z'get_testing_overrides.<locals>.<lambda>  s      ]_ r4   c                     dS r:  r<  )r>  r  r  rh  r  r  r  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rf  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  )r>  r  r  r  rm  r  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   -q=c                     dS r:  r<  )r>  r  r|  r  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r[   num_classess     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   r   c                     dS r:  r<  )r>  rQ  r  rf  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r
  r4   c                     dS r:  r<  )r>  r  	log_inputr   r  r  r  r  s           r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r4   c                     dS r:  r<  r>  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4         ?UUUUUU?c                     dS r:  r<  r>  lowerr  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  s    wy r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  )r  r  rf  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r  r  r`  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  )r>  r  r  deltas       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rr  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rr  r4      c                     dS r:  r<  )r>  r`  	thresholds      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  r^  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  r  rf  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c
                     dS r:  r<  
anchorpositivenegativerh  r  r  swapr  r  r  s
             r1   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r4   )distance_functionrh  r  r  c                    dS r:  r<  )r  r  r  r	  rh  r  r  s          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  rU  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r[   r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  )r[   r  stdr  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r[   vals     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   fan_in
leaky_reluc                     dS r:  r<  )r[   r  r  nonlinearityr  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  as_tuples     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   )r'  c                    dS r:  r<  )r>  rx  r'  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   froc                     dS r:  r<  r>  r  r|  r  r?  ra   s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  rK  r|  r  r?  ra   s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  s     13 r4   c                     dS r:  r<  )vpowr|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r:  r<  rl  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  input3r  	transposes        r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  r|  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  qr  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  r6  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   V瞯<c                     dS r:  r<  )r>  rconds     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r1  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  )r>  upscale_factors     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  )r>  downscale_factors     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  )r>  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r   r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r'  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rV  r4   c                     dS r:  r<  r!  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  ra   s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  )r>  r0  rw  r{  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>   rY  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  )r>  somer?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   reducedc                     dS r:  r<  )r>  r  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   linearc                     dS r:  r<  r>  r-  r|  r  interpolationr?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rE  r4   c                     dS r:  r<  rK  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  rj  r4   c                     dS r:  r<  )r>  scaleszero_pointsr  ra   s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  ra   s       r1   r@  z'get_testing_overrides.<locals>.<lambda>	  r  r4   c                     dS r:  r<  )r>  ra   reduce_ranges      r1   r@  z'get_testing_overrides.<locals>.<lambda>
  r  r4   c                     dS r:  r<  )r>  r  r  r  r  r  output_scaleoutput_zero_points           r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  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                 r1   r@  z'get_testing_overrides.<locals>.<lambda>        _a r4   c                     dS r:  r<  rX  s                 r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   r<  r   rZ  c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  s
     " r4   r   r   rZ  rZ  c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  s     !# r4   r   r   r   rZ  rZ  rZ  c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  s     !#	 r4   c                     dS r:  r<  rX  s                 r1   r@  z'get_testing_overrides.<locals>.<lambda>%  ra  r4   c                     dS r:  r<  rX  s                 r1   r@  z'get_testing_overrides.<locals>.<lambda>(  ra  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>*  rU  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>+  rV  r4   c                     dS r:  r<  )r>  highra   ro   r`   r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>,  r  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>-  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>.  r;  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>/  rL  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>0  r  r4   c                     dS r:  r<  rm  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>1  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>2  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>3  rS  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>4  rY  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>5  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>6  r  r4   c                     dS r:  r<  )r>  r  r|  maxnormr?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>7  rJ  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r4   c                     dS r:  r<  )r>  shapes     r1   r@  z'get_testing_overrides.<locals>.<lambda>9  rL  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>:  ra  r4   c	                     dS r:  r<  rK  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>;  r  r4   c                     dS r:  r<  rU  s         r1   r@  z'get_testing_overrides.<locals>.<lambda><  r  r4   c	                     dS r:  r<  rK  s	            r1   r@  z'get_testing_overrides.<locals>.<lambda>=  r  r4   c                     dS r:  r<  rU  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>>  r  r4   c                     dS r:  r<  )r>  shiftsr  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>?  r  r4   r   rZ  c                     dS r:  r<  )r>  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>@  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>A  rS  r4   c                     dS r:  r<  r4  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>B  r  r4   c                     dS r:  r<  )r  r  compressed_indices_dtypes      r1   r@  z'get_testing_overrides.<locals>.<lambda>C  r  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>D  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>E  rS  r4   c                     dS r:  r<  )r>  rX  r_  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>F  rY  r4   c                     dS r:  r<  rj  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>G  r  r4   c                     dS r:  r<  r>  r|  r0  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>H  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>I  r  r4   c                     dS r:  r<  )r>  r|  r0  r  r  include_selfs         r1   r@  z'get_testing_overrides.<locals>.<lambda>J  rr  r4   c                     dS r:  r<  )sorted_sequencer>  r  r  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>K  r  r4   r.  c                     dS r:  r<  )r  r  lengthsrz  r  r  unsafes          r1   r@  z'get_testing_overrides.<locals>.<lambda>L  r  r4   c                     dS r:  r<  )r>  r|  r0  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>M  rC  r4   c                     dS r:  r<  )r>  r  r|  r0  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>N  r  r4   c                     dS r:  r<  r>  r  r|  r8  r  steps         r1   r@  z'get_testing_overrides.<locals>.<lambda>O  r  r4   c                     dS r:  r<  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>P  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>Q  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>R  rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>S  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>T  rU  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>U  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>V  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>W  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>X  rL  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>Y  rO  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>Z  rL  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>[  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>\  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>]  r  r4   c                     dS r:  r<  )r  r  r  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>^  rG  r4   c                     dS r:  r<  )r  r  r  r  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>_  r  r4   )stabler?  c                    dS r:  r<  )r>  r|  r-  r  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>`  r'  r4   c                     dS r:  r<  r[   split_size_or_sectionsr|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>a  ro  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>b  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>c  rL  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>d  rQ  r4   c                     dS r:  r<  ry  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>e  r  r4   c                     dS r:  r<  rj  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>f  r9  r4   c                     dS r:  r<  r   s      r1   r@  z'get_testing_overrides.<locals>.<lambda>g  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>h  rA  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>i  rC  r4   reflectc
                     dS r:  r<  )
r>  r  r  r  r  r  pad_moder  r  r  s
             r1   r@  z'get_testing_overrides.<locals>.<lambda>k  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>m  rY  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>n  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>o  rA  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>p  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>q  rO  r4   c                     dS r:  r<  r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>r  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>s  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>t  rO  r4   c                     dS r:  r<  )r  r  cs      r1   r@  z'get_testing_overrides.<locals>.<lambda>u  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>v  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>w  rI  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>x  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>y  rI  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>z  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>{  rI  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>|  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>}  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>~  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  )r>  rF  
compute_uvr?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4      c                     dS r:  r<  )r>  r-  r  Ms       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  )r>  full_matricesr?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r>  dim0r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  )r>  axis0axis1s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r6  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  rW  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  rX  r?  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  s    r r4   c                     dS r:  r<  )r>  r0  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  rz  r|  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  )r  inds     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  r?  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r|  r-  r?  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  rN  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r(  unitriangulars        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r  r  r  r/  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r
  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c
                     dS r:  r<  r  s
             r1   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r=  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  )r>  r|  sizesr  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  sortedreturn_inversereturn_countsr|  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  r;  r<  r|  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r   r4   c                     dS r:  r<  )rz  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r)  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  ry  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  rE  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r4  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  )	conditionr  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r>  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  levels     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )primaltangentrT  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  r  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  )r  rx  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   )implicitc                    dS r:  r<  )r  rx  r`  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r|  r8  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  rx  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  )r  r|  r0  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r|  r8  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r!  r4   c                     dS r:  r<  )r  
split_sizer|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  split_sizesr|  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r+  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>   r  r4   c                     dS r:  r<  r+  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>	  rS  r4   c                     dS r:  r<  r+  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>
  rU  r4   c                     dS r:  r<  r  ra   s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  r  	dimensionrx  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rJ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  rX  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rC  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r{  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>   rI  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>!  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>"  rO  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>#  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>$  rL  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>%  rS  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>&  rS  r4   c                     dS r:  r<  )r  arrays     r1   r@  z'get_testing_overrides.<locals>.<lambda>'  r  r4   c                     dS r:  r<  )r  idxs     r1   r@  z'get_testing_overrides.<locals>.<lambda>(  rQ  r4   c                     dS r:  r<  )r  memos     r1   r@  z'get_testing_overrides.<locals>.<lambda>)  rC  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>*  rO  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>+  rI  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>,  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>-  rO  r4   c                     dS r:  r<  )r  format_specs     r1   r@  z'get_testing_overrides.<locals>.<lambda>.  r  r4   c                     dS r:  r<  )r  protos     r1   r@  z'get_testing_overrides.<locals>.<lambda>/  rY  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>0  rD  r4   )tensor_contentsc                    dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>1  rJ  r4   c                     dS r:  r<  )r  r  r!  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>2  rU  r4   c                     dS r:  r<  )r  ds     r1   r@  z'get_testing_overrides.<locals>.<lambda>3  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>4  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>5  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>6  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>7  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>8  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>9  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>:  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>;  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda><  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>=  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>>  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>?  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>@  rQ  r4   c                     dS r:  r<  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r1   r@  z'get_testing_overrides.<locals>.<lambda>A  r  r4   c                     dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>B  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>C  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>D  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>E  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>F  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>G  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>H  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>I  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>J  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>K  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>L  rY  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>M  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>N  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>O  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>P  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>Q  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>R  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>S  rY  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>T  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>U  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>V  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>W  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>X  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>Y  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>Z  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>[  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>\  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>]  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>^  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>`  rQ  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>a  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>b  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>c  rG  r4   c                     dS r:  r<  )r  ra   non_blockingr0   s       r1   r@  z'get_testing_overrides.<locals>.<lambda>d  r!  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>e  r;  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>f  r;  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>g  rI  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>h  rI  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>i      " r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>j  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>k  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>l  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>m  r  r4   c                     dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>n  rR  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>o  rO  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>p  rO  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>q  rS  r4   c                     dS r:  r<  )r  orderellipsis_idxs      r1   r@  z'get_testing_overrides.<locals>.<lambda>r  rR  r4   c                     dS r:  r<  )r  callables     r1   r@  z'get_testing_overrides.<locals>.<lambda>s  rQ  r4   c                     dS r:  r<  re  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>t  r  r4   c                     dS r:  r<  re  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>u  r  r4   c                     dS r:  r<  )r  gradientretain_graphcreate_graphrI  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>v  s    ik r4   c                     dS r:  r<  r  rh   s     r1   r@  z'get_testing_overrides.<locals>.<lambda>w  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>x  r@  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>y  r@  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>z  r@  r4   )r  c                    dS r:  r<  )r  mediansigmar  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>{  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>|  rI  r4   c                     dS r:  r<  )r  	coalesceds     r1   r@  z'get_testing_overrides.<locals>.<lambda>}  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>~  ra  r4   c                     dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  r  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                    dS r:  r<  )r  r_  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  rf  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rt  r4   c                    dS r:  r<  )r  r  r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r[   s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                    dS r:  r<  )r  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r+  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  )r  r[   r  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  rl  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  )r  rX  assigns      r1   r@  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r:  r<  )r  r~  r8  r  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  ro  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r:  r<  r+  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  rz  r[   r{  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                    dS r:  r<  )r  from_tor  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r:  r<  r  streams     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  hooks     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rI  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  names     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r0  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rQ  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  rw  r  rx  r  s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r|  r0  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rp  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  r|  r8  r  r  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  )r  r  accumulate_matchess      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  size1size2	dense_dims       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  ra  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r:  r<  )r  rk  rl  r`  r_  r?  s         r1   r@  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  )r  repss     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  ra   r  copyrh   s        r1   r@  z'get_testing_overrides.<locals>.<lambda>  s    ln r4   )masked_gradc                    dS r:  r<  r  ra   rp  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r@  r4   c                     dS r:  r<  rr  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  r}  s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  rR  r4   c                     dS r:  r<  )r  rB  rC  s      r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  r  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  rL  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r:  r<  rE  s     r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  rU  r4   c                     dS r:  r<  )r  r  r   drivers       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  )r  r`   r  r0   s       r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r:  r<  rY  s    r1   r@  z'get_testing_overrides.<locals>.<lambda>  r  r4   is______i__rbitwise_N)rZ  rZ  N)rZ  N)r}  r~  F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)r  r  )r  F)FN)NNN)rZ  NN)r<  F)NrZ  r   rZ  rZ  )NrZ  r   r   rZ  rZ  )r   NNr  )rZ  r~  )r;  N)r   r  Fr:  )rZ  r;  NNN)r   r   rZ  )r   r  r;  )r<  )r  N)NNr  FF)Nr<  Fr  FNN)NNNF)FF)Nr;  N)Nr  N)r   r;  )TT)NF)NNrZ  )NNr}  T)r\  )r  NNr  )re  r   r   N)re  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rZ  F)Nr   rZ  FF)NFNN)r;  FN)r9  NNN)NNr}  )r<  NFN)r\  FF)Nr   FTN)NNFrb  r}  )NNNr  )NNrk  Nr  r9  )r\  TF)	NNr<  Fr  FNFN)rZ  r   rZ  )NNFN)Fr  r  )r  )r;  )r  r   N)rZ  Fr  r;  )r  r  F)NNNNTrb  r}  )NNr  NNF)NNr  )r  F)r  r  r  )Nr  N)Nr   N)TNTNFNNNNNNF)rZ  r  NNNr  )NNrk  Nr  )r<  rZ  r  N)r   r   )r  r  F)TFNr~  Nr  )Nr  )r  r  FF)Nr9  )NNr  r  )r  r  )rZ  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)rZ  r  )r.  NNNr   F)r   NNrZ  )r;  F)NNNTr  FTN)TTN)r  r<  N)r<  N)TFF)TFFN)rZ  r<  )Nr   NN(  r!   r6   rG  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshaddaddbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullir   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffr  diagonal_scatteras_strided_scatterdigammadistdivdivide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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igammacrC  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr.  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr-  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm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_hotrQ  pairwise_distancepoisson_nll_lossprelurelurelu6rms_normrreluselusilumishscaled_dot_product_attentionsmooth_l1_loss
huber_losssoft_margin_losssoftmaxsoftminsoftplus
softshrinksoftsign
tanhshrinkr  triplet_margin_loss!triplet_margin_with_distance_lossunfoldr   uniform_normal_	constant_kaiming_uniform_nonzerononzero_staticargwherer  vector_normmatrix_normnorm_except_dimnuclear_normr#  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar  	ones_liker"  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2deg	rand_likerandint_like
randn_likeravelrB  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_ite	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer(  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr'  HmTmH_backward_hooks_post_accumulate_grad_hooksr4  _cdatar5  r6  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precisionr  r`   ra   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itemsizero   rM  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_rr   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrrd  r|  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_rZ   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_ri   random_record_streamrefine_namesregister_hook"register_post_accumulate_grad_hookrenamerepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_shortrx  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizerC  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)r6   retprivateuse1_backend_nameret2ignoredr  r!  r  subnamerM  r"   s              r1   r   r     sS_   6 \Fw%	---w%222w% 	!#@#@w% 	!#A#A	w%
 	
...w% 	''w% 	000w% 	///w% 	111w% 		444w% 	QQQw% 	LLLw% 	LLLw% 	LLLw% 	JJJw%  	
KKK!w%" 	#%J%J#w% w%$ 		---%w%& 	XXX'w%( 	FFF)w%* 	
...+w%, 	
...-w%. 	JJJ/w%0 	///1w%2 		FFF3w%4 	&&5w%6 	&&7w%8 	1119w%: 	
...;w%< 	22=w%> 	000?w%@ 	///Aw%B 	111Cw%D 	
...Ew% w% w%F 	000Gw%H 	666Iw%J 	888Kw%L 	///Mw%N 	111Ow%P 	--Qw%R 	--Sw%T 	--Uw%V 	xxxWw%X 	RRRYw%Z 	{{[w%\ 	'){){]w%^ 	(*u*u_w%` 	 Q Qaw%b 	%'v'vcw%d 	1  4C  4Cew%f 	 5 5gw% w% w%h 	%'\'\iw%j 	CCCkw%l 	??mw%n 	.tttqw%t 	CCCuw%v 	>>>ww%x 	<<<yw%z 	555{w%| 	;;;}w%~ 	<<<w%@ 	 "C"C"CAw%B 	!#D#D#DCw%D 	--Ew%F 		333Gw%H 	!4!4Iw%J 	11Kw%L 	]]]Mw% w% w%N 	11Ow%P 		666Qw%R 	999Sw%T 	>>>Uw%V 	aaaWw%X 	
...Yw%Z 	
>>>[w%\ 	$::::]w%^ 	77_w%` 	???aw%b 	999cw%d 	 "P"P"Pew%f 	 G G Ggw%h 	NNNiw%j 	&(Y(Ykw%l 	444mw%n 	CCCow% w% w%p 	
BBBqw%r 	888sw%t 	888uw%v 	888ww%x 		OOOyw%z 	%%{w%| 	III}w%~ 	,,w%@ 	999Aw%B 	((Cw%D 	555Ew%F 	
...Gw%H 	777Iw%J 	666Kw%L 	555Mw%N 	===Ow%P 	dddQw% w% w%R 	dddSw%T 	dddUw%V 	wwWw%X 	===Yw%Z 	  !A  !A  !A[w%\ 	  !A  !A  !A]w%^ 	  !A  !A  !A_w%` 	((aw%b 		---cw%d 	#  &C  &C  &Cew%f 	
...gw%h 	!C!C!Ciw%j 	--kw%l 	@@@mw%n 	EEEow%p 	xxxsw%v 	555ww% w% w%x 	555yw%z 	BBB{w%| 	AAA}w%~ 	"$@$@$@w%@ 	;;;Aw%B 	111Cw%D 	**Ew%F 		##Gw%H 	**Iw%J 	&&Kw%L 	
:::Mw%N 	@@@Ow%P 	222Qw%R 	
VVVSw%T 	BBBUw%V 	KKKWw%X 	 O O OYw% w% w%Z 	 "Y"Y"Y[w%\ 	111]w%^ 	
000_w%` 		HHHaw%b 	KKKcw%d 		444ew%f 	@@@gw%h 	
**iw%j 	
))kw%l 	;;mw%n 	222ow%p 	444qw%r 	888sw%t 	???uw%v 	CCCww%x 	44yw%z 	|||}w% w% w%@ 	 j  j  jCw%F 	eeeGw%H 	333Iw%J 	,,Kw%L 		---Mw%N 	
...Ow%P 	000Qw%R 		---Sw%T 	
...Uw%V 	///Ww%X 	.0o0oYw%Z 	-/h/h[w%\ 	- C  C  C_w%b 	')N)Ncw%d 	79^9^ew%f 	')})}gw%h 	7``kw% w% w%n 	+-=-=ow%p 	*,<,<qw%r 	*,B,Bsw%t 	#%?%?uw%v 	99ww%x 		CCCyw%z 		CCC{w%| 		DDD}w%~ 		CCCw%@ 		DDDAw%B 		JJJCw%D 		KKKEw%F 		DDDGw%H 		EEEIw%J 		EEEKw%L 		FFFMw%N 		FFFOw% w% w%P 		GGGQw%R 		IIISw%T 		JJJUw%V 		JJJWw%X 		KKKYw%Z 		666[w%\ 		777]w%^ 		BBB_w%` 		---aw%b 	@@@cw%d 	
**ew%f 	&&gw%h 	&&iw%j 	QQQkw%l 	///mw%n 	33ow%p 	???qw% w% w%r 	
555sw%t 	
...uw%v 	///ww%x 	t4PUP]fjz  D  D  Dyw%z 	&(H(H{w%| 	\\\}w%~ 	OOOw%@ 		444Aw%B 	333Cw%D 	>>>Ew%F 	///Gw%H 	,,,Iw%J 	666Kw%L 	555Mw%N 		333Ow%P 	NNNQw%R 	ccSw% w% w%T 	ffUw%V 	ffWw%X 	mmmYw%Z 		ss[w%\ 	NNN]w%^ 	333_w%` 	888aw%b 	555cw%d 	;;;ew%f 	"$z$z$zgw%h 	GGGiw%j 	mmmkw%l 	YYYmw%n 	(*?*?ow%p 	444qw%r 	;;sw%t 	222uw% w% w%v 	666ww%x 	777yw%z 	888{w%| 	
...}w%~ 	==w%@ 	>>Aw%B 	LLLCw%D 	BBBEw%F 	==Gw%H 	\\\Iw%J 	))Kw%L 	
GGGMw%N 	&&Ow%P 	''Qw%R 	222Sw%T 	222Uw%V 	ttYw% w% w%\ 	((]w%^ 	111_w%` 	444aw%b 	KKKcw%d 	**ew%f 	''gw%h 	&&iw%j 	..kw%l 	,,mw%n 	!1!1ow%p 	**qw%r 	33sw%t 	))uw%v 	WWWww%x 	%%yw%z 	 d  d  d}w%@	 	rrrA	w% w% w%B	 	
++C	w%D	 	NNNE	w%F	 	"$c$c$cG	w%H	 	!L!L!LI	w%J	 	 S S SK	w%L	 	sssM	w%N	 		444O	w%P	 	666Q	w%R	 	333S	w%T	 	;;;U	w%V	 	
;;;W	w%X	 	000Y	w%Z	 	  K  K  K[	w%\	 		---]	w%^	 	<<<_	w%`	 	///a	w%b	 	///c	w% w% w%d	 	
...e	w%f	 	:::g	w%h	 	;;;i	w%j	 	&&k	w%l	 	...m	w%n	 	<<<o	w%p	 	555q	w%r	 	;;;s	w%t	 	<<<u	w%v	 	///w	w%x	 	IIIy	w%z	 	
ss{	w%|	 	OOO}	w%~	 	333	w%@
 	
555A
w%B
 	EEEC
w%D
 	BBBE
w% w% w%F
 	!  $A  $A  $AG
w%H
 	88I
w%J
 	<<K
w%L
 	===M
w%N
 	777O
w%P
 	???Q
w%R
 	 F F FS
w%T
 	!#]#]#]U
w%V
 	[[[W
w%X
 	>>>Y
w%Z
 	//[
w%\
 	!#@#@#@]
w%^
 	 "M"M"M_
w%`
 	 < < <a
w%b
 	**c
w%d
 	!1!1e
w%f
 		---g
w% w% w%h
 	888i
w%j
 	
555k
w%l
 	lllm
w%n
 	lllo
w%p
 	lllq
w%r
 	%tttu
w%x
 	
...y
w%z
 	VVV{
w%|
 	000}
w%~
 	333
w%@ 	55Aw%B 		---Cw%D 	888Ew%F 	
555Gw%H 	ttKw%N 	 "}"}Ow%P 	)+v+vQw% w% w%R 	%'h'hSw%T 	*wwWw%Z 	*gg]w%` 	 d  dcw%f 	222gw%h 	
EEEiw%j 	<<kw%l 	==mw%n 	AAAow%p 		444qw%r 	999sw%t 	UUUuw%v 	111ww%x 	++yw%z 	::{w%| 	WWW}w%~ 	!s!sw% w% w%@ 	&(_(_Aw%B 	88Cw%D 	!f!f!fEw%F 	!V!VGw%H 	UUUIw%J 	$&>&>Kw%L 	333Mw%N 	:::Ow%P 		---Qw%R 	222Sw%T 	:::Uw%V 	/1N1NWw%X 	/1N1NYw%Z 	/1d1d1d[w%\ 	<>q>q>q]w%^ 	/1d1d1d_w%` 	<>q>q>qaw% w% w%b 	/1d1d1dcw%d 	<>q>q>qew%f 	')S)S)Sgw%h 	)+a+a+aiw%j 	& B  B  Bmw%p 	& B  B  Bsw%v 	&xxxyw%| 	$&R&R&R}w%~ 	0cccAw%D 	<tttGw%J 	 "L"L"LKw%L 	1iiiOw%R 	) L  L  LUw%X 	$xxx[w%^ 	#%Z%Z%Z_w%` 	%'\'\'\aw%b 	%'\'\'\cw% w% w%d 	%'\'\'\ew%f 	!K!K!Kgw%h 	%|||kw%n 	) J  J  Jqw%t 	13i3i3iuw%v 	 "m"m"mww%x 	1zzz{w%~ 	>zzzAw%D 	1zzzGw%J 	>zzzMw%P 	-/u/u/uQw%R 	 "F"F"FSw%T 	!9!9!9Uw%V 	')z)z)zWw%X 	&(g(g(gYw%Z 	*,c,c,c[w%\ 	&(C(C(C]w% w% w%^ 	$&`&`&`_w%` 	0bbbcw%f 	) I  I  Iiw%l 	' M  M  Mow%r 	"  %A  %A  %Asw%t 	#%o%o%ouw%v 	&(m(m(mww%x 	&(\(\(\yw%z 	"$G$G$G{w%| 	/1g1g1g}w%~ 	')^)^)^w%@ 	&(8(8Aw%B 	%'m'm'mCw%D 	%'m'm'mEw%F 	%'m'm'mGw%H 	/iiiKw%N 	&tttQw% w% w%T 	3tttWw%Z 	&ttt]w%` 	3tttcw%f 	&tttiw%l 	3tttow%r 	(*z*z*zsw%t 	(*z*z*zuw%v 	(*z*z*zww%x 	$&p&p&pyw%z 	8 _  _  _}w%@ 	-tttCw%F 	2VVVIw%L 	7cccOw%R 	$vvvUw%X 	%'X'X'XYw%Z 	#%F%F%F[w%\ 	!P!P!P]w% w% w%^ 	-/a/a/a_w%` 	,}}}cw%f 	!#;#;gw%h 	 "A"A"Aiw%j 	!#B#B#Bkw%l 	$&_&_&_mw%n 	!#y#y#yow%p 	 "A"A"Aqw%r 	 "A"A"Asw%t 	 "A"A"Auw%v 	8:u:u:uww%x 	*  -A  -A  -Ayw%z 	&(](](]{w%| 	,.x.x.x}w%~ 	#%Z%Z%Zw%@ 	#%Z%Z%ZAw%B 	$&L&L&LCw% w% w%D 	&(C(C(CEw%F 	$&6&6Gw%H 	&(8(8Iw%J 	%'X'X'XKw%L 	/ L  L  LOw%R 	=DHQT[`lrvvvvUw%X 	"$b$b$bYw%Z 	 O O O[w%\ 	SSS]w%^ 	!7!7_w%` 	&(x(x(xaw%b 	777cw%d 	FFFFew%f 	((gw%h 	
\\\iw%j 	dddkw%l 	 "h"h"hmw% w% w%n 	  +/#3 #3 #3ow%v 	999ww%x 	dddyw%z 	%%{w%| 	**}w%~ 	QQQw%@ 	!S!S!SAw%B 	++Cw%D 	IIIEw%F 	***Gw%H 	555Iw%J 	IIIKw%L 	==Mw%N 	AAOw%P 	777Qw%R 	 Y YSw%T 	666Uw% w% w%V 	222Ww%X 	--Yw%Z 	ddd[w%\ 		777]w%^ 	
000_w%` 		DDDaw%b 	 "2"2cw%d 	"$4$4ew%f 	')9)9gw%h 	''iw%j 	,,kw%l 	777mw%n 	CCCow%p 	fffqw%r 	iiisw%t 	"$V$Vuw%v 	!#M#Mww% w% w%x 	)+P+Pyw%z 	"$s$s{w%| 	  a  aw%B 	! a  aEw%H 	"J# # #Kw%R 	"L # # #Uw%^ 	"O 	# # #aw%l 	% a  aow%r 	% a  auw%x 	111yw%z 	ddd{w%| 	d5=Y]mrvvv}w%~ 	eeew%@ 	%%Aw%B 	
...Cw%D 	
555Ew%F 	FFFGw% w% w%H 	,,Iw%J 	//Kw%L 	444Mw%N 	
333Ow%P 	:::Qw%R 	AAASw%T 	!;!;!;Uw%V 	..Ww%X 	QQQYw%Z 	xx[w%\ 	SSS]w%^ 	xx_w%` 	SSSaw%b 	
777cw%d 	777ew%f 	///gw%h 	555iw% w% w%j 	PPkw%l 	bbbmw%n 	///ow%p 	
444qw%r 	MMMsw%t 	88uw%v 	<<ww%x 	ZZZyw%z 	eee{w%| 	|||}w%~ 	22w%@ 	??Aw%B 	WWWCw%D 	WWWEw%F 	
333Gw%H 	111Iw%J 	
...Kw% w% w%L 	111Mw%N 		---Ow%P 		---Qw%R 	
...Sw%T 	
...Uw%V 	''Ww%X 	..Yw%Z 		))[w%\ 	
**]w%^ 	888_w%` 	@@@aw%b 	WWWcw%d 	
YeQUYYYYew%f 	EEEgw%h 	 P P Piw%j 	
...kw%l 	000mw% w% w%n 	;;;ow%p 	OOOqw%r 	888sw%t 		---uw%v 	222ww%x 	
 j  j  j{w%~ 		444w%@ 	999Aw%B 		---Cw%D 	))Ew%F 	''Gw%H 	Iw%J 	Kw%L 	''Mw%N 	))Ow%P 	))Qw%R 	((Sw% w% w%T 	))Uw%V 	((Ww%X 	))Yw%Z 	(([w%\ 	))]w%^ 	))_w%` 	))aw%b 	))cw%d 	000ew%f 		IIIgw%h 	AAAiw%j 	HHHkw%l 	888mw%n 	44ow%p 	66qw%r 	//sw%t 	!1!1uw% w% w%v 	!1!1ww%x 	!1!1yw%z 	!1!1{w%| 	,.K.K.K}w%~ 	,.K.K.Kw%@ 	,.K.K.KAw%B 	,.K.K.KCw%D 	//Ew%F 	,,Gw%H 	++Iw%J 	,,Kw%L 	--Mw%N 	..Ow%P 	,,Qw%R 	--Sw%T 	--Uw%V 	 A A AWw% w% w%X 	!B!B!BYw%Z 	//[w%\ 	*,I,I,I]w%^ 	+-J-J-J_w%` 	**aw%b 	++cw%d 	**ew%f 	++gw%h 	+-J-J-Jiw%j 	+-J-J-Jkw%l 	--mw%n 	 0 0ow%p 	!#D#D#Dqw%r 	--sw%t 	!O!O!Ouw%v 	(*:*:ww%x 	(*:*:yw% w% w%z 	(*:*:{w%| 	(*:*:}w%~ 	"$7$7w%@ 	,,Aw%B 	--Cw%D 	!>!>!>Ew%F 	++Gw%H 	--Iw%J 	/1A1AKw%L 	/1A1AMw%N 	46S6S6SOw%P 	46S6S6SQw%R 	46S6S6SSw%T 	46S6S6SUw%V 	,,Ww%X 	@@@Yw%Z 	)+;+;[w% w% w%\ 	@@@]w%^ 	>>>_w%` 	<<<aw%b 	!!cw%d 	
++ew%f 	KKKgw%h 		---iw%j 	
...kw%l 	 3 3 3mw%n 	 "<"<"<ow%p 	:::qw%r 	HHHsw%t 	JJJuw%v 	
**ww%x 	
KKKyw%z 	%%{w%| 	55}w% w% w%~ 	111w%@ 	555Aw%B 	 e e eCw%D 	%'a'a'aEw%F 	
:::Gw%H 	! L  L  LKw%N 	
:::Ow%P 	22Qw%R 	///Sw%T 	---Uw%V 	<<Ww%X 	hhhYw%Z 	 "g"g"g[w%\ 	66]w%^ 	;;;_w%` 	LLLaw%b 	%'W'W'Wcw% w% w%d 	888ew%f 	111gw%h 		---iw%j 	222kw%l 	;;mw%n 	222ow%p 	999qw%r 	%'`'`sw%t 	1ooww%z 	eee{w%| 	55}w%~ 	@@w%@ 	Aw%B 	"OOCw%D 	//Ew%F 	ooGw%H 	QQQIw% w% w%J 	')>)>Kw%L 	FFFMw%N 	%CCCCOw%P 	>>Qw%R 	11Sw%T 	!#@#@Uw%V 	66Ww%X 	??Yw%Z 	NNN[w%\ 	<<<]w%^ 	#%H%H%H_w%` 	00aw%b 	oocw%d 	99ew%f 	22gw%h 	__iw%j 	OOkw% w% w%l 	OOmw%n 	??ow%p 	qw%r 	sw%t 	uw%v 	ww%x 	111yw%z 	//{w%| 	AA}w%~ 	//w%@ 	33Aw%B 	44Cw%D 	44Ew%F 	22Gw%H 	33Iw%J 	33Kw%L 	11Mw% w% w%N 	22Ow%P 	22Qw%R 	11Sw%T 	22Uw%V 	22Ww%X 	..Yw%Z 	--[w%\ 	..]w%^ 	//_w%` 	OOaw%b 	00cw%d 	ew%f 	33gw%h 	iw%j 	??kw%l 	..mw%n 	//ow% w% w%p 	//qw%r 	55sw%t 	00uw%v 	22ww%x 	yw%z 	{w%| 	//}w%~ 	w%@ 	77Aw%B 	44Cw%D 	__Ew%F 	AAAAGw%H 	11Iw%J 	//Kw%L 	//Mw%N 	//Ow%P 		??Qw% w% w%R 		??Sw%T 	&Uw%V 	*2OOWw%X 	ooYw%Z 	[w%\ 	__]w%^ 	oo_w%` 	aw%b 	cw%d 	ew%f 	-/p/pgw%h 	*,V,Viw%j 	__kw%l 	mw%n 	ooow%p 	qw%r 	sw% w% w%t 	uw%v 	ww%x 	yw%z 	{w%| 	#__}w%~ 	w%@ 	Aw%B 	Cw%D 	 //Ew%F 	Gw%H 	 //Iw%J 	#__Kw%L 	 //Mw%N 	$ooOw%P 	 //Qw%R 	Sw%T 	Uw% w% w%V 	__Ww%X 	ooYw%Z 	[w%\ 	__]w%^ 	 //_w%` 	$ooaw%b 	oocw%d 	ew%f 	__gw%h 	__iw%j 	'/kw%l 	NNNmw%n 	ooow%p 	ooqw%r 	sw%t 	uw%v 	__ww% w% w%x 	__yw%z 	OO{w%| 	__}w%~ 	OOw%@ 	==Aw%B 	Cw%D 	Ew%F 	//Gw%H 	==Iw%J 	00Kw%L 	88Mw%N 	99Ow%P 	kkkQw%R 	E4IMMMSw%T 	0EIIIUw%V 	0EIIIWw%X 	0EIIIYw% w% w%Z 	MTMMMM[w%\ 	]w%^ 	66_w%` 	e6MQQQaw%b 	>>>cw%d 	
u/DHHHew%f 	0EIIIgw%h 	0EIIIiw%j 	
u/DHHHkw%l 	
u/DHHHmw%n 	ow%p 	//qw%r 	!O!O!Osw%t 	
OOuw%v 	//ww%x 	%2GKKKyw%z 	53HLLL{w% w% w%| 	__}w%~ 	,,w%@ 	00Aw%B 	HHHHHCw%D 	,,Ew%F 	55Gw%H 	1FJJJIw%J 	%2GKKKKw%L 	@@@@Mw%N 	??Ow%P 	0EIIIQw%R 	1FJJJSw%T 	//Uw%V 	Ww%X 	
u/DHHHYw%Z 	__[w%\ 	oo]w% w% w%^ 	___w%` 	//aw%b 	11cw%d 	//ew%f 	__gw%h 	MTMMMMiw%j 	00kw%l 	0EIIImw%n 	66ow%p 	55qw%r 		((sw%t 	@@@uw%v 	EEww%x 	??yw%z 	{w%| 	"OO}w%~ 	-w% w% w%@  	%A w%B  	C w%D  	ooE w%F  	,,G w%H  	??I w%J  	GGGK w%L  	M w%N  	LDLLLLO w%P  	55Q w%R  	33S w%T  	33U w%V  	13H3HW w%X  	,,Y w%Z  	--[ w%\  	BBB] w%^  	11_ w%`  	--a w% w% w%b  	--c w%d  	00e w%f  	 "8"8g w%h  	OOi w%j  	[[[k w%l  	??m w%n  	ooo w%p  	1FJJJq w%r  	__s w%t  	WWWu w%v  	??w w%x  	11y w%z  	&(W(W(W{ w%|  	GG} w%~  	')Q)Q w%@! 	OOOA!w%B! 	C!w% w% w%D! 	E!w%F! 	G!w%H! 	__I!w%J! 	11K!w%L! 	++M!w%N! 		EUZUjnnnO!w%P! 	IIIIIQ!w%R! 	GGGS!w%T! 	//U!w%V! 	W!w%X! 	//Y!w%Z! 	..[!w%\! 	==]!w%^! 	777_!w%`! 	a!w%b! 	++c!w%d! 	..e!w% w%f! 	oo777 //FFFm!w% w%Ct! 	(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Jr4   
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/   r0   relevant_argsr
  r"   wrappeds      r1   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((((r4   )	functoolsr   )r"   r
  r
  s   `@r1   rY  z"wrap_torch_function.<locals>.inner&  sD    				) 	) 	) 	) 	) 	) 
		) r4   r<  )r
  rY  s   ` r1   r   r     s#    0	 	 	 	 	 Lr4   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   )r\
  r!   _C_is_torch_function_enabledsetr
  r   _disabled_torch_function_implr  r
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer0  iold_args	            r1   _get_overloaded_argsr
  4  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$,: #&%r4   
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<  )r\
  ).0r
  s     r1   
<listcomp>z)handle_torch_function.<locals>.<listcomp>  s    EEEcS		EEEr4   z nor in mode )r
  tuplemapr\
  r   _pop_mode_temporarilyr   NotImplementedr
  r
  r!   r
  r
  r+   warnDeprecationWarning
__module__r
  _get_current_function_mode	TypeError)r
  r
  r/   r0   r
  typesr  resultoverloaded_argtorch_func_method	func_namer  s               r1   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   r6   r  r   r	  r
  endswithislowerr
  object
isinstancer
  
ModuleType
__future___Featurer  r
  r'  __set__r   r   formatr
  r  )
overridable_funcsr0  tested_namespacesnamespace_str	namespacens_funcsr  r(   r"   r  s
             r1   _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##r4   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<  r4   r1   r   r   b  s     &''**r4   c                     t          | t          j        j        t          j        j        f          rt          |           S t                      d                             |           S )a&  Get a human readable string name for a function passed to
    __torch_function__

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

    Returns
    -------
    str
        Name of the function; if eval'ed it should give back the input
        function.
    rZ  )r  r!   _ops
OpOverloadOpOverloadPacketstrr  get)fs    r1   r   r   o  sO      !ej+UZ-HIJJ 1vv%''*..q111r4   c                  `    t                      } t          | t          j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r
  r!   r6   )r  methodss     r1   _get_tensor_methodsr#    s*     233#EL122GNr4   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    r1   r   r     s"    . &(((FDMY,FFr4   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   )r\
  r!   r6   r
  )inps    r1   r   r     s(    D 99$J5I(J(JJr4   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!)
    rY  r%   Nc                     d S r  r<  rY  s    r1   r   zTorchFunctionMode.__init__  s    r4   r<  c                     t           r  )NotImplementedErrorr  r"   r
  r/   r0   s        r1   r   z$TorchFunctionMode.__torch_function__  s    !!r4   c                 $    t          |            | S r  )
_push_moderY  s    r1   	__enter__zTorchFunctionMode.__enter__  s    4r4   c                 "    t                       d S r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r1   __exit__zTorchFunctionMode.__exit__  s    r4   c                 >    t          j        d            | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`)r+   r
  )clsr/   r0   instances       r1   pushzTorchFunctionMode.push  s4    ^	
 	
 	
 3'''r4   )r%   Nr<  N)r
  r
  __qualname____doc____annotations__r   r   r/  r5  classmethodr9  r<  r4   r1   r(  r(    s          <    " " " "       [  r4   r(  c                  R    t                      } | dk    rt          | dz
            nd S )Nr   rZ  )r   r   	stack_lens    r1   r
  r
    s-    )++I4=MM!)a-000tKr4   c                  P    t                      } d t          |           D             S )Nc                 ,    g | ]}t          |          S r<  )r   )r
  r
  s     r1   r
  z4_get_current_function_mode_stack.<locals>.<listcomp>
  s!    @@@!"1%%@@@r4   )r   r   r@  s    r1    _get_current_function_mode_stackrD    s)    )++I@@uY/?/?@@@@r4   c                 $    t          |            d S r  )r   )r  s    r1   r.  r.    s    !$'''''r4   c                  "    t                      } | S r  )r   olds    r1   r1  r1    s    
#
%
%CJr4   c               #   v   K   t                      } 	 | V  t          |            d S # t          |            w xY wr  )r1  r.  rG  s    r1   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<  r,  s        r1   r   z(BaseTorchFunctionMode.__torch_function__   s!    >FtT$V$$$r4   r:  )r
  r
  r;  r   r<  r4   r1   rK  rK    s(        % % % % % %r4   rK  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<  r4   r1   r   r   &  s       
	+	+	-	-  	EEEDDDDD	                 s   ;(;*;??)r    r!   r  )9r<  r  r  
contextlibr
  r
  r+   r   typingr   r   r   r   r	   r
   r   r   r!   torch._Cr   r   r   r   r   r   r   r   r   r  r  r3   	lru_cacher   r7  r   r   r
  r   r   r   r   r  r   r   r#  ro
  r   r   r(  r
  rD  r.  r1  contextmanagerr
  rK  r   r<  r4   r1   <module>rT     sN   ,                         H H H H H H H H H H H H H H H H H H H H 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
    1   
     	       F TSs8} S S S  Sl Tc(m    2 TwtHh$67 w w w  wt##H # # # #P *.K KC=K3%+&K 
#YK K K K\UUC=U
 	U U U Up ![  . ';	 	  *k 	   TS$	$sDN"
#T(C-%8
89S$ S$ S$ S$l 	+4T(^(;#< 	+ 	+ 	+ 	+ 2 2 2( T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
( ( (  
   % % % % %- % % %     r4   