
    #h             	       $   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mZmZm	Z	m
Z
mZmZmZmZ ddlmZ ddl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)DictSetListAnyCallableIterableTypeTuplewraps)	_has_torch_function_has_torch_function_unary_has_torch_function_variadic_add_docstr_push_on_torch_function_stack_pop_torch_function_stack_get_function_stack_at_len_torch_function_stack_is_torch_function_mode_enabled)
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     O/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperK   s    $&& 	) 	)#H{EZ`aaaa4(((	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)s   %AAAr   )r"   r#   r$   r2   s   ``` r1   _disable_user_warningsr3   6   sA    * 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        jO        t           jN        jN        t           jN        j        t           jN        jP        t           jN        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{        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        | 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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dynamic_dim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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_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_make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activationr6   s    r1   r   r   S   s   ( \Fzzz 	z 	%	z
 	 z 	z 	z 	z 	z 	
z 	
z 	
z 	z 	
z 	z  	%!z" 	#z$ 	%z& 	'z( 	*)z* 	 +z, 	*-z. 	/z0 	*1z2 	3z4 	5z6 	 7z8 	!9z: 	 ;z< 	%=z> 	?z@ 	
AzB 	CzD 	EzF 	GzH 	IzJ 	KzL 	MzN 	OzP 	QzR 	SzT 	UzV 	WzX 	YzZ 	[z\ 	]z^ 	_z` 	azb 	czd 	ezf 	gzh 	izj 	kzl 	mzn 	'ozp 	qzr 	szt 	uzv 	wzx 	yzz 	{z| 	}z~ 	z@ 	AzB 	)CzD 	EzF 	GzH 	)IzJ 	$KzL 	(MzN 	 OzP 	!QzR 	SzT 	UzV 	WzX 	YzZ 	 [z\ 	]z^ 	_z` 	'azb 	czd 		ezf 		gzh 		izj 	kzl 	
mzn 	
ozp 	qzr 	szt 	uzv 	wzx 	yzz 	({z| 	 }z~ 	z@ 	AzB 	,CzD 	EzF 	GzH 	
IzJ 	KzL 	
MzN 	OzP 	QzR 	SzT 	UzV 	WzX 	YzZ 	[z\ 	&]z^ 	_z` 	azb 	czd 	ezf 	!gzh 	*izj 	kzl 	mzn 	ozp 	qzr 	,szt 	.uzv 	$wzx 	-yzz 	,{z| 	.}z~ 	4z@ 	7AzB 	1CzD 	#EzF 	'GzH 	 IzJ 	+KzL 	*MzP 	$QzT 	UzV 	WzX 	YzZ 	[z\ 	]z^ 	$_z` 	#azb 	%czd 	$ezf 	 gzh 	izj 	%kzl 	mzn 	ozp 	"qzr 	!szt 	"uzv 	&wzx 	%yzz 	&{z| 	%}z~ 	&z@ 	%AzB 	$CzD 	$EzF 	$GzH 	$IzJ 	$KzL 	$MzN 	$OzP 	$QzR 	(SzT 	(UzV 	'WzX 	(YzZ 	%[z\ 	!]z^ 	2_z` 	*azb 	;czd 	*ezf 	*gzh 	*izj 	*kzl 	mzn 	$ozp 	qzr 	szt 	uzv 	wzx 	
yzz 	
{z| 	}z~ 	z@ 	-AzB 	CzD 	EzF 	GzH 	IzJ 	KzL 	MzN 	 OzP 	QzR 	SzT 	!UzV 	!WzX 	YzZ 	[z\ 	]z^ 	_z` 	azb 	
czd 	ezf 	gzh 	
izj 	kzl 	mzn 	 ozp 	qzr 	szt 	uzv 	wzx 	yzz 	{z| 	}z~ 	z@ 	AzB 	CzD 	EzF 	GzH 	IzJ 	KzL 	MzN 	OzP 	QzR 	SzT 	UzV 	"WzX 	YzZ 	[z\ 	 ]z^ 	%_z` 	'azb 	*2czd 	ezf 	gzh 	izj 	kzl 	mzn 	ozp 	 qzr 	sz z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_gradr,  s    r1   get_default_nowrap_functionsr1  e  s-    $ \F r4   c            
      .   t           j        } i t           j        ddt           j        ddt           j        d t           j        d t           j        ddt           j        d t           j        ddt           j	        dd	t           j
        dd
t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d i t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d  t           j        dd!t           j        dd"t           j        d# t           j         dd$t           j!        dd%t           j"        dd&t           j#        dd'i t           j$        dd(t           j%        dd)t           j&        dd*t           j'        dd+t           j(        dd,t           j)        d- t           j*        d. t           j+        d/ t           j,        dd2t           j-        dd3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        dd<t           j7        d= t           j8        	 dd?t           j9        dd@t           j:        ddAt           j;        ddBt           j<        ddCt           j=        ddDt           j>        ddEt           j?        ddFt           j@        ddGt           jA        dH t           jB        ddIt           jC        dJ t           jD        dK t           jE        ddLi t           jF        dM t           jG        ddNt           jH        ddOt           jI        ddPt           jJ        ddSt           jK        ddTt           jL        ddVt           jM        ddWdXt           jN        dY t           jO        ddZt           jP        jO        dd[t           jP        jQ        dd\t           jR        dd]t           jS        dd^t           jT        d_ t           jU        d/d`t           jV        ddai t           jW        ddbt           jX        ddct           jY        dddt           jZ        ddet           j[        ddft           j\        dg t           j]        ddit           j^        dj t           j_        ddkt           j`        dl t           jP        ja        ddmt           jb        ddnt           jc        ddot           jd        ddpt           je        ddqt           jf        d/drt           jg        d dsi t           jh        d dtt           ji        d dut           jj        dv t           jk        d/dwt           jl        ddxt           jm        ddyt           jn        ddzt           jo        d{ t           jp        dd|t           jq        dd}t           jr        dd~t           js        ddt           jt        d t           ju        ddt           jP        ju        ddt           jv        	 ddt           jw        ddi t           jx        ddt           jy        ddt           jz        ddt           j{        ddt           j|        ddt           j}        ddt           j~        d t           j        d t           jP        j        d t           j        d t           j        ddt           j        ddt           j        d/dt           j        ddt           j        ddt           jP        j        d	dt           j        ddi t           j        ddt           j        ddt           j        d
dt           j        ddt           j        ddt           j        ddt           j        ddt           j        d t           j        d t           j        d t           j        ddt           jP        j        ddt           jP        j        dd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        ddt           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        dd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        ddʄt           j        j        dd˄t           j        j        dd̄i t           j        j        dd̈́t           j        j        dd΄t           j        j        ddτt           j        j        ddЄt           j        j        ddфt           j        j        dd҄t           j        j        ddӄt           j        j        ddԄt           j        ddՄt           j        ddքt           j        dׄ t           j        d؄ t           j        dل t           j        ddڄt           j        ddۄt           j        d܄ t           j        dd݄i t           j        ddބt           j        dd߄t           j        ddt           j        ddt           j        ddfdt           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        d i t           j        d t           j        d t           j        ddt           j        d t           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           j        ddt           jP        j        d t           j        dd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        d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        	 dd"t           j        d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        	 dd0t           j        dd1t           j        dd2t           j        dd3t           j         dd4i t           j!        dd5t           j"        dd6t           j#        dd7t           j$        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        d 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        d!dKt           jP        j7        d!dLt           jP        j8        d dMt           jP        j1        d 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@        d"dYt           jA        d"dZt           jB        d"d[t           jC        	 d#d\t           jD        dd]t           jE        d$d^t           jF        dd_t           jG        dd`t           jH        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        	 d,dt           jk        jl        j8        	 ddt           jk        jl        jL        ddt           jk        jl        jq        	 ddt           jk        jl        jx        	 d-dt           jk        jl        jv        	 ddt           jk        jl        j        d.dt           jk        jl        jy        d.dt           jk        jl        jz        d.di 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}        d0dt           jk        jl        j~        	 d1dt           jk        jl        j        	 	 d1dt           jk        jl        j        	 d1dt           jk        jl        j        	 	 d1dt           jk        jl        j        d2dt           jk        jl        j        d3dt           jk        jl        j        d4dt           jk        jl        j        d5dt           jk        jl        j        d'dt           jk        jl        j        d6dt           jk        jl        j        ddi t           jk        jl        j        d7dt           jk        jl        j        	 ddt           jk        jl        j        	 d8dt           jk        jl        j        	 d9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ń t           jk        jl        j        ddƄt           jk        jl        j        ddǄt           jk        jl        j2        	 ddȄt           jk        jl        j@        	 d#dɄt           jk        jl        jC        	 d#dʄi t           jk        jl        jA        	 d#d˄t           jk        jl        j        	 d#d̄t           jk        jl        jB        	 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        	 dAdׄt           jk        jl        j        dBdلt           jk        jl        j        d4dڄt           jk        jl        j        dCd܄t           jk        jl        j        dDd݄i t           jk        jl        j        	 dEdބt           jk        jl        j        d߄ t           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        dFdt           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        ddt           jk        jl        j        dGdt           jk        jl        j        dHdt           jk        jl        j        dIdt           jk        jl        j        d:dt           jk        jl        j        d=dt           jk        jl        j        d=dt           jk        jl        j        dJdt           jk        jl        j        ddt           jk        jl        j        d i t           jk        jl        j        d t           jk        jl        j        ddt           jk        jl        j        	 dKdt           jk        jl        j        ddUdd>ddt           jk        jl        j        d0dt           jk        j        j        dLdt           jk        j        j        dLdt           jk        j        j        d t           jk        j        j        dMdt           j        ddt           j        ddd t           j        d t           j        dNdt           jP        j        dOdt           jP        j        dPdt           jP        j        dQdt           j        dRdi t           j        dNdt           j        d	 t           j        d
 t           j        dSdt           j        dDdt           j        d t           j        dTdt           j        d
dt           j        dUdt           jP        j        dV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 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  t           j        d! t           j        d" t           j        d!d#t           jP        j        dWd%t           j        dXd't           j        dXd(t           j        d) t           j        d* t           j        d+ t           j        d, i t           j        d- t           j        d. t           j        t                      d/d0dfd1t           j        t                      d2d3dfd4t           j        t                      d5d6dfd7t           j        d8 t           j        d9 t           j        dd:t           j        dd;t           j        dt           j        ddfd<t           j        dd=t           j        d> t           j        dd?t           j        dd@t           jP        j        ddAt           j        dB t           j        dC i t           j        ddDt           j        ddEt           j        ddFt           j        ddGt           j        ddHt           j        dI t           j        dJ t           j        ddKt           j        dL t           j        ddMt           j        ddNt           j        dYdPt           j        ddQt           j        ddRt           j        dS t           j        dFdTt           j        ddUi t           j         d0dVt           j        ddWt           j        dX t           j        dY t           j        ddZt           j        dd[t           j        dZd]t           j        d^ t           j        d_ t           j	        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i t           j        ddht           j        di t           jP        j        dj t           j        dk t           j        dl t           j        ddmt           jP        j        d!dnt           jP        j        d dot           j        d\dddpdqt           j        d/drt           j        d/dst           j        ddtt           j        ddut           j        ddvt           j        ddwt           j        ddxt           j        ddyi t           j        ddzt           j         	 d]d|t           j!        dd}t           j"        dd~t           j#        ddt           j$        d t           j%        d t           j&        d t           j'        d t           j(        d t           j)        d t           j*        d t           j+        ddt           j,        d^dt           j-        d_dt           jP        j,        d!dt           jP        j.        ddi t           j/        d t           j0        d t           j1        j2        d t           j1        j3        d t           j1        j4        d t           j1        j5        d t           j1        j6        d t           j1        j7        ddt           j1        j8        ddt           j1        j9        ddt           j1        j:        ddt           j1        j        d t           j1        j;        d t           j1        j        d t           j1        j        d t           j1        j<        d t           j1        j        d i t           j1        j        d t           j1        j=        d t           j1        j        d t           j1        j>        ddt           j1        j?        ddt           j1        j@        d t           j1        jA        ddt           j1        jB        ddt           j1        j        d t           j1        jC        d t           j1        jD        d t           j1        jE        d t           j1        jF        ddt           j1        jG        ddt           j1        j         d t           j1        jH        d t           j1        j        ddi t           j1        j*        d t           j1        j+        ddt           j1        jI        d t           j1        jJ        d t           j1        jK        d t           j1        jL        d t           j1        jM        d t           j1        jN        d t           j1        jO        d t           j1        j        ddt           j1        jP        d t           j1        j        d t           j1        jQ        d t           j1        jR        d t           j1        jS        ddt           j1        jT        ddt           j1        jU        ddi t           j1        jV        ddt           j1        j        d t           j1        j        ddt           j1        jW        dÄ t           j1        jX        ddĄt           j1        j%        ddńt           j1        jY        ddƄt           jZ        dǄ t           j[        dȄ t           j\        ddɄt           j]        ddʄt           j^        dd˄t           jP        j_        d
d̄t           jP        j`        dd̈́t           ja        d`d΄t           jb        d/dτt           j        ddЄi t           jc        dф t           jd        d%d҄t           je        dӄ t           jf        dԄ t           jg        ddՄt           jh        ddքt           ji        dadׄt           jP        jj        dSd؄t           jk        ddلt           j        	 dKdڄt           jl        ddۄt           jm        d܄ t           jn        dd݄t           jo        d/dބt           jp        d߄ t           jq        dbdt           jr        ddi t           js        d t           jt        d/dt           ju        d/dt           jv        d/dt           jw        ddt           jP        jx        ddt           jy        ddt           jz        ddt           j{        d t           j|        ddt           j}        ddt           j~        ddt           j        d t           j        d t           j        d t           j        d t           j        d i t           j        d t           j        ddt           j        d t           j        ddt           j        dddt           j        d t           j        d t           j        d t           j        d t           j        d t           j        d[dt           j        d/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        d
 t           j        d t           j        d t           j        d/dt           j        d t           j        d t           j        d | j        d | j        d | j        d | j        d | j        d i | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | j        d | 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        d/ | j        d0 | j        d1 | j        d2 | j        d3 | j        dd4d5| j        d6 | j        d7 | j        j        d8 i | j        j        d9 | j        j        d: | j        j        d; | 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        dE | j        dF | j        j        dG | j        j        dH | j        j        dI i | j        j        dJ | j        j        dK | j        j        dL | j        j        dM | j        j        dN | j        j        dO | j        j        dP | j        j        dQ | j        j        dR | j        j        dS | j        j        dT | j        j        dU | j        j        dV | j        j        dW | j        j        dX | j        j        dY | j        j        dZ i | j        j        d[ | 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        ddh| j        di | j        dj | j        dk i | j        dl | j        dm | j        dn | j        do | j        dp | j        dq | j         dr | j        ds | j        dt | j        du | j        dv | j        dw | j        dx | j        dy | j        d1dz| j        t           j	        fd{| j
        t           j	        fd|i | j        t           j	        fd}| j        t           j	        fd~| j        dOd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        d | j        d | j        dd| j        d | j        d | j        t           j	        fdi | j        t           j	        fd| j        d | j        d | j        d | j         d0ddd| j!        d | j"        d | j#        t           j	        fd| j$        t           j	        fd| j%        ddd| j&        d | j'        t           j	        fd| j(        t           j	        fd| j)        d | j*        d | j+        t           j	        fd| j,        d i | j-        d | j        d | j.        d | j/        d | j0        d | j1        d | j2        dcddd| j        d | j3        t           j	        fd| j4        d | j5        d | jS        d | j        d | j6        d | j7        d | j8        d | j9        d i | j:        d | j        d | j;        d | j        d | j<        d | j=        dd| j>        d | j?        dddd| j@        d | jA        d | jB        d | jC        d | jD        d | jE        d | jF        dd| jG        d | jH        d i | jI        d | jJ        dÄ | jK        dĄ | jL        dń | jM        dddƄ| j        dǄ | jN        dȄ | jO        t           j	        fdɄ| jP        dʄ | j	        d[d˄| jQ        d̄ | jR        d̈́ | jS        dd΄| jT        dτ | jU        dЄ | j        ddф| jV        d҄ i | jW        dӄ | jX        dԄ | jY        dՄ | jZ        dք | jc        dׄ | j[        ddt           j	        fd؄| j\        dddٜdڄ| j]        ddۄ| j^        d܄ | j_        d݄ | j`        dބ | ja        d߄ | j        d | j        dOd| jb        d | jc        d | jd        d | je        d | jf        dd| jg        d t           jP        jh        ddi}i }t                      }|j                                D ]\  }}|jk        |jk        dz   d|jk        z   dz   d|jk        z   dz   d|jk        z   dz   g}|jk        l                    d          rP|jk        t          d          d         }|n                    d|z   dz   d|z   dz   d|z   dz   g           |D ]1}t          | |d          }	t          |	          r|	|vr	|	|vr|||	<   2|q                    |           |S (e  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 r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  r8  output_sizes     r1   r:  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r4  r6  )inputsr@  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  r8  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r4  r6  r8  otherr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4      c                     dS r4  r6  r8  batch1batch2alphabetar9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r4  r6  r8  tensor1tensor2valuer9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r4  r6  r]  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  r8  mat1mat2rZ  rY  r9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  )r8  matvecrZ  rY  r9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  )r8  vec1vec2rZ  rY  r9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  thetasizealign_cornerss      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rj  r4   c                     dS r4  r6  r8  dims     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   h㈵>:0yE>Fc                     dS r4  r6  )r8  rR  trolatol	equal_nans        r1   r:  z'get_testing_overrides.<locals>.<lambda>      VX r4   c                     dS r4  r6  r8  ptraininplaces       r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r8  rv  keepdimr9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rj  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  )r8  msgs     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                      dS r4  r6  tensorss    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                      dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                      dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   r   Tc                     dS r4  r6  )r8  kernel_sizer  padding	ceil_modecount_include_pads         r1   r:  z'get_testing_overrides.<locals>.<lambda>      vx r4   c                     dS r4  r6  rV  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      PR r4   c	                     dS r4  r6  )	r8  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r1   r:  z'get_testing_overrides.<locals>.<lambda>      y{ r4   c                     dS r4  r6  )grad_outr8  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r8  r  r  r  input_gweight_gbias_gs           r1   r:  z'get_testing_overrides.<locals>.<lambda>      su r4   c                     dS r4  r6  )r8  r  r  r  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r[  r4   c                     dS r4  r6  r8  r  r  r  r  r  r  counts           r1   r:  z'get_testing_overrides.<locals>.<lambda>      tv r4   c                     dS r4  r6  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>        AC r4   c                     dS r4  r6  r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r4  r6  )r8  r  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>      Z\ r4   c                     dS r4  r6  )r8  	generatorr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  input1input2r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   r  c                     dS r4  r6  r8  targetr  size_averagereduce	reduction
pos_weights          r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    TV r4   c                     dS r4  r6  )r8  weights	minlengths      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  probr  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4   c                      dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r8  rf  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                      dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  selfrr  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  )r8  
boundaries	out_int32rightr9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      [] r4   c                      dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  rv  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4          @#use_mm_for_euclid_dist_if_necessaryc                     dS r4  r6  )x1x2r  compute_modes       r1   r:  z'get_testing_overrides.<locals>.<lambda>      _a r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4         ?c                     dS r4  r6  r8  rY  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      2 r4   )r9  c                     dS r4  r6  )r9  matricess     r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  r8  groupss     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  upperr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  check_errorsr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r4  r6  )r  r  r  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>      B r4   c                     dS r4  r6  )r8  numeln_binsratio	bit_widths        r1   r:  z'get_testing_overrides.<locals>.<lambda>      WY r4   c                     dS r4  r6  r8  chunksrv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r8  minmaxr9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r%  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  )r8  r&  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r'  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  r9  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  
correctionfweightsaweightss       r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>      2 r4      c                     dS r4  r6  )r8  rwith_replacements      r1   r:  z'get_testing_overrides.<locals>.<lambda>      r r4   c                     dS r4  r6  )realimags     r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )absangs     r1   r:  z'get_testing_overrides.<locals>.<lambda>      b r4   c                     dS r4  r6  )r8  ords     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>      R r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>   r  r4   c                     dS r4  r6  )r8  padr`  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r  r  r  dilationr  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>      bd r4   c                     dS r4  r6  rM  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rM  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c	                     dS r4  r6  )	r8  r  r  r  r  rN  
transposedoutput_addingr  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    uw r4   c                     dS r4  r6  )r8  r  r  rK  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r  r  r  output_paddingr  rN  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>        A r4   c                     dS r4  r6  rW  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r4  r6  rW  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>	  rY  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>
  rB  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r  r  r  marginr  r  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  )r   r  rv  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r8  rR  rv  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   r5  c                     dS r4  r6  re  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r4  r6  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    .0 r4   c                     dS r4  r6  r8  rv  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rq  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  rv  r9  r`   s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r)  r4   c                     dS r4  r6  rt  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  yxrv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  rq  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  diagonalr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>   rA  r4   c                     dS r4  r6  )r8  offsets     r1   r:  z'get_testing_overrides.<locals>.<lambda>!  r=  r4   c                     dS r4  r6  )r8  nrv  prependappendr9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>"      TV r4   c                     dS r4  r6  r8  r  dim1dim2s       r1   r:  z'get_testing_overrides.<locals>.<lambda>#  r)  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>$  rn  r4   c                     dS r4  r6  )r8  srcr  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>%  r2  r4   c                     dS r4  r6  )r  r  rr  r  storage_offsets        r1   r:  z'get_testing_overrides.<locals>.<lambda>&  r   r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>'  rO  r4   c                     dS r4  r6  )r8  rR  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>(  rK  r4   c                     dS r4  r6  r8  rR  rounding_moder9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>)      b r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>*  rn  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>+  rS  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>,  rA  r4   c                     dS r4  r6  r8  rf  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>-  r}  r4   c                     dS r4  r6  )re  rf  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>.      r r4   c                     dS r4  r6  r8  indices_or_sectionss     r1   r:  z'get_testing_overrides.<locals>.<lambda>/  r  r4   c                     dS r4  r6  r-  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>0  r=  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>1  rS  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>2  r  r4   Lc                     dS r4  r6  r8  UPLOr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>3  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>4  r  r4   c                     dS r4  r6  )equationoperandss     r1   r:  z'get_testing_overrides.<locals>.<lambda>5  rS  r4   c                     dS r4  r6  r8  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r1   r:  z'get_testing_overrides.<locals>.<lambda>6  s
    (* r4   c
                     dS r4  r6  )
r8  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r1   r:  z'get_testing_overrides.<locals>.<lambda>8  s
    df r4   c                     dS r4  r6  r8  r`   rn   r_   requires_grads        r1   r:  z'get_testing_overrides.<locals>.<lambda>:      ce r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>;  r  r4   c                     dS r4  r6  r8  rR  s     r1   r:  z'get_testing_overrides.<locals>.<lambda><  r=  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>=  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>>  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>?  rK  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>@  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>A  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>B  rM  r4   c                     dS r4  r6  )r8  scale
zero_pointaxis	quant_min	quant_maxs         r1   r:  z'get_testing_overrides.<locals>.<lambda>C      mo r4   c                     dS r4  r6  )r8  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>D      fh r4   c                     dS r4  r6  )ry  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>E  s
    ac r4   c                     dS r4  r6  r8  packed_weightr  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>H  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>I      \^ r4   c                     dS r4  r6  r8  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>J      {} r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>K  
    SU r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>M  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>N  r  r4   c                     dS r4  r6  )r8  abs      r1   r:  z'get_testing_overrides.<locals>.<lambda>O  r)  r4   c                     dS r4  r6  r8  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>P  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>Q  r  r4   c                     dS r4  r6  r8  r  rv  norms       r1   r:  z'get_testing_overrides.<locals>.<lambda>R  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>S  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>T  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>U  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>V  r  r4   r  r5  c                     dS r4  r6  r8  srv  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>W  rj  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>X  rn  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>Y  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>Z  rg  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>[  rg  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>\  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>]  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>^  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>_  r9  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>`  rj  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>a  rj  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>b  rn  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>c  r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>d  rG  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>e  r)  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>f  r;  r4   c                     dS r4  r6  )r8  	start_dimend_dims      r1   r:  z'get_testing_overrides.<locals>.<lambda>g  rA  r4   c                     dS r4  r6  r8  dimss     r1   r:  z'get_testing_overrides.<locals>.<lambda>h  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>i  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>j  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>k  r[  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>l  rM  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>m  r  r4   c                     dS r4  r6  r8  exponentr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>n  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>o  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>p  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>q  rM  r4   c                     dS r4  r6  )r8  
fill_valuer9  r`   rn   r_   r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>r  s      BD r4   c                     dS r4  r6  )r8  r  	dep_tokens      r1   r:  z'get_testing_overrides.<locals>.<lambda>s  r  r4   c                     dS r4  r6  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r1   r:  z'get_testing_overrides.<locals>.<lambda>t  r  r4   c                     dS r4  r6  )r8  rv  indexr9  sparse_grads        r1   r:  z'get_testing_overrides.<locals>.<lambda>u  r2  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>v  rS  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>w  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>x  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>y  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>z  r=  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>{  r  r4   c                     dS r4  r6  r8  rm  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>|  r  r4   c                     dS r4  r6  r/  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>}  r  r4   c                     dS r4  r6  )r8  spacingrv  
edge_orders       r1   r:  z'get_testing_overrides.<locals>.<lambda>~  r  r4   c                     dS r4  r6  r8  gridinterpolation_modepadding_moders  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      ac r4   c                     dS r4  r6  r5  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      df r4   c                     dS r4  r6  r5  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  )r8  
num_groupsr  r  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      km r4   c	                     dS r4  r6  	r8  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r1   r:  z'get_testing_overrides.<locals>.<lambda>      qs r4   c                     dS r4  r6  r8  rB  w_ihw_hhb_ihb_hhs         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4         ?c                     dS r4  r6  r8  lambds     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  valuesr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r`  r  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      xz r4   d   c                     dS r4  r6  )r8  binsr&  r'  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r]  r&  r'  r  densityr9  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  r?  r4   c                     dS r4  r6  )r8  r]  r   r  r_  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r   r4   c                     dS r4  r6  r8  taus     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )re  rf  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r-  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r8  rv  r&  sources       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rl  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  indicesrW  
accumulates       r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  )r8  rv  r&  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r)  r4   c                     dS r4  r6  )r8  rv  r&  r`  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  rv  r&  rm  r  include_inputs         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rZ   s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )eteassume_uniqueinverts       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rw  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rw  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c	                     dS r4  r6  )	r8  r  r  r  r  use_input_statsr  r  r  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    -/ r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rn  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  rR  rtolr{  r|  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      UW r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c
                     dS r4  r6  )
r8  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    Z\ r4   c                     dS r4  r6  r8  r  r  r  r  
log_targets         r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    pr r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>       r4   c                     dS r4  r6  )r8  krv  r  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  	hermitianr  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r4  r6  )r8  r  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  )LDpivotsBr  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>      QS r4   c                     dS r4  r6  )r8  normalized_shaper  r  espr  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  endr  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  )r8  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    RT r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r8  rv  r`   s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )ry  rx  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  )r8  namesr  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c	                     dS r4  r6  )	databatch_sizesrB  rC  rD  rE  rF  r  rG  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  rK  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r2  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )Apivot	get_infosr9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r4  r6  )r  r!  r"  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r)  r4   c                     dS r4  r6  r_  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  rY  r4   c                     dS r4  r6  )r8  maskr`  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  rm  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  r8  r  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )LUr  r  leftadjointr9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      Y[ r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r8  r  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  )r8  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      2 r4   c                     dS r4  r6  r-  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r  r  rN  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>      jl r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r  r  rN  return_indicesr  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  
    OQ r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  )r8  rv  r  r`   r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                      dS r4  r6  )r  r0   s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r  r  r  exponential_average_factorepsilons           r1   r:  z'get_testing_overrides.<locals>.<lambda>  
    GI r4   c	                     dS r4  r6  	r8  r  r  r  r  rN  r  	benchmarkdeterministics	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c	                     dS r4  r6  )	r8  r  zrY  r  r  r  rN  r  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rM  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c
                     dS r4  r6  )
r8  r  r  r  rX  r  rN  r  r  r  s
             r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c	                     dS r4  r6  r  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    <> r4   c                     dS r4  r6  )r8  r  weight_stride0rB  cxr  hidden_sizerE  rH  rF  r  rG  r  dropout_states                 r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    VX r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r4  r6  r8  rm  destinations      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  
descendingr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>	  r  r4   c                     dS r4  r6  )r8  num_samplesreplacementr9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>
      SU r4   c                     dS r4  r6  )r8  ri  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  rv  startr  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4           c                     dS r4  r6  )r8  nanposinfneginfr9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r  r  r  r  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  )r8  r  r  r  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    ]_ r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  )r8  r  r  NCHxWgroupr  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  rv  r  r`   s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r)  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r?  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r?  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r@  r   s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rH  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>      oq r4   c                     dS r4  r6  rH  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rH  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>   rJ  r4   c                     dS r4  r6  rH  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>!  rO  r4   c                     dS r4  r6  rH  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>"  rJ  r4   c                     dS r4  r6  rp  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>#  r  r4   c                     dS r4  r6  r8  r  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>$  r  r4   c                     dS r4  r6  r8  r  r  r  r  r  divisor_overrides          r1   r:  z'get_testing_overrides.<locals>.<lambda>%  
    XZ r4   c                     dS r4  r6  rS  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>'  rU  r4   皙?c                     dS r4  r6  )r8  r  r  r  r  r  r  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>)  s
    BD r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>+  r  r4   c                     dS r4  r6  r8  r  r  r  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>,  
    EG r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>.  s
    oq r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>0  ra  r4   c                     dS r4  r6  r_  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>1  r  r4   c                     dS r4  r6  )r8  r  r  r  ignore_indexr  r  label_smoothings           r1   r:  z'get_testing_overrides.<locals>.<lambda>3  
    `b r4   c                     dS r4  r6  ri  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>5  s
    NP r4   c                     dS r4  r6  rQ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>7      XZ r4   c                     dS r4  r6  rQ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>8  r  r4   c                     dS r4  r6  rQ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>9  r  r4   c                     dS r4  r6  rQ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>:  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>;  rn  r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda><  
    PR r4   c                     dS r4  r6  )r8  r  r  r  r  r  r  r   r  include_last_offsetr  s              r1   r:  z'get_testing_overrides.<locals>.<lambda>>  s
    Y[ r4   c                     dS r4  r6  rQ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>A      gi r4   c                     dS r4  r6  )r8  r@  r  rN  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>B  r?  r4   c                     dS r4  r6  r8  r  r@  output_ratior   _random_sampless         r1   r:  z'get_testing_overrides.<locals>.<lambda>C  rd  r4   c                     dS r4  r6  rt  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>F  
    "$ r4   c                     dS r4  r6  rt  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>H  rd  r4   c                     dS r4  r6  rt  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>K  rx  r4   ư>c                     dS r4  r6  )r8  r  varr   r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>M  r  r4   nonec                     dS r4  r6  )r8  approximates     r1   r:  z'get_testing_overrides.<locals>.<lambda>N  r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>O  r  r4   bilinearr   c                     dS r4  r6  )r8  r6  r  r8  rs  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>P  rZ  r4   c                     dS r4  r6  )r8  r>  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>Q      eg r4   绽|=c                     dS r4  r6  )logitsrc  hardr  rv  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>R  r9  r4   c                     dS r4  r6  rT  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>S  r  r4         c                     dS r4  r6  )r8  min_valmax_valr  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>T  r  r4   c                     dS r4  r6  rY  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>U  r\  r4   c                     dS r4  r6  )r8  r  r  r  r  r  r  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>W  s
    [] r4   nearestc                     dS r4  r6  )r8  rr  scale_factorr  rs  recompute_scale_factor	antialiass          r1   r:  z'get_testing_overrides.<locals>.<lambda>Y  rU  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>[  rY  r4   c                     dS r4  r6  r8  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>\  r  r4   c                     dS r4  r6  r8  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>]  r?  r4   {Gz?c                     dS r4  r6  )r8  negative_sloper  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>^  r  r4   c                     dS r4  r6  )r8  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>_  r  r4   -C6?      ?c                     dS r4  r6  )r8  rr  rY  rZ  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>`  r  r4      c                     dS r4  r6  r8  rv  _stacklevelr`   s       r1   r:  z'get_testing_overrides.<locals>.<lambda>a  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>b  r  r4   c                     dS r4  r6  r8  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>c  r?  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>d  r?  r4   c                     dS r4  r6  r_  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>e  s
    QS r4   c                     dS r4  r6  r8  r  r  r  rN  r  r   s          r1   r:  z'get_testing_overrides.<locals>.<lambda>g  rm  r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>i  
    ]_ r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>k  rm  r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>m  r  r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>o  rm  r4   c                     dS r4  r6  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>q  r  r4   c                     dS r4  r6  r8  rp  r  r  r  r@  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>s  rZ  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>t  rZ  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>u  rZ  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>v      np r4   c                     dS r4  r6  )querykeyr`  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>x  s     km r4   c                     dS r4  r6  )r8  r  r  r`  r  r  r  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>|  r  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>~  r  r4   c                     dS r4  r6  r[  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  
    Y[ r4   c                     dS r4  r6  )r8  r  r  r  rb  r  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    FH r4   -q=c                     dS r4  r6  )r8  r  rv  r  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  )rZ   num_classess     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   r   c                     dS r4  r6  )r8  rK  r  r`  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r   r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  	log_inputr   r  r  r  r  s           r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r)  r4         ?UUUUUU?c                     dS r4  r6  r8  lowerr  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    wy r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  )r  r  r`  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r  r  rZ  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    } r4   c                     dS r4  r6  )r8  r  r  deltas       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4      c                     dS r4  r6  )r8  rZ  	thresholds      r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  rT  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  r  r`  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c
                     dS r4  r6  
anchorpositivenegativer`  r  r  swapr  r  r  s
             r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    pr r4   )distance_functionr`  r  r  c                    dS r4  r6  )r  r  r  r   r`  r  r  s          r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    ^` r4   c                     dS r4  r6  )r8  r  rN  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    `b r4   c                     dS r4  r6  )rZ   r  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )rZ   r  stdr  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r[  r4   c                     dS r4  r6  )rZ   vals     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   fan_in
leaky_reluc                     dS r4  r6  )rZ   r  r  nonlinearityr  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  as_tuples     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   )r  c                    dS r4  r6  )r8  rr  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   froc                     dS r4  r6  r8  r  rv  r  r9  r`   s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  rD  rv  r  r9  r`   s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )vpowrv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c                     dS r4  r6  rb  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  )r8  r  input3r  	transposes        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r[  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  rv  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  qr  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r4  r6  r,  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   V瞯<c                     dS r4  r6  )r8  rconds     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r(  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r4  r6  )r8  upscale_factors     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  downscale_factors     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  )r8  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  )r8  r  r  r   r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r   r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  )r8  r`   s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  )r8  r&  rm  rq  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  )r8  somer9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   reducedc                     dS r4  r6  )r8  r  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   linearc                     dS r4  r6  r8  r$  rv  r  interpolationr9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  rB  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  rq  r4   c                     dS r4  r6  )r8  scaleszero_pointsr  r`   s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r`   s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r`   reduce_ranges      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  r  r  r}  r  output_scaleoutput_zero_points           r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r8  rB  rL  rM  rN  rO  	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>  s
    eg r4   c                     dS r4  r6  rO  s                 r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    fh r4   )r   )rT  c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r\  r4   )r   r   )rT  rT  c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   )r   r   r   )rT  rT  rT  c                     dS r4  r6  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    JL r4   c                     dS r4  r6  rO  s                 r1   r:  z'get_testing_overrides.<locals>.<lambda>  
    jl r4   c                     dS r4  r6  rO  s                 r1   r:  z'get_testing_overrides.<locals>.<lambda>  r]  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  )r8  highr`   rn   r_   r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  re  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  rv  maxnormr9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  shapes     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c	                     dS r4  r6  rA  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rK  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c	                     dS r4  r6  rA  s	            r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rK  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  shiftsr  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   r   rT  c                     dS r4  r6  )r8  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r-  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  compressed_indices_dtypes      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    ^` r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  )r8  rR  rY  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  rd  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r8  rv  r&  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  rv  r&  r  r  include_selfs         r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r4  r6  )sorted_sequencer8  r  r  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   r'  c                     dS r4  r6  )r  r  lengthsrp  r  r  unsafes          r1   r:  z'get_testing_overrides.<locals>.<lambda>   s    z| r4   c                     dS r4  r6  )r8  rv  r&  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  )r8  r  rv  r&  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r  rv  r.  r  steps         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>	  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>
  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  r  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  )r  r  r  r  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   )stabler9  c                    dS r4  r6  )r8  rv  r#  r  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r   r4   c                     dS r4  r6  rZ   split_size_or_sectionsrv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rg  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  rq  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rd  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>  r2  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   reflectc
                     dS r4  r6  )
r8  r  r  r  r  r  pad_moder  r  r  s
             r1   r:  z'get_testing_overrides.<locals>.<lambda>  s
    _a r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>   r  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>!  r;  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>"  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>#  rI  r4   c                     dS r4  r6  r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>$  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>%  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>&  rI  r4   c                     dS r4  r6  )r  r  cs      r1   r:  z'get_testing_overrides.<locals>.<lambda>'  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>(  rB  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>)  rK  r4   c                     dS r4  r6  )r8  r=  
compute_uvr9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>*  r9  r4      c                     dS r4  r6  )r8  r$  r  Ms       r1   r:  z'get_testing_overrides.<locals>.<lambda>+  rD  r4   c                     dS r4  r6  )r8  full_matricesr9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>,  r  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>-  r  r4   c                     dS r4  r6  r8  dim0r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>.  rS  r4   c                     dS r4  r6  )r8  axis0axis1s      r1   r:  z'get_testing_overrides.<locals>.<lambda>/  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>0  rM  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>1  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>2  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>3  rO  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>4  rO  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>5  rn  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>6  rn  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>7  rn  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>8  rn  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>9  rM  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>:  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>;  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda><  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>=  r;  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>>  rF  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>?  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>@  r;  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>A  r;  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>B  rD  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>C  r)  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>D  rM  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>E  r9  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>F  rj  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>G  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>H  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>I  r}  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>J  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>K  rj  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>L  rj  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>M  r;  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>N  rK  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>O  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>P  r;  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>Q  r2  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>R  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>S  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>T  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>U  r  r4   c                     dS r4  r6  r,  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>V  rG  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>W  r=  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>X  r;  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>Y  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>Z  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>[  r;  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>\  rD  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>]  rD  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>^  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>_  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>`  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>a  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>b  r=  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>c  rA  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>d  r  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>e  rA  r4   c                     dS r4  r6  rQ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>f  r  r4   c                     dS r4  r6  )r  rR  r9  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>g  r  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>h  s    r r4   c                     dS r4  r6  )r8  r&  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>i  r  r4   c                     dS r4  r6  )r8  rp  rv  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>j  rn  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>k  r;  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>l  rF  r4   c                     dS r4  r6  )r  inds     r1   r:  z'get_testing_overrides.<locals>.<lambda>m  r  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>n  r  r4   c                     dS r4  r6  )r  r  r  r9  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>o  r  r4   c                     dS r4  r6  )r8  r  rv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>p  r  r4   c                     dS r4  r6  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>q  rj  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>r  r}  r4   c                     dS r4  r6  )r8  r  rv  r#  r9  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>s  rn  r4   c                     dS r4  r6  rH  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>t  r4  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>u  r  r4   c                     dS r4  r6  rw  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>v  rO  r4   c                     dS r4  r6  rw  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>w  r  r4   c                     dS r4  r6  )r8  r  r  r  unitriangulars        r1   r:  z'get_testing_overrides.<locals>.<lambda>x  r  r4   c                     dS r4  r6  )r8  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>y  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>z  r  r4   c
                     dS r4  r6  r  s
             r1   r:  z'get_testing_overrides.<locals>.<lambda>{  s
    VX r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>~  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r7  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  )r8  rv  sizesr  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  sortedreturn_inversereturn_countsrv  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r8  r   r!  rv  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )rp  rq  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r"  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  ra  r4   c                     dS r4  r6  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rq  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )ry  r:  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  ru  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r-  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  )	conditionry  rx  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  levels     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )primaltangentr1  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  r  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  )r  rr  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r[  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   )implicitc                    dS r4  r6  )r  rr  r=  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  rv  r.  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  rr  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  )r  rv  r&  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  )r  rv  r.  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  
split_sizerv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  split_sizesrv  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r"  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r"  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r"  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r  r`   s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r  	dimensionrr  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  rR  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rX  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  )r  arrays     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  idxs     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  )r  memos     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  )r  format_specs     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  )r  protos     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   )tensor_contentsc                    dS r4  r6  )r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rD  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  )r  ds     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r}  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rS  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>   rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rA  r4   c                     dS r4  r6  )r  r`   non_blockingr0   s       r1   r:  z'get_testing_overrides.<locals>.<lambda>	  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>
  r4  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rB  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>      " r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rI  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rM  r4   c                     dS r4  r6  )r  orderellipsis_idxs      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  callables     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rK  r4   c                     dS r4  r6  rB  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  rB  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  gradientretain_graphcreate_graphrC  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>  s    ik r4   c                     dS r4  r6  r  rg   s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r9  r4   )r  c                    dS r4  r6  )r  mediansigmar  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>   r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>!  rB  r4   c                     dS r4  r6  )r  	coalesceds     r1   r:  z'get_testing_overrides.<locals>.<lambda>"  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>#  r[  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>$  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>%  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>&  r9  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>'  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>(  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>)  rB  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>*  r  r4   c                     dS r4  r6  )r  r  r  r  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>+  r2  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>,  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>-  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>.  rn  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>/  ra  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>0  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>1  r=  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>2  rK  r4   c                    dS r4  r6  )r  rU  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>3  r  r4   c                     dS r4  r6  r  r`  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>4  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>5  r  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>6  rj  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>7  rn  r4   c                    dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>8  rA  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>9  r}  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>:  r9  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>;  rj  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda><  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>=  rI  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>>  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>?  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>@  r;  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>A  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>B  r  r4   c                     dS r4  r6  )r  rZ   s     r1   r:  z'get_testing_overrides.<locals>.<lambda>C  rO  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>D  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>E  r  r4   c                    dS r4  r6  )r  r  r  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>F  r  r4   c                     dS r4  r6  r"  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>G  rK  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>H  r9  r4   c                     dS r4  r6  )r  rZ   r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>I  r  r4   c                     dS r4  r6  )r  ry  rx  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>J  r  r4   c                     dS r4  r6  )r  rf  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>K  rB  r4   c                     dS r4  r6  )r  r[  r.  r  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>L  rg  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>M  r}  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>N  rB  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>O  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>P  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>Q  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>R  rI  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>S  r4  r4   c                     dS r4  r6  r"  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>T  r=  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>U  r}  r4   c                     dS r4  r6  )r  rp  rZ   rq  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>V  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>W  rI  r4   c                    dS r4  r6  )r  from_tor  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>X  ra  r4   c                     dS r4  r6  r  streams     r1   r:  z'get_testing_overrides.<locals>.<lambda>Y  r  r4   c                     dS r4  r6  )r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>Z  r  r4   c                     dS r4  r6  r  hooks     r1   r:  z'get_testing_overrides.<locals>.<lambda>[  r  r4   c                     dS r4  r6  r#  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>\  r  r4   c                     dS r4  r6  )r  names     r1   r:  z'get_testing_overrides.<locals>.<lambda>]  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>^  r;  r4   c                     dS r4  r6  )r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>_  r)  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>`  rO  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>a  r;  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>b  r;  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>c  rK  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>d  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>e  r  r4   c                     dS r4  r6  )r  rm  r  rr  r  s        r1   r:  z'get_testing_overrides.<locals>.<lambda>f  r  r4   c                     dS r4  r6  )r  r  rv  r&  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>g  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>h  r;  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>i  rj  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>j  r  r4   c                     dS r4  r6  )r  r  rv  r.  r  r  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>k  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>l  r}  r4   c                     dS r4  r6  )r  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>m  rO  r4   c                     dS r4  r6  )r  r  accumulate_matchess      r1   r:  z'get_testing_overrides.<locals>.<lambda>n  r  r4   c                     dS r4  r6  r  size1size2	dense_dims       r1   r:  z'get_testing_overrides.<locals>.<lambda>o  r  r4   c                     dS r4  r6  r;  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>p  r[  r4   c                     dS r4  r6  )r  re  rf  rZ  rY  r9  s         r1   r:  z'get_testing_overrides.<locals>.<lambda>q  r2  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>r  rI  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>s  rM  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>t  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>u  r=  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>v  rO  r4   c                     dS r4  r6  )r  repss     r1   r:  z'get_testing_overrides.<locals>.<lambda>w  r  r4   c                     dS r4  r6  )r  r`   r  copyrg   s        r1   r:  z'get_testing_overrides.<locals>.<lambda>x  s    ln r4   )masked_gradc                    dS r4  r6  r  r`   rJ  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>y  r9  r4   c                     dS r4  r6  rL  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>z  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>{  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>|  r  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>}  r  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>~  rF  r4   c                     dS r4  r6  rZ  s       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  r  r  s      r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  )r  rq  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   c                     dS r4  r6  r^  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rF  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  r4  r4   c                     dS r4  r6  r  s     r1   r:  z'get_testing_overrides.<locals>.<lambda>  rG  r4   c                     dS r4  r6  r6  s    r1   r:  z'get_testing_overrides.<locals>.<lambda>  rO  r4   c                     dS r4  r6  )r  r  r   drivers       r1   r:  z'get_testing_overrides.<locals>.<lambda>  r  r4   _____i__rbitwise_N)rT  rT  N)rT  N)rw  rx  F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)r  r  )r  F)FN)NNN)rT  NN)r5  F)NrT  r   rT  rT  )NrT  r   r   rT  rT  )r   NNr  )rT  rx  )r5  N)r   r  Fr4  )rT  r5  NNN)r   r   rT  )r   r  r5  )r5  )r  N)NNr  FF)Nr5  Fr  FNN)NNNF)FF)Nr5  N)Nr  N)r   r5  )TT)NF)NNrT  )NNrw  T)rR  )r  NNr  )r[  r   r   N)r[  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rT  F)Nr   rT  FF)NFNN)r5  FN)r/  NNN)NNrw  )r5  NFN)rR  FF)Nr   FTN)NNFrW  rw  )NNNr  )NNr`  Nr  r/  )rR  TF)	NNr5  Fr  FNFN)rT  r   rT  )NNFN)Fr{  r  )r~  )r5  )r  r   N)rT  Fr  r5  )r  r  F)NNNNTrW  rw  )NNr  NNF)NNr  )r  F)r  r  r  )Nr  N)Nr   N)TNTNFNNNNNNF)rT  r  NNNr  )NNr`  Nr  )r5  rT  r  N)r   r   )r  r{  F)TFNrx  Nr  )r  r  FF)Nr/  )NNr  r  )r  r  )rT  r  )r  r5  r{  FNNr  )r/  r  N)r   r  r	  N)r  NFNN)NNFNN)r5  NFNN)r  r  FNN)r5  r   )TF)NTr5  )r&  )r&  F)r>  N)NFr@  N)rT  rx  )r'  NNNr   F)r   NNrT  )r5  F)NNNTr  FTN)TTN)r  r5  N)r5  N)TFF)TFFN)rT  r5  )Nr   NN(r  r!   r6   r@  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dotrF  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igammacr<  	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	  r@  local_response_norm
logsigmoid	lp_pool1d	lp_pool2d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_hotrK  pairwise_distancepoisson_nll_lossprelurelurelu6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tuplequantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2deg	rand_likerandint_like
randn_likeravelr;  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_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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
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_hooksr.  _cdatar/  r0  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precisionr  r_   r`   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_ort	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizern   r'  r  nbytesndim	output_nrr  rq  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_rq   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudaxpuipudata_ptrr>  rv  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_rY   halfchalf	has_namesrp  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rh   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_shortrr  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizer  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asrW  viewview_aszero_
__dlpack____dlpack_device__r  r   items__name__
startswithlenextendgetattrr  update)
r6   retret2ignoredr  r  r  subnamer'  r"   s
             r1   r   r     s^   6 \Fm%	---m%222m% 	!#@#@m% 	!#A#A	m%
 	
...m% 	''m% 	000m% 	///m% 	111m% 		444m% 	QQQm% 	LLLm% 	LLLm% 	LLLm% 	JJJm%  	
KKK!m%" 	#%J%J#m% m%$ 		---%m%& 	XXX'm%( 	FFF)m%* 	
...+m%, 	
...-m%. 	JJJ/m%0 	///1m%2 		FFF3m%4 	&&5m%6 	&&7m%8 	1119m%: 	
...;m%< 	22=m%> 	000?m%@ 	///Am%B 	111Cm%D 	
...Em% m% m%F 	000Gm%H 	666Im%J 	888Km%L 	///Mm%N 	111Om%P 	--Qm%R 	--Sm%T 	--Um%V 	xxxWm%X 	RRRYm%Z 	{{[m%\ 	'){){]m%^ 	(*u*u_m%` 	 Q Qam%b 	%'v'vcm%d 	1  4C  4Cem%f 	 5 5gm% m% m%h 	%'\'\im%j 	CCCkm%l 	??mm%n 	.NR2W 2W 2Wom%r 	CCCsm%t 	>>>um%v 	<<<wm%x 	555ym%z 	;;;{m%| 	<<<}m%~ 	 "C"C"Cm%@ 	!#D#D#DAm%B 	--Cm%D 		333Em%F 	!4!4Gm%H 	11Im%J 	]]]Km% m% m%L 	11Mm%N 		666Om%P 	999Qm%R 	>>>Sm%T 	aaaUm%V 	
...Wm%X 	
===Ym%Z 	$::::[m%\ 	88]m%^ 	???_m%` 	999am%b 	 "P"P"Pcm%d 	 G G Gem%f 	NNNgm%h 	&(Y(Yim%j 	444km%l 	CCCmm% m% m%n 	
BBBom%p 	888qm%r 	888sm%t 	888um%v 		OOOwm%x 	%%ym%z 	III{m%| 	,,}m%~ 	999m%@ 	((Am%B 	555Cm%D 	
...Em%F 	777Gm%H 	666Im%J 	555Km%L 	===Mm%N 	dddOm% m% m%P 	dddQm%R 	dddSm%T 	wwUm%V 	===Wm%X 	  !A  !A  !AYm%Z 	  !A  !A  !A[m%\ 	  !A  !A  !A]m%^ 	((_m%` 		---am%b 	#  &C  &C  &Ccm%d 	
...em%f 	!C!C!Cgm%h 	--im%j 	@@@km%l 	EEEmm%n 	',1 1 1om%r 	555sm% m% m%t 	555um%v 	BBBwm%x 	AAAym%z 	"$@$@$@{m%| 	;;;}m%~ 	111m%@ 	**Am%B 		##Cm%D 	**Em%F 	&&Gm%H 	
:::Im%J 	@@@Km%L 	222Mm%N 	
VVVOm%P 	BBBQm%R 	KKKSm%T 	 O O OUm% m% m%V 	 "Y"Y"YWm%X 	111Ym%Z 	
000[m%\ 		HHH]m%^ 	KKK_m%` 		444am%b 	@@@cm%d 	
**em%f 	
))gm%h 	;;im%j 	222km%l 	444mm%n 	888om%p 	???qm%r 	CCCsm%t 	44um%v 	!&+ + +wm% m% m%z 	^bg g g{m%~ 	eeem%@ 	333Am%B 	,,Cm%D 		---Em%F 	
...Gm%H 	000Im%J 		---Km%L 	
...Mm%N 	///Om%P 	.0o0oQm%R 	-/h/hSm%T 	-Z_1d 1d 1dUm%Z 	')N)N[m%\ 	79^9^]m%^ 	')})}_m%` 	7 ;V ;Vam% m% m%d 	+-=-=em%f 	*,<,<gm%h 	*,B,Bim%j 	#%?%?km%l 	99mm%n 		CCCom%p 		CCCqm%r 		DDDsm%t 		CCCum%v 		DDDwm%x 		JJJym%z 		KKK{m%| 		DDD}m%~ 		EEEm%@ 		EEEAm%B 		FFFCm%D 		FFFEm% m% m%F 		GGGGm%H 		IIIIm%J 		JJJKm%L 		JJJMm%N 		KKKOm%P 		666Qm%R 		777Sm%T 		BBBUm%V 		---Wm%X 	@@@Ym%Z 	
**[m%\ 	&&]m%^ 	&&_m%` 	QQQam%b 	///cm%d 	33em%f 	???gm% m% m%h 	
555im%j 	
...km%l 	///mm%n 	t4PUP]fjz  D  D  Dom%p 	&(H(Hqm%r 	\\\sm%t 	OOOum%v 		444wm%x 	333ym%z 	>>>{m%| 	///}m%~ 	,,,m%@ 	666Am%B 	555Cm%D 		333Em%F 	NNNGm%H 	ccIm% m% m%J 	ffKm%L 	ffMm%N 	mmmOm%P 		ssQm%R 	NNNSm%T 	333Um%V 	888Wm%X 	555Ym%Z 	;;;[m%\ 	"$z$z$z]m%^ 	GGG_m%` 	mmmam%b 	YYYcm%d 	(*?*?em%f 	444gm%h 	;;im%j 	222km% m% m%l 	666mm%n 	777om%p 	888qm%r 	
...sm%t 	==um%v 	>>wm%x 	LLLym%z 	BBB{m%| 	==}m%~ 	\\\m%@ 	))Am%B 	
GGGCm%D 	&&Em%F 	''Gm%H 	222Im%J 	222Km%L 	 0 0Mm% m% m%P 	((Qm%R 	111Sm%T 	444Um%V 	KKKWm%X 	**Ym%Z 	''[m%\ 	&&]m%^ 	.._m%` 	,,am%b 	!1!1cm%d 	**em%f 	33gm%h 	))im%j 	WWWkm%l 	%%mm%n 	SX] ] ]om%r 	rrrsm% m% m%t 	
++um%v 	NNNwm%x 	"$c$c$cym%z 	!L!L!L{m%| 	 S S S}m%~ 	sssm%@	 		444A	m%B	 	666C	m%D	 	333E	m%F	 	;;;G	m%H	 	
;;;I	m%J	 	000K	m%L	 	LPU U UM	m%P	 		---Q	m%R	 	<<<S	m%T	 	///U	m%V	 	///W	m% m% m%X	 	
...Y	m%Z	 	:::[	m%\	 	;;;]	m%^	 	&&_	m%`	 	...a	m%b	 	<<<c	m%d	 	555e	m%f	 	;;;g	m%h	 	<<<i	m%j	 	///k	m%l	 	IIIm	m%n	 	
sso	m%p	 	OOOq	m%r	 	333s	m%t	 	
555u	m%v	 	EEEw	m%x	 	BBBy	m% m% m%z	 	!  $A  $A  $A{	m%|	 	88}	m%~	 	<<	m%@
 	===A
m%B
 	777C
m%D
 	???E
m%F
 	 F F FG
m%H
 	!#]#]#]I
m%J
 	[[[K
m%L
 	>>>M
m%N
 	//O
m%P
 	!#@#@#@Q
m%R
 	 "M"M"MS
m%T
 	 < < <U
m%V
 	**W
m%X
 	!1!1Y
m%Z
 		---[
m% m% m%\
 	888]
m%^
 	
555_
m%`
 	llla
m%b
 	lllc
m%d
 	llle
m%f
 	%HM)R )R )Rg
m%j
 	
...k
m%l
 	VVVm
m%n
 	000o
m%p
 	333q
m%r
 	55s
m%t
 		---u
m%v
 	888w
m%x
 	
555y
m%z
 	 #J #J{
m%~
 	 "}"}
m%@ 	)+v+vAm% m% m%B 	%'h'hCm%D 	* .R .REm%H 	* .? .?Im%L 	 Y YMm%P 	222Qm%R 	
EEESm%T 	<<Um%V 	==Wm%X 	AAAYm%Z 		444[m%\ 	999]m%^ 	UUU_m%` 	111am%b 	++cm%d 	::em%f 	WWWgm%h 	!s!sim% m% m%j 	&(_(_km%l 	88mm%n 	!f!f!fom%p 	!V!Vqm%r 	UUUsm%t 	$&?&?um%v 	333wm%x 	:::ym%z 		---{m%| 	222}m%~ 	:::m%@ 	/1N1NAm%B 	/1N1NCm%D 	/1d1d1dEm%F 	<>q>q>qGm%H 	/1d1d1dIm%J 	<>q>q>qKm% m% m%L 	/1d1d1dMm%N 	<>q>q>qOm%P 	')S)S)SQm%R 	)+a+a+aSm%T 	&RV*[ *[ *[Um%X 	&RV*[ *[ *[Ym%\ 	&;@*E *E *E]m%` 	$&R&R&Ram%b 	0=C4H 4H 4Hcm%f 	<im@r @r @rgm%j 	 "L"L"Lkm%l 	1KQ5V 5V 5Vmm%p 	)[^-c -c -cqm%t 	$GL(Q (Q (Qum%x 	#%Z%Z%Zym%z 	%'\'\'\{m%| 	%'\'\'\}m% m% m%~ 	%'\'\'\m%@ 	!K!K!KAm%B 	%IN)S )S )SCm%F 	)txSW-\ -\ -\Gm%L 	13i3i3iMm%N 	 "m"m"mOm%P 	1Z^5c 5c 5cQm%T 	>[` % % %Wm%Z 	1Z^5c 5c 5c[m%^ 	>[` % % %am%d 	-/u/u/uem%f 	 "F"F"Fgm%h 	!9!9!9im%j 	')z)z)zkm%l 	&(g(g(gmm%n 	*,c,c,com%p 	&(C(C(Cqm% m% m%r 	$&^&^&^sm%t 	0=C4H 4H 4Hum%x 	)TY-^ -^ -^ym%| 	'QV+[ +[ +[}m%@ 	"  %A  %A  %AAm%B 	#%o%o%oCm%D 	&(m(m(mEm%F 	&(\(\(\Gm%H 	"$G$G$GIm%J 	/1g1g1gKm%L 	')^)^)^Mm%N 	&(8(8Om%P 	%'m'm'mQm%R 	%'m'm'mSm%T 	/IO3T 3T 3TUm%X 	&IN*S *S *SYm%\ 	3V[7` 7` 7`]m% m% m%` 	&IN*S *S *Sam%d 	3V[7` 7` 7`em%h 	&IN*S *S *Sim%l 	3V[7` 7` 7`mm%p 	(*z*z*zqm%r 	(*z*z*zsm%t 	(*z*z*zum%v 	$&p&p&pwm%x 	8fjquchm m m{m%B 	-GM1R 1R 1RCm%F 	2?E6J 6J 6JGm%J 	7QW;\ ;\ ;\Km%N 	$>D(I (I (IOm%R 	%'X'X'XSm%T 	#%F%F%FUm%V 	!P!P!PWm%X 	-/a/a/aYm% m% m%Z 	,QW0\ 0\ 0\[m%^ 	!#;#;_m%` 	 "A"A"Aam%b 	!#B#B#Bcm%d 	!#y#y#yem%f 	 "A"A"Agm%h 	 "A"A"Aim%j 	 "A"A"Akm%l 	8:u:u:umm%n 	*,,,om%p 	&(\(\(\qm%r 	,.x.x.xsm%t 	#%Z%Z%Zum%v 	#%Z%Z%Zwm%x 	$&L&L&Lym%z 	&(C(C(C{m%| 	$&6&6}m% m% m%~ 	&(8(8m%@ 	%'X'X'XAm%B 	/hn3s 3s 3sCm%F 	=RV_bEJV\Aa Aa Aa AaGm%L 	"$b$b$bMm%N 	 M M MOm%P 	QQQQm%R 	!7!7Sm%T 	&(x(x(xUm%V 	777Wm%X 	FFFFYm%Z 	(([m%\ 	
\\\]m%^ 	ddd_m%` 	 "h"h"ham%b 	 "p"p"pcm%d 	999em% m% m%f 	dddgm%h 	%%im%j 	**km%l 	QQQmm%n 	!S!S!Som%p 	++qm%r 	IIIsm%t 	***um%v 	555wm%x 	IIIym%z 	=={m%| 	AA}m%~ 	777m%@ 	 Y YAm%B 	666Cm%D 	222Em%F 	--Gm% m% m%H 	dddIm%J 		777Km%L 	
000Mm%N 		DDDOm%P 	 "2"2Qm%R 	"$4$4Sm%T 	')9)9Um%V 	''Wm%X 	,,Ym%Z 	777[m%\ 	CCC]m%^ 	fff_m%` 	iiiam%b 	"$V$Vcm%d 	!#M#Mem%f 	)+P+Pgm%h 	"$s$sim% m% m%j 	  $h $hkm%p 	! %i %iqm%t 	"uwwX\.2e&H &H &Hum%x 	"uwwX^.4&J &J &Jym%| 	"uwwXa.75&M &M &M}m%@ 	% )m )mAm%D 	% )m )mEm%H 	111Im%J 	dddKm%L 	d5=Y]mrvvvMm%N 	eeeOm%P 	%%Qm%R 	
...Sm%T 	
555Um%V 	FFFWm%X 	,,Ym%Z 	//[m% m% m%\ 	444]m%^ 	
333_m%` 	:::am%b 	AAAcm%d 	!;!;!;em%f 	..gm%h 	xxim%j 	SSSkm%l 	xxmm%n 	SSSom%p 	
777qm%r 	777sm%t 	///um%v 	555wm%x 	PPym%z 	```{m%| 	///}m% m% m%~ 	
444m%@ 	MMMAm%B 	88Cm%D 	<<Em%F 	ZZZGm%H 	eeeIm%J 	|||Km%L 	22Mm%N 	??Om%P 	WWWQm%R 	
333Sm%T 	111Um%V 	
...Wm%X 	111Ym%Z 		---[m%\ 		---]m%^ 	
..._m% m% m%` 	
...am%b 	''cm%d 	..em%f 		))gm%h 	
**im%j 	888km%l 	@@@mm%n 	WWWom%p 	
YeQUYYYYqm%r 	EEEsm%t 	 P P Pum%v 	
...wm%x 	000ym%z 	;;;{m%| 	OOO}m%~ 	888m%@ 		---Am% m% m%B 	222Cm%D 	
Y]b b bEm%H 		444Im%J 	999Km%L 		---Mm%N 	))Om%P 	''Qm%R 	Sm%T 	Um%V 	''Wm%X 	))Ym%Z 	(([m%\ 	000]m%^ 		III_m%` 	AAAam%b 	HHHcm%d 	888em% m% m%f 	44gm%h 	66im%j 	//km%l 	!1!1mm%n 	!1!1om%p 	!1!1qm%r 	!1!1sm%t 	,.K.K.Kum%v 	,.K.K.Kwm%x 	,.K.K.Kym%z 	,.K.K.K{m%| 	//}m%~ 	,,m%@ 	++Am%B 	,,Cm%D 	--Em%F 	..Gm% m% m%H 	,,Im%J 	--Km%L 	--Mm%N 	 A A AOm%P 	!B!B!BQm%R 	//Sm%T 	*,I,I,IUm%V 	+-J-J-JWm%X 	**Ym%Z 	++[m%\ 	**]m%^ 	++_m%` 	+-J-J-Jam%b 	+-J-J-Jcm%d 	--em%f 	 0 0gm%h 	!#D#D#Dim% m% m%j 	--km%l 	!O!O!Omm%n 	(*:*:om%p 	(*:*:qm%r 	(*:*:sm%t 	(*:*:um%v 	"$7$7wm%x 	,,ym%z 	--{m%| 	!>!>!>}m%~ 	++m%@ 	--Am%B 	/1A1ACm%D 	/1A1AEm%F 	46S6S6SGm%H 	46S6S6SIm%J 	46S6S6SKm% m% m%L 	46S6S6SMm%N 	,,Om%P 	@@@Qm%R 	)+;+;Sm%T 	@@@Um%V 	>>>Wm%X 	<<<Ym%Z 	!![m%\ 	
++]m%^ 	KKK_m%` 		---am%b 	
...cm%d 	 3 3 3em%f 	 "<"<"<gm%h 	:::im%j 	HHHkm%l 	JJJmm% m% m%n 	
**om%p 	
KKKqm%r 	%%sm%t 	55um%v 	111wm%x 	555ym%z 	 e e e{m%| 	%'a'a'a}m%~ 	
:::m%@ 	!NT%Y %Y %YAm%F 	
:::Gm%H 	22Im%J 	///Km%L 	---Mm%N 	<<Om%P 	hhhQm%R 	 "g"g"gSm% m% m%T 	66Um%V 	;;;Wm%X 	LLLYm%Z 	%'W'W'W[m%\ 	888]m%^ 	111_m%` 		---am%b 	222cm%d 	;;em%f 	222gm%h 	999im%j 	eeekm%l 	55mm%n 	@@om%p 	qm%r 	"OOsm%t 	//um% m% m%v 	oowm%x 	QQQym%z 	')>)>{m%| 	FFF}m%~ 	%CCCCm%@ 	>>Am%B 	11Cm%D 	!#@#@Em%F 	66Gm%H 	??Im%J 	NNNKm%L 	<<<Mm%N 	#%H%H%HOm%P 	00Qm%R 	ooSm%T 	99Um%V 	22Wm% m% m%X 	__Ym%Z 	OO[m%\ 	OO]m%^ 	??_m%` 	am%b 	cm%d 	em%f 	gm%h 	111im%j 	//km%l 	AAmm%n 	//om%p 	33qm%r 	44sm%t 	44um%v 	22wm%x 	33ym% m% m%z 	33{m%| 	11}m%~ 	22m%@ 	22Am%B 	11Cm%D 	22Em%F 	22Gm%H 	..Im%J 	--Km%L 	..Mm%N 	//Om%P 	OOQm%R 	00Sm%T 	Um%V 	33Wm%X 	Ym%Z 	??[m% m% m%\ 	..]m%^ 	//_m%` 	//am%b 	55cm%d 	00em%f 	22gm%h 	im%j 	km%l 	//mm%n 	om%p 	77qm%r 	44sm%t 	__um%v 	AAAAwm%x 	11ym%z 	//{m%| 	//}m% m% m%~ 	//m%@ 		??Am%B 		??Cm%D 	&Em%F 	*2OOGm%H 	ooIm%J 	Km%L 	__Mm%N 	ooOm%P 	Qm%R 	Sm%T 	Um%V 	-/p/pWm%X 	*,V,VYm%Z 	__[m%\ 	]m%^ 	oo_m% m% m%` 	am%b 	cm%d 	em%f 	gm%h 	im%j 	km%l 	#__mm%n 	om%p 	qm%r 	sm%t 	 //um%v 	wm%x 	 //ym%z 	#__{m%| 	 //}m%~ 	$oom%@ 	 //Am% m% m%B 	Cm%D 	Em%F 	__Gm%H 	ooIm%J 	Km%L 	__Mm%N 	 //Om%P 	$ooQm%R 	ooSm%T 	Um%V 	__Wm%X 	__Ym%Z 	'/[m%\ 	NNN]m%^ 	oo_m%` 	ooam%b 	cm% m% m%d 	em%f 	__gm%h 	__im%j 	OOkm%l 	__mm%n 	OOom%p 	==qm%r 	sm%t 	um%v 	//wm%x 	==ym%z 	00{m%| 	88}m%~ 	99m%@ 	kkkAm%B 	E4IMMMCm%D 	0EIIIEm% m% m%F 	0EIIIGm%H 	0EIIIIm%J 	MTMMMMKm%L 	Mm%N 	66Om%P 	e6MQQQQm%R 	>>>Sm%T 	
u/DHHHUm%V 	0EIIIWm%X 	
u/DHHHYm%Z 	
u/DHHH[m%\ 	]m%^ 	//_m%` 	!O!O!Oam%b 	
OOcm%d 	//em%f 	%2GKKKgm% m% m%h 	53HLLLim%j 	__km%l 	,,mm%n 	00om%p 	HHHHHqm%r 	,,sm%t 	55um%v 	1FJJJwm%x 	%2GKKKym%z 	@@@@{m%| 	??}m%~ 	0EIIIm%@ 	1FJJJAm%B 	//Cm%D 	Em%F 	
u/DHHHGm%H 	__Im% m% m%J 	ooKm%L 	__Mm%N 	//Om%P 	11Qm%R 	//Sm%T 	__Um%V 	MTMMMMWm%X 	00Ym%Z 	0EIII[m%\ 	66]m%^ 	55_m%` 		((am%b 	EEcm%d 	??em%f 	gm%h 	"OOim%j 	-km% m% m%l 	%mm%n 	om%p 	ooqm%r 	,,sm%t 	??um%v 	GGGwm%x 	ym%z 	LDLLLL{m%| 	55}m%~ 	33m%@ 	33Am%B 	13H3HCm%D 	,,Em%F 	--Gm%H 	BBBIm%J 	11Km%L 	--Mm% m% m%N 	--Om%P 	00Qm%R 	 "8"8Sm%T 	OOUm%V 	[[[Wm%X 	??Ym%Z 	oo[m%\ 	1FJJJ]m%^ 	___m%` 	WWWam%b 	??cm%d 	11em%f 	&(W(W(Wgm%h 	GGim%j 	')Q)Qkm%l 	OOOmm%n 	om% m% m%p 	qm%r 	sm%t 	__um%v 	11wm%x 	++ym%z 		EUZUjnnn{m%| 	IIIII}m%~ 	GGGm%@ 	//Am%B 	Cm%D 	//Em%F 	..Gm%H 	==Im%J 	777Km%L 	Mm%N 	++Om%P 	..Qm% m%R 	oo777 //FFFYm% m%C^ D#%%G		  1 JJ1:$AJ%AJ%
 :  ,, 	 jZ!1!12GLL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 ra  )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   r0  z"wrap_torch_function.<locals>.inner  sD    				) 	) 	) 	) 	) 	) 
		) r4   r6  )r
  r0  s   ` r1   r   r     s#    .	 	 	 	 	 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sethasattrr   _disabled_torch_function_implri  r
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer&  iold_args	            r1   _get_overloaded_argsr
    s/   D  8..00 	"%%%!#O ( (;s## ,,,CW1X1X,+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 r6  )r&
  ).0r
  s     r1   
<listcomp>z)handle_torch_function.<locals>.<listcomp>]  s    EEEcS		EEEr4   z nor in mode )r
  r'	  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   L +=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 *   +=$j11 	.6G6PTb6b6b!)OOOM [,. . . #":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   rr	  r
  endswithislowerr
  object
isinstancer
  
ModuleType
__future___Featurer  r
  r#  __set__r   r   formatr
  r  )
overridable_funcsr&  tested_namespacesnamespace_str	namespacens_funcsr
  r(   r"   r  s
             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]););<	 /@ ;6 ;6*y(! :	6 :	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#:#:::@C</D/F/FFFF

S\^b^kHlHlFFF%d+224<@@@D>> *88Y88E$K  ,....<#8#:#::::CJJyRVR_<`<`:::i(//5555u:	6v 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
  r6  r4   r1   r   r     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.
    rT  )r
  r!   _ops
OpOverloadOpOverloadPacketstrr
  get)fs    r1   r   r     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                   N    e Zd ZU dZd e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!)
    r0  c                     d S ra  r6  r6  s    r1   r   zTorchFunctionMode.__init__d  s    r4   r6  Nc                     t                      ra  )NotImplementedErrorr  r"   r
  r/   r0   s        r1   r   z$TorchFunctionMode.__torch_function__g  s    !###r4   c                 $    t          |            | S ra  )
_push_moder6  s    r1   	__enter__zTorchFunctionMode.__enter__j  s    4r4   c                 "    t                       d S ra  )	_pop_mode)r  exc_typeexc_valexc_tbs       r1   __exit__zTorchFunctionMode.__exit__n  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.pushq  s,    hiii3'''r4   r6  N)r
  r
  __qualname____doc____annotations__r   r   r
  r
  classmethodr
  r6  r4   r1   r
  r
  C  s          :   $ $ $ $       [  r4   r
  c                  R    t                      } | dk    rt          | dz
            nd S )Nr   rT  )r   r   	stack_lens    r1   r
  r
  x  s-    )++I4=MM!)a-000tKr4   c                  P    t                      } d t          |           D             S )Nc                 ,    g | ]}t          |          S r6  )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_stackr  }  s)    )++I@@uY/?/?@@@@r4   c                 $    t          |            d S ra  )r   )r  s    r1   r
  r
    s    !$'''''r4   c                  "    t                      } | S ra  )r   olds    r1   r
  r
    s    
#
%
%CJr4   c               #   v   K   t                      } 	 | V  t          |            d S # t          |            w xY wra  )r
  r
  r
  s    r1   r
  r
    s<      
++C			3
3s   ' 8c                       e Zd ZddZdS )BaseTorchFunctionModer6  Nc                     |i } ||i |S ra  r6  r
  s        r1   r   z(BaseTorchFunctionMode.__torch_function__  s!    >FtT$V$$$r4   r
  )r
  r
  r
  r   r6  r4   r1   r  r    s(        % % % % % %r4   r  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 ra  )r!   r
  _RestorePythonTLSSnapshotr6  r4   r1   r   r     s       
	+	+	-	-  	EEEDDDDD	                 s   ;(;*;??)r    r!   ra  )9r
  r
  r
  r
  r
  r+   typingr   r   r   r   r   r   r	   r
   r   
contextlibr!   torch._Cr   r   r   r   r   r   r   r   r   r
  r
  r3   	lru_cacher   r1  r   r   r
  r   r   r   r   r
  r   r   r
  r9
  r   r   r
  r
  r  r
  r
  contextmanagerr
  r  r   r6  r4   r1   <module>r     s@   ,               H H H H H H H H H H H H H H H H H H H H           % % % % % % % % % % % % % % % % % % % % % %   U\ "NQ`h   : TMs8} M M M  M` Tc(m    2 TftHh$67 f f f  fP!"H " " " "HE E EHcUTX[DY Eeijmen E E E EPLL-5c]LPSL L L L\ ![  . ';	 	  *k 	   TI$E$sDN/B*CT(TW-EX*X$Y I$ I$ I$ I$V 	+4T(^(;#< 	+ 	+ 	+ 	+ 2 2 2& TS]     Gx GD G G G G0"K "K "KH2 2 2 2 2 2 2 2jL L L
A A A( ( (  
   % % % % %- % % %     r4   