
    +jZ                       d Z ddlZddlZddlZddlZddlZddlZddlZddlm	Z	m
Z
 ddlmZ ddlmZmZmZ ddlmZ ddlZddlmZmZmZmZmZmZmZmZmZ g dZ ed	          Z ed
          Z	 	 d2de	eef         de de de	eef         fdZ!ej"        e!de#e	         fd                        Z$ej"        de#e	         fd            Z%ej"        e!de&e	e	f         fd                        Z'de	ee
e         f         de	e	eef         ge	eef         f         fdZ(	 d3de
e         de	ege)f         dz  de*e         fdZ+de	eef         de
e         dej,        dej-        def
dZ. eed          Z/ eed          Z0 eed           Z1ej"        de2e&ee*e	         f         e&e	e f         f         fd!            Z3e!de&ee*e	         f         fd"            Z4e!d#             Z5ej"        de#e	         fd$            Z6e!de	de7fd%            Z8d& Z9 G d' d(          Z:d) Z;d* Z<d+ Z=d, Z>ej?        d-             Z@ G d. d/e:          ZAej?        d0             ZBej?        d1             ZCdS )4aE  
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)CallableIterable)wraps)AnycastTypeVar)	ParamSpec)	_add_docstr_get_function_stack_at_has_torch_function_has_torch_function_unary_has_torch_function_variadic_is_torch_function_mode_enabled_len_torch_function_stack_pop_torch_function_stack_push_on_torch_function_stack)
get_ignored_functionsget_overridable_functionsget_testing_overrideshandle_torch_functionhas_torch_functionresolve_nameis_tensor_likeis_tensor_method_or_propertywrap_torch_functionenable_reentrant_dispatch_P_R.*is deprecated, please use.*torchfuncregexmodulereturnc                      t                     dt          j        dt          j        dt          f fd            }|S )a  
    Decorator that temporarily disables ``UserWarning``s for the given ``module`` if the warning message matches the
    given ``regex`` pattern.

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

    Returns
    -------
    function
        The wrapped function.
    argskwargsr$   c                      t          j                    5  t          j        dt                      | i |cd d d            S # 1 swxY w Y   d S )Nignore)categorymessager#   )warningscatch_warningsfilterwarningsUserWarning)r&   r'   r!   r#   r"   s     R/home/wildlama/comfy/ComfyUI/.venv/lib/python3.11/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapper[   s    $&& 	) 	)#;f    4(((		) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)s   %AAA)r   r   r&   r'   r   )r!   r"   r#   r1   s   ``` r0   _disable_user_warningsr2   C   s_    0 4[[)rw )") ) ) ) ) ) ) ) ) [) N    c                  0   t           j        } h t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j	        t           j
        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j         t           j!        t           j"        t           j#        t           j$        t           j%        t           j&        t           j'        t           j(        t           j)        t           j*        t           j+        t           j,        t           j-        t           j.        t           j/        t           j0        t           j1        t           j2        t           j3        t           j4        t           j5        t           j6        t           j7        t           j8        t           j9        t           j:        t           j;        t           j<        t           j=        t           j>        t           j?        t           j@        t           jA        t           jB        t           jC        t           jD        t           jE        t           jF        t           jG        t           jH        t           jI        t           jJ        t           jK        t           jL        t           jM        t           jN        t           jO        t           jP        jP        t           jP        j        t           jP        jQ        t           jP        j        t           jR        t           jS        jT        t           jS        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        t           jz        t           j{        t           j|        t           j}        t           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        j        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        jd        t           j        j        j        t           j        j        jR        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        j        t           j        j        t          t          t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        j        j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        t           j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        | j        j        | j        j        | j        | j        | j        | j        | j         | j        | j        | j        | j        | j        | j        | j        }t          j        dk    r|	                    | j
                   |S )a%  
    Return public functions that cannot be overridden by ``__torch_function__``.

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

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

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

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

    Examples
    --------
    >>> import inspect
    >>> my_add = torch.overrides.get_testing_overrides()[torch.add]
    >>> inspect.signature(my_add)
    <Signature (input, other, out=None)>
    Nc                     dS N inputouts     r0   <lambda>z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  rK  output_sizes     r0   rM  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS rG  rI  )inputsrS  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  rK  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS rG  rI  rK  otherrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      " r3      c                     dS rG  rI  rK  batch1batch2alphabetarL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS rG  rI  rK  tensor1tensor2valuerL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS rG  rI  rp  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  rK  mat1mat2rm  rl  rL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  )rK  matvecrm  rl  rL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  )rK  vec1vec2rm  rl  rL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  thetasizealign_cornerss      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  rK  dims     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   h㈵>:0yE>Fc                     dS rG  rI  rK  re  rtolatol	equal_nans        r0   rM  z'get_testing_overrides.<locals>.<lambda>      VX r3   c                     dS rG  rI  rK  ptraininplaces       r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rK  r  keepdimrL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  )rK  msgs     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                      dS rG  rI  tensorss    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                      dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                      dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   r   Tc                     dS rG  rI  )rK  kernel_sizer  padding	ceil_modecount_include_pads         r0   rM  z'get_testing_overrides.<locals>.<lambda>      vx r3   c                     dS rG  rI  ri  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      PR r3   c	                     dS rG  rI  )	rK  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r0   rM  z'get_testing_overrides.<locals>.<lambda>      y{ r3   c                     dS rG  rI  )grad_outrK  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  rK  r  r  r  input_gweight_gbias_gs           r0   rM  z'get_testing_overrides.<locals>.<lambda>      su r3   c                     dS rG  rI  )rK  r  r  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  rK  r  r  r  r  r  r  counts           r0   rM  z'get_testing_overrides.<locals>.<lambda>      tv r3   c                     dS rG  rI  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>        AC r3   c                     dS rG  rI  rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS rG  rI  )rK  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>      Z\ r3   c                     dS rG  rI  )rK  	generatorrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  input1input2r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   r  c                     dS rG  rI  rK  targetr  size_averagereduce	reduction
pos_weights          r0   rM  z'get_testing_overrides.<locals>.<lambda>      rt r3   c                     dS rG  rI  )rK  weights	minlengths      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  probr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      " r3   c                      dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rK  ry  	out_dtyperL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                      dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  selfr  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  )rK  
boundaries	out_int32rightrL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>      [] r3   c                      dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>	  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>
      r r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3          @#use_mm_for_euclid_dist_if_necessaryc                     dS rG  rI  )x1x2r  compute_modes       r0   rM  z'get_testing_overrides.<locals>.<lambda>      _a r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3         ?c                     dS rG  rI  rK  rl  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   )rL  c                     dS rG  rI  )rL  matricess     r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  rK  groupss     r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS rG  rI  rK  upperrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  check_errorsrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      b r3   c                     dS rG  rI  r%  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS rG  rI  )r  r  r&  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>      B r3   c                     dS rG  rI  )rK  numeln_binsratio	bit_widths        r0   rM  z'get_testing_overrides.<locals>.<lambda>      WY r3   c                     dS rG  rI  rK  chunksr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  rK  minmaxrL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r:  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  )rK  r;  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r<  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  rL  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  
correctionfweightsaweightss       r0   rM  z'get_testing_overrides.<locals>.<lambda>      R r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>      2 r3      c                     dS rG  rI  )rK  rwith_replacements      r0   rM  z'get_testing_overrides.<locals>.<lambda>      r r3   c                     dS rG  rI  )realimags     r0   rM  z'get_testing_overrides.<locals>.<lambda>       " r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>!  r  r3   c                     dS rG  rI  )absangs     r0   rM  z'get_testing_overrides.<locals>.<lambda>"      b r3   c                     dS rG  rI  )rK  ords     r0   rM  z'get_testing_overrides.<locals>.<lambda>#  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>$  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>%  r#  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>&  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>'  r  r3   c                     dS rG  rI  )rK  padrs  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>(      2 r3   c                     dS rG  rI  rK  r  r  r  r  dilationr"  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>)      bd r3   c                     dS rG  rI  rb  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>*  rd  r3   c                     dS rG  rI  rb  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>+  rd  r3   c	                     dS rG  rI  )	rK  r  r  r  r  rc  
transposedoutput_addingr"  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>,      uw r3   c                     dS rG  rI  )rK  r  r  r_  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>-  r`  r3   c                     dS rG  rI  rK  r  r  r  r  output_paddingr"  rc  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>.        A r3   c                     dS rG  rI  rm  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>/  ro  r3   c                     dS rG  rI  rm  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>0  ro  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>1  rW  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>2  rN  r3   c                     dS rG  rI  r  r  r  marginr  r  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>3  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>4  rY  r3   c                     dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>5  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>6  rN  r3   c                     dS rG  rI  rK  re  r  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>7  rT  r3   rH  c                     dS rG  rI  r{  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>8      2 r3   c                     dS rG  rI  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r0   rM  z'get_testing_overrides.<locals>.<lambda>:  r  r3   c                     dS rG  rI  rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda><  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>=  r  r3   c                     dS rG  rI  rK  r  rL  rb   s       r0   rM  z'get_testing_overrides.<locals>.<lambda>>  r>  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>?  rW  r3   c                     dS rG  rI  yxr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>@  rT  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>B  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>C       r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>D       r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>E  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>F  r  r3   c                     dS rG  rI  rK  diagonalrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>G  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>H  rT  r3   c                     dS rG  rI  )rK  offsets     r0   rM  z'get_testing_overrides.<locals>.<lambda>I  rP  r3   c                     dS rG  rI  )rK  nr  prependappendrL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>J      TV r3   c                     dS rG  rI  rK  r  dim1dim2s       r0   rM  z'get_testing_overrides.<locals>.<lambda>K  r>  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>L  r  r3   c                     dS rG  rI  )rK  srcr  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>M  rG  r3   c                     dS rG  rI  )r  r  r  r  storage_offsets        r0   rM  z'get_testing_overrides.<locals>.<lambda>N  r5  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>O  rb  r3   c                     dS rG  rI  )rK  re  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>P  r^  r3   c                     dS rG  rI  rK  re  rounding_moderL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>Q      b r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>R  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>S  rf  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>T  rT  r3   c                     dS rG  rI  rK  ry  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>U  r  r3   c                     dS rG  rI  )rx  ry  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>V      r r3   c                     dS rG  rI  rK  indices_or_sectionss     r0   rM  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS rG  rI  rB  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>X  rP  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>Y  rf  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>Z  r  r3   Lc                     dS rG  rI  rK  UPLOrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>[  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>\  r  r3   c                     dS rG  rI  )equationoperandss     r0   rM  z'get_testing_overrides.<locals>.<lambda>]  rf  r3   c                     dS rG  rI  rK  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r0   rM  z'get_testing_overrides.<locals>.<lambda>_      z| r3   c
                     dS rG  rI  )
rK  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r0   rM  z'get_testing_overrides.<locals>.<lambda>b  s      hj r3   c                     dS rG  rI  rK  rb   rp   ra   requires_grads        r0   rM  z'get_testing_overrides.<locals>.<lambda>d      ce r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>e       r3   c                     dS rG  rI  rK  re  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>f  rR  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>g  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>h  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>i  r^  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>j  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>k  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>l  r`  r3   c                     dS rG  rI  )rK  scale
zero_pointaxis	quant_min	quant_maxs         r0   rM  z'get_testing_overrides.<locals>.<lambda>m      mo r3   c                     dS rG  rI  )rK  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>n      fh r3   c                     dS rG  rI  )r  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r0   rM  z'get_testing_overrides.<locals>.<lambda>p  s      AC r3   c                     dS rG  rI  rK  packed_weightr  outputs       r0   rM  z'get_testing_overrides.<locals>.<lambda>r  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>s      df r3   c                     dS rG  rI  rK  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>t      {} r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>v      ^` r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>x  r`  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>y  r  r3   c                     dS rG  rI  )rK  abs      r0   rM  z'get_testing_overrides.<locals>.<lambda>z  r>  r3   c                     dS rG  rI  rK  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>{  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>|  r  r3   c                     dS rG  rI  rK  r  r  norms       r0   rM  z'get_testing_overrides.<locals>.<lambda>}  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>~  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   r  rH  c                     dS rG  rI  rK  sr  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  )rK  	start_dimend_dims      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  rK  dimss     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  exponentrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )rK  
fill_valuerL  rb   rp   ra   r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  s      BD r3   c                     dS rG  rI  )rK  r  	dep_tokens      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  indexrL  sparse_grads        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rI  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  spacingr  
edge_orders       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS rG  rI  rK  gridinterpolation_modepadding_moder  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>      ac r3   c                     dS rG  rI  rO  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rO  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  
num_groupsr  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      km r3   c	                     dS rG  rI  	rK  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r0   rM  z'get_testing_overrides.<locals>.<lambda>      qs r3   c                     dS rG  rI  rK  r[  w_ihw_hhb_ihb_hhs         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3         ?c                     dS rG  rI  rK  lambds     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  r  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  valuesrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  rv  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      xz r3   d   c                     dS rG  rI  )rK  binsr;  r<  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  )rK  rw  r;  r<  r  densityrL  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS rG  rI  )rK  rw  r   r  ry  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r5  r3   c                     dS rG  rI  rK  taus     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rx  ry  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rB  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rK  r  r?  sources       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  indicesrq  
accumulates       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  )rK  r  r?  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  )rK  r  r?  rs  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )rK  r  r?  r  r  include_inputs         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r\   s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )eteassume_uniqueinverts       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c	                     dS rG  rI  )	rK  r  r  r  r  use_input_statsr  r  r  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  r)  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>      UW r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c
                     dS rG  rI  )
rK  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r0   rM  z'get_testing_overrides.<locals>.<lambda>        bd r3   c                     dS rG  rI  rK  r  r  r  r  
log_targets         r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    pr r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>       r3   c                     dS rG  rI  )rK  kr  r  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS rG  rI  )rK  	hermitianr*  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS rG  rI  )rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  )LDpivotsBr  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>      QS r3   c                     dS rG  rI  )rK  normalized_shaper  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  endr  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  )rK  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r0   rM  z'get_testing_overrides.<locals>.<lambda>  s      IK r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rK  r  rb   s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )rK  namesr  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c	                     dS rG  rI  )	databatch_sizesr[  r\  r]  r^  r_  r  r`  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>   rb  r3   c                     dS rG  rI  rd  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )Apivot	get_infosrL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  )r  r:  r;  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  ru  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS rG  rI  )rK  maskrs  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>	  r`  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>
  r#  r3   c                     dS rG  rI  rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r*  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )LUr  r  leftadjointrL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      Y[ r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  )rK  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>      2 r3   c                     dS rG  rI  rB  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  r  r  rc  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      jl r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  r  r  rc  return_indicesr  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  )rK  r  r  rb   rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>!  r^  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>"  r  r3   c                      dS rG  rI  )r  r'   s     r0   rM  z'get_testing_overrides.<locals>.<lambda>#  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>$  rN  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>%  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>&  r  r3   c                     dS rG  rI  )rK  r  r  r  r  r  exponential_average_factorepsilons           r0   rM  z'get_testing_overrides.<locals>.<lambda>(  r  r3   c	                     dS rG  rI  	rK  r  r  r  r  rc  r"  	benchmarkdeterministics	            r0   rM  z'get_testing_overrides.<locals>.<lambda>*  r  r3   c	                     dS rG  rI  )	rK  r  zrl  r  r  r  rc  r"  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>+  r  r3   c                     dS rG  rI  rb  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>,  r  r3   c
                     dS rG  rI  )
rK  r  r  r  rn  r  rc  r"  r'  r(  s
             r0   rM  z'get_testing_overrides.<locals>.<lambda>.  rj  r3   c	                     dS rG  rI  r&  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>1      eg r3   c                     dS rG  rI  )rK  r  weight_stride0r[  cxr  hidden_sizer^  ra  r_  r  r`  r  dropout_states                 r0   rM  z'get_testing_overrides.<locals>.<lambda>4  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>6  r>  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>7  r}  r3   c                     dS rG  rI  rK  r  destinations      r0   rM  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS rG  rI  r7  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>9  r`  r3   c                     dS rG  rI  )rK  
descendingrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>:  rW  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>;  rf  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda><  r  r3   c                     dS rG  rI  )rK  num_samplesreplacementrL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>=      SU r3   c                     dS rG  rI  )rK  r|  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>>  rb  r3   c                     dS rG  rI  rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>?  r  r3   c                     dS rG  rI  )rK  r  startr  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>@  r  r3           c                     dS rG  rI  )rK  nanposinfneginfrL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>A  r  r3   c                     dS rG  rI  )rK  r  r  r  r  r  r  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>B  rb  r3   c                     dS rG  rI  )rK  r  r  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>C      ]_ r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>D  r  r3   c                     dS rG  rI  rK  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>E  r  r3   c                     dS rG  rI  rK  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>F  r5  r3   c                     dS rG  rI  )rK  r  r  NCHxWgroupr  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>G  r  r3   c                     dS rG  rI  )rK  r  r  r  rb   s        r0   rM  z'get_testing_overrides.<locals>.<lambda>H  rA  r3   c                     dS rG  rI  r!  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>K  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>L  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>M  rP  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>N  r  r3   c                     dS rG  rI  rR  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>O  r/  r3   c                     dS rG  rI  rR  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>P  r/  r3   c                     dS rG  rI  rK  rS  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>Q  rd  r3   c                     dS rG  rI  rc  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>R      oq r3   c                     dS rG  rI  rc  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>S  rd  r3   c                     dS rG  rI  rc  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>T  re  r3   c                     dS rG  rI  rc  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>U  rd  r3   c                     dS rG  rI  rc  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>V  re  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS rG  rI  rK  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>X  r  r3   c                     dS rG  rI  rK  r  r  r  r  r  divisor_overrides          r0   rM  z'get_testing_overrides.<locals>.<lambda>Z        @B r3   c                     dS rG  rI  rn  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>]  rp  r3   皙?c                     dS rG  rI  )rK  r  r  r  r  r  r  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>`  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>b  r  r3   c                     dS rG  rI  rK  r  r  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>d  rS  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>g  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>i  rt  r3   c                     dS rG  rI  ru  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>k      gi r3   c                     dS rG  rI  )rK  r  r  r  ignore_indexr  r  label_smoothings           r0   rM  z'get_testing_overrides.<locals>.<lambda>n        JL r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>q  r  r3   c                     dS rG  rI  rl  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>s      XZ r3   c                     dS rG  rI  rl  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>t  r  r3   c                     dS rG  rI  rl  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>u  r  r3   c                     dS rG  rI  rl  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>v  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>w  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>y  r  r3   c                     dS rG  rI  )rK  r  r  r  r  r  r  r   r  include_last_offsetr  s              r0   rM  z'get_testing_overrides.<locals>.<lambda>|  s      HJ r3   c                     dS rG  rI  rl  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>~  rz  r3   c                     dS rG  rI  )rK  rS  r  rc  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS rG  rI  rK  r  rS  output_ratior  _random_sampless         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   ư>c                     dS rG  rI  )rK  r  varr   r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   nonec                     dS rG  rI  )rK  approximates     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   bilinearr   c                     dS rG  rI  )rK  rP  r  rR  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  )rK  rW  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r.  r3   绽|=c                     dS rG  rI  )logitsr}  hardr  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS rG  rI  rm  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3         c                     dS rG  rI  )rK  min_valmax_valr  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rs  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>      `b r3   c                     dS rG  rI  )rK  r  r  r  r  r  r  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>  s      GI r3   nearestc                     dS rG  rI  )rK  r  scale_factorr  r  recompute_scale_factor	antialiass          r0   rM  z'get_testing_overrides.<locals>.<lambda>  s      KM r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS rG  rI  rK  r  r  r  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rQ  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   {Gz?c                     dS rG  rI  )rK  negative_sloper  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   -C6?      ?c                     dS rG  rI  )rK  r  rl  rm  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r.  r3   r5   c                     dS rG  rI  rK  r  _stacklevelrb   s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    \^ r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rX  r3   c                     dS rG  rI  ru  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  rz  r3   c                     dS rG  rI  rK  r  r  r  rc  r  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  r  r  r  rS  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )querykeyrs  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r0   rM  z'get_testing_overrides.<locals>.<lambda>  s      ]_ r3   c                     dS rG  rI  )rK  r  r  rv  r  r  r  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rv  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rS  r3   c                     dS rG  rI  )rK  r  r  r  r}  r  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   -q=c                     dS rG  rI  )rK  r  r  r  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r\   num_classess     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   r   c                     dS rG  rI  )rK  r_  r  rs  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r+  r3   c                     dS rG  rI  r  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  	log_inputr   r  r  r  r  s           r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  rS  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3         ?UUUUUU?c                     dS rG  rI  rK  lowerr&  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    wy r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  )r  r  rs  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  r  r  rm  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  ro  r3   c                     dS rG  rI  )rK  r  r  deltar  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    hj r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3      c                     dS rG  rI  )rK  rm  	thresholds      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  rm  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  r  rs  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS rG  rI  
anchorpositivenegativerv  r  r  swapr  r  r  s
             r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   )distance_functionrv  r  r  c                    dS rG  rI  )r  r  r  r  rv  r  r  s          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  rc  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r\   r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  )r\   r  stdr  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r\   vals     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   fan_in
leaky_reluc                     dS rG  rI  )r\   r  r  nonlinearityr  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  as_tuples     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   )r6  c                    dS rG  rI  )rK  r  r6  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   froc                     dS rG  rI  rK  r  r  r  rL  rb   s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  rY  r  r  rL  rb   s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS rG  rI  r+  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r+  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  s     13 r3   c                     dS rG  rI  )vpowr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r)  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>   rd  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS rG  rI  r|  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  input3r  	transposes        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  qr  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rD  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   V瞯<c                     dS rG  rI  )rK  rconds     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r?  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>	  rN  r3   c                     dS rG  rI  )rK  upscale_factors     r0   rM  z'get_testing_overrides.<locals>.<lambda>
  r`  r3   c                     dS rG  rI  )rK  downscale_factors     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  )rK  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  )rK  r  r  r   r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r5  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS rG  rI  r0  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  )rK  rb   s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  )rK  r?  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  )rK  somerL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   reducedc                     dS rG  rI  )rK  r  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   linearc                     dS rG  rI  rK  r;  r  r  interpolationrL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rY  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rz  r3   c                     dS rG  rI  )rK  scaleszero_pointsr  rb   s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  rb   s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  rb   reduce_ranges      r0   rM  z'get_testing_overrides.<locals>.<lambda>   r+  r3   c                     dS rG  rI  )rK  r  r  r  r  r  output_scaleoutput_zero_points           r0   rM  z'get_testing_overrides.<locals>.<lambda>!  rb  r3   c                     dS rG  rI  rK  r[  re  rf  rg  rh  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r0   rM  z'get_testing_overrides.<locals>.<lambda>#        _a r3   c                     dS rG  rI  rf  s                 r0   rM  z'get_testing_overrides.<locals>.<lambda>&  ro  r3   rI  r   rg  c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>)  s
     " r3   r   r   rg  rg  c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>.  s     !# r3   r   r   r   rg  rg  rg  c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>4  s     !#	 r3   c                     dS rG  rI  rf  s                 r0   rM  z'get_testing_overrides.<locals>.<lambda>;  ro  r3   c                     dS rG  rI  rf  s                 r0   rM  z'get_testing_overrides.<locals>.<lambda>>  ro  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>@  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>A  rI  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>B  rY  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>C  r  r3   c                     dS rG  rI  r{  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>D  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>E  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>F  r`  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>G  rf  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>H  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS rG  rI  )rK  r  r  maxnormrL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>J  rW  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>K  r  r3   c                     dS rG  rI  )rK  shapes     r0   rM  z'get_testing_overrides.<locals>.<lambda>L  rY  r3   c                     dS rG  rI  rS  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>M  rn  r3   c	                     dS rG  rI  rZ  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>N  r  r3   c                     dS rG  rI  rd  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>O  r  r3   c	                     dS rG  rI  rZ  s	            r0   rM  z'get_testing_overrides.<locals>.<lambda>P  r  r3   c                     dS rG  rI  rd  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>Q  r  r3   c                     dS rG  rI  )rK  shiftsr)  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>R  r#  r3   r   rg  c                     dS rG  rI  )rK  r  r)  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>S  r#  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>T  r`  r3   c                     dS rG  rI  rB  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>U  r  r3   c                     dS rG  rI  )r  r  compressed_indices_dtypes      r0   rM  z'get_testing_overrides.<locals>.<lambda>V  r+  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>X  r`  r3   c                     dS rG  rI  )rK  re  rl  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>Y  rf  r3   c                     dS rG  rI  rw  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>Z  r  r3   c                     dS rG  rI  rK  r  r?  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>[  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>\  r  r3   c                     dS rG  rI  )rK  r  r?  r  r  include_selfs         r0   rM  z'get_testing_overrides.<locals>.<lambda>]  r  r3   c                     dS rG  rI  )sorted_sequencerK  r	  r
  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>^  r  r3   r<  c                     dS rG  rI  )r  r  lengthsr  r  r  unsafes          r0   rM  z'get_testing_overrides.<locals>.<lambda>_  r  r3   c                     dS rG  rI  )rK  r  r?  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>`  rP  r3   c                     dS rG  rI  )rK  r  r  r?  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>a  r  r3   c                     dS rG  rI  rK  r  r  rF  r  steps         r0   rM  z'get_testing_overrides.<locals>.<lambda>b  r  r3   c                     dS rG  rI  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>c  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>d  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>e  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>f  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>g  rb  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>h  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>i  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>j  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>k  rY  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>l  r\  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>m  rY  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>n  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>o  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>p  r  r3   c                     dS rG  rI  )r  r  r  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>q  rT  r3   c                     dS rG  rI  )r  r  r  r*  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>r  r  r3   )stablerL  c                    dS rG  rI  )rK  r  r;  r  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>s  r5  r3   c                     dS rG  rI  r\   split_size_or_sectionsr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>t  r}  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>u  r+  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>v  rY  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>w  r^  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>x  r  r3   c                     dS rG  rI  rw  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>y  rG  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>z  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>{  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>|  rP  r3   reflectc                     dS rG  rI  )rK  r  r  r  r  r  pad_moder  r  r  align_to_windows              r0   rM  z'get_testing_overrides.<locals>.<lambda>~  s      ~@ r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  )r  r  cs      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r&   s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  )rK  rT  
compute_uvrL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3      c                     dS rG  rI  )rK  r;  r  Ms       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  )rK  full_matricesrL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rK  dim0r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rf  r3   c                     dS rG  rI  )rK  axis0axis1s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rD  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r	  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  rd  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  re  rL  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    r r3   c                     dS rG  rI  )rK  r?  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  )r  inds     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r)  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r)  rL  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r(  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r  r;  rL  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r[  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r&  r6  unitriangulars        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  r  r&  r  r;  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c
                     dS rG  rI  r  s
             r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  rJ  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  )rK  r  sizesr  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  sortedreturn_inversereturn_countsr  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  rG  rH  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r.  r3   c                     dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r7  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  rU  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rB  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  )	conditionr  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )rK  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  levels     r0   rM  z'get_testing_overrides.<locals>.<lambda>   r  r3   c                     dS rG  rI  )primaltangentr`  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  r  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   )implicitc                    dS rG  rI  )r  r  rl  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>	  r  r3   c                     dS rG  rI  )r  r  rF  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>
  r  r3   c                     dS rG  rI  )r  r)  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  )r  r  r?  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  rF  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r/  r3   c                     dS rG  rI  )r  
split_sizer  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  split_sizesr  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r9  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r9  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rP  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r9  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  rb   s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  r  	dimensionr  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>   rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>!  r  r3   c                     dS rG  rI  r  re  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>"  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>#  rf  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>$  rf  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>%  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>&  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>'  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>(  rb  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>)  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>*  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>+  rb  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>,  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>-  rP  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>.  rY  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>/  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>0  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>1  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>2  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>3  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>4  rW  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>5  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>6  r\  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>7  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>8  rY  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>9  r`  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>:  r`  r3   c                     dS rG  rI  )r  arrays     r0   rM  z'get_testing_overrides.<locals>.<lambda>;  r  r3   c                     dS rG  rI  )r  idxs     r0   rM  z'get_testing_overrides.<locals>.<lambda><  r^  r3   c                     dS rG  rI  )r  memos     r0   rM  z'get_testing_overrides.<locals>.<lambda>=  rP  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>>  r\  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>?  rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>@  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>A  r\  r3   c                     dS rG  rI  )r  format_specs     r0   rM  z'get_testing_overrides.<locals>.<lambda>B  r#  r3   c                     dS rG  rI  )r  protos     r0   rM  z'get_testing_overrides.<locals>.<lambda>C  rf  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>D  rR  r3   )tensor_contentsc                    dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>E  rW  r3   c                     dS rG  rI  )r  r  r/  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>F  rb  r3   c                     dS rG  rI  )r  ds     r0   rM  z'get_testing_overrides.<locals>.<lambda>G  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>H  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>I  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>J  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>K  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>L  r#  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>M  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>N  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>O  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>P  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>Q  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>R  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>S  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>T  rP  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>U  r^  r3   c                     dS rG  rI  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r0   rM  z'get_testing_overrides.<locals>.<lambda>V  s    np r3   c                     dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>W  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>X  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>Y  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>Z  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>[  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>\  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>]  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>^  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>_  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>`  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>a  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>b  rf  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>c  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>d  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>e  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>f  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>g  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>h  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>i  rf  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>j  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>k  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>l  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>m  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>n  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>o  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>p  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>q  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>r  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>s  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>t  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>u  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>v  r^  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>w  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>x  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>y  rT  r3   c                     dS rG  rI  )r  rb   non_blockingr'   s       r0   rM  z'get_testing_overrides.<locals>.<lambda>z  r/  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>{  rI  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>|  rI  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>}  rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>~  rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>      " r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )r  orderellipsis_idxs      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )r  callables     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  rq  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rq  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  gradientretain_graphcreate_graphrV  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    ik r3   c                     dS rG  rI  r  ri   s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   )r  c                    dS rG  rI  )r  mediansigmar  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  )r  	coalesceds     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  ambiguity_checks     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                    dS rG  rI  )r  rn  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  rs  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r*  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                    dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r\   s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                    dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r9  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  )r  r\   r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  ry  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  re  assigns      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rT  r3   c                     dS rG  rI  )r  r  rF  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rW  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS rG  rI  r9  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r\   r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                    dS rG  rI  )r  from_tor  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rt  r3   c                     dS rG  rI  )r  streams     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  hooks     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  rW  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  names     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r>  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r^  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  r?  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r}  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r  rF  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  )r  r  accumulate_matchess      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  size1size2	dense_dims       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  ro  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  rn  r3   c                     dS rG  rI  )r  rx  ry  rm  rl  rL  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>  rG  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r\  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rR  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  )r  repss     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  rb   r  copyri   s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  s    ln r3   )masked_gradc                    dS rG  rI  r  rb   r~  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  rN  r3   c                     dS rG  rI  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r-  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r`  r3   c                     dS rG  rI  )r  rQ  rR  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r#  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  r  s     r0   rM  z'get_testing_overrides.<locals>.<lambda>  rY  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rI  r3   c                     dS rG  rI  )r  rT  max_version	dl_devicer}  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>  rd  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  rb  r3   c                     dS rG  rI  )r  r  r  s      r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  r  r   drivers       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  )r  ra   r  r'   s       r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   c                     dS rG  rI  re  s    r0   rM  z'get_testing_overrides.<locals>.<lambda>  r  r3   is______i__rbitwise_c                     dS rG  rI  )r\   r  rX  async_op	group_srcs        r0   rM  z'get_testing_overrides.<locals>.<lambda>+  r.  r3   c                     dS rG  rI  )r\   oprX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>,  r  r3   c                     dS rG  rI  )r\   dstr  rX  r  	group_dsts         r0   rM  z'get_testing_overrides.<locals>.<lambda>-  rX  r3   c                     dS rG  rI  )r  r  rX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>.  r  r3   c                     dS rG  rI  )tensor_listr\   rX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>/  r  r3   c                     dS rG  rI  )output_tensorinput_tensorrX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>0  r  r3   c                     dS rG  rI  )output_tensor_listsinput_tensor_listrX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>1  r  r3   c                     dS rG  rI  )r\   gather_listr  rX  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>2  r  r3   c                     dS rG  rI  )r\   scatter_listr  rX  r  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>3  r  r3   c                     dS rG  rI  )r  
input_listr  rX  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>4  r.  r3   c                     dS rG  rI  )r  rK  r  rX  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>5  rz  r3   c                     dS rG  rI  )r  rK  output_split_sizesinput_split_sizesrX  r  s         r0   rM  z'get_testing_overrides.<locals>.<lambda>6  s      LN r3   c                     dS rG  rI  )output_tensor_listr  rX  r  s       r0   rM  z'get_testing_overrides.<locals>.<lambda>7  rX  r3   c                     dS rG  rI  r\   r  rX  tagr  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>8  r  r3   c                     dS rG  rI  r\   r  rX  r  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>9  r  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>:  r5  r3   c                     dS rG  rI  r  s        r0   rM  z'get_testing_overrides.<locals>.<lambda>;  r5  r3   N)rg  rg  N)rg  N)r  r  F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)r  r  )r  F)FN)NNN)rg  NN)rJ  F)Nrg  r   rg  rg  )Nrg  r   r   rg  rg  )r   NNr  )rg  r  )rH  N)r   r  FrG  )rg  rH  NNN)r   r   rg  )r   r  rH  )rJ  )r  N)NNr  FF)NrJ  Fr  FNN)NNNF)FF)NrH  N)Nr  N)r   rH  )TT)NF)NNrg  )NNr  T)rk  )NFr   N)r  NNr  )ru  r   r   N)ru  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rg  F)Nr   rg  FF)NFNN)rH  FN)rG  NNN)NNr  )Nr  )rJ  NFN)rk  FF)Nr   FTN)NNFrr  r  )NNNr  )NNr{  Nr  rG  )rk  TF)	NNrJ  Fr  FNFN)rg  r   rg  )NNFN)Fr  r  )r  )rH  )r  r   N)rg  Fr  rH  )r  r  F)NNNNTrr  r  )NNr  NNF)NNr  N)r  F)r  r  r  )Nr5   N)Nr   N)TNTNFNNNNNNF)rg  r  NNNr  )NNr  )NNr{  Nr  )rJ  rg  r  N)r   r   )r  r  F)TFNr  Nr  )Nr  )r  r  FF)NrG  )NNr  r  )r  r  N)rg  r	  )r  rJ  r  FNNr  )rG  r  N)r   r  r   N)r'  NFNN)NNFNN)rJ  NFNN)r'  r  FNN)rJ  r   )TF)NTrJ  )r=  )r=  F)rU  N)NFrW  N)rI  rq  rr  F)rI  rt  ru  F)rI  rw  rx  F)rg  r  )r<  NNNr   F)r   NNrg  )rH  F)	NNNTr  FTNN)TTN)r  rJ  N)rJ  N)TFF)TFFN)rg  rJ  )Nr   NN)NNNN)NNNFN)NNr   N(  r    r7   rU  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshr=  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dotr_  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdge
get_devicegreater_equalgeqrfi0inneroutergerr  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrinkhash_tensor	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacrQ  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr<  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr;  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_fused_rms_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_lossr   rW  local_response_norm
logsigmoid	lp_pool1d	lp_pool2d	lp_pool3dmax_pool2d_with_indicesmax_pool3d_with_indicesmax_unpool1dmax_unpool2dmax_unpool3dmse_lossmulti_head_attention_forwardmulti_margin_lossmultilabel_margin_lossmultilabel_soft_margin_lossnll_loss	normalizeone_hotr_  pairwise_distancepoisson_nll_lossprelurelurelu6rms_normrreluselusilumishscaled_dot_product_attentionsmooth_l1_loss
huber_losssoft_margin_losssoftmaxsoftminsoftplus
softshrinksoftsign
tanhshrinkr  triplet_margin_loss!triplet_margin_with_distance_lossunfoldr   uniform_normal_	constant_kaiming_uniform_nonzerononzero_staticargwherer  vector_normmatrix_normnorm_except_dimnuclear_normr1  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar  	ones_liker0  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2degravelrP  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_itesym_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer6  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr.  HmTmH_backward_hooks_post_accumulate_grad_hooksrA  _cdatarB  rC  _grad_fngrad_fn
grad_dtype_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  ra   rb   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizerp   r[  r  nbytesndim	output_nrr  r  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_update_names_valuesalign_asalign_toapply_rs   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrrr  r  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesr  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rj   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_shortr  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizerR  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asrq  viewview_aszero_
__dlpack____dlpack_device__r?  r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrr   items__name__
startswithlenextendr  updatedistributedis_availabletorch.distributed	broadcast
all_reducer  all_reduce_coalesced
all_gatherall_gather_into_tensorall_gather_coalescedreduce_scatterreduce_scatter_tensorall_to_all_single
all_to_allisendirecvsendrecv)r7   retprivateuse1_backend_nameret2ignoredr  r/  r  subnamer[  r!   rA  s               r0   r   r     s9a   6 \F{%	---{%222{% 	!#@#@{% 	!#A#A	{%
 	
...{% 	''{% 	000{% 	///{% 	111{% 		444{% 	QQQ{% 	LLL{% 	LLL{% 	LLL{% 	JJJ{%  	
KKK!{%" 	#%J%J#{% {%$ 		---%{%& 	XXX'{%( 	FFF){%* 	
...+{%, 	
...-{%. 	JJJ/{%0 	///1{%2 		FFF3{%4 	&&5{%6 	&&7{%8 	1119{%: 	
...;{%< 	22={%> 	000?{%@ 	///A{%B 	111C{%D 	
...E{% {% {%F 	000G{%H 	666I{%J 	888K{%L 	///M{%N 	111O{%P 	--Q{%R 	--S{%T 	--U{%V 	xxxW{%X 	RRRY{%Z 	{{[{%\ 	'){){]{%^ 	(*u*u_{%` 	 Q Qa{%b 	%'v'vc{%d 	1  4C  4Ce{%f 	 5 5g{% {% {%h 	%'\'\i{%j 	CCCk{%l 	??m{%n 	.tttq{%t 	CCCu{%v 	>>>w{%x 	<<<y{%z 	555{{%| 	;;;}{%~ 	<<<{%@ 	 "C"C"CA{%B 	!#D#D#DC{%D 	--E{%F 		CCCG{%H 	!4!4I{%J 	11K{%L 	]]]M{% {% {%N 	11O{%P 		666Q{%R 	999S{%T 	>>>U{%V 	aaaW{%X 	
...Y{%Z 	
>>>[{%\ 	$::::]{%^ 	77_{%` 	???a{%b 	999c{%d 	 "P"P"Pe{%f 	 G G Gg{%h 	NNNi{%j 	&(Y(Yk{%l 	444m{%n 	CCCo{% {% {%p 	
BBBq{%r 	888s{%t 	888u{%v 	888w{%x 		OOOy{%z 	%%{{%| 	III}{%~ 	,,{%@ 	999A{%B 	((C{%D 	555E{%F 	
...G{%H 	777I{%J 	666K{%L 	555M{%N 	===O{%P 	dddQ{% {% {%R 	dddS{%T 	dddU{%V 	wwW{%X 	===Y{%Z 	  !A  !A  !A[{%\ 	  !A  !A  !A]{%^ 	  !A  !A  !A_{%` 	((a{%b 		---c{%d 	#  &C  &C  &Ce{%f 	
...g{%h 	!C!C!Ci{%j 	--k{%l 	@@@m{%n 	EEEo{%p 	xxxs{%v 	555w{% {% {%x 	555y{%z 	BBB{{%| 	AAA}{%~ 	"$@$@$@{%@ 	;;;A{%B 	111C{%D 	**E{%F 		##G{%H 	**I{%J 	&&K{%L 	
:::M{%N 	@@@O{%P 	222Q{%R 	
VVVS{%T 	BBBU{%V 	KKKW{%X 	 O O OY{% {% {%Z 	 "Y"Y"Y[{%\ 	111]{%^ 	
000_{%` 		HHHa{%b 	KKKc{%d 		444e{%f 	@@@g{%h 	
:::i{%j 	
))k{%l 	;;m{%n 	222o{%p 	444q{%r 	888s{%t 	???u{%v 	CCCw{%x 	44y{%z 	|||}{% {% {%@ 	 j  j  jC{%F 	eeeG{%H 	333I{%J 	,,K{%L 		---M{%N 	
...O{%P 	000Q{%R 		---S{%T 	
...U{%V 	///W{%X 	.0o0oY{%Z 	-/h/h[{%\ 	- C  C  C_{%b 	')V)Vc{%d 	79f9fe{%f 	')})}g{%h 	7``k{% {% {%n 	+-=-=o{%p 	*,<,<q{%r 	*,B,Bs{%t 	#%?%?u{%v 	99w{%x 		CCCy{%z 		CCC{{%| 		DDD}{%~ 		CCC{%@ 		DDDA{%B 		JJJC{%D 		KKKE{%F 		DDDG{%H 		EEEI{%J 		EEEK{%L 		FFFM{%N 		FFFO{% {% {%P 		GGGQ{%R 		IIIS{%T 		JJJU{%V 		JJJW{%X 		KKKY{%Z 		666[{%\ 		777]{%^ 		BBB_{%` 		---a{%b 	@@@c{%d 	
**e{%f 	&&g{%h 	&&i{%j 	QQQk{%l 	///m{%n 	33o{%p 	???q{% {% {%r 	
555s{%t 	
...u{%v 	///w{%x 	t4PUP]fjz  D  D  Dy{%z 	&(H(H{{%| 	\\\}{%~ 	OOO{%@ 		444A{%B 	333C{%D 	**E{%F 	>>>G{%H 	///I{%J 	,,,K{%L 	666M{%N 	555O{%P 		333Q{%R 	NNNS{% {% {%T 	ccU{%V 	ffW{%X 	ffY{%Z 	mmm[{%\ 		ss]{%^ 	NNN_{%` 	333a{%b 	888c{%d 	555e{%f 	VVVg{%h 	;;;i{%j 	"$z$z$zk{%l 	GGGm{%n 	mmmo{%p 	YYYq{%r 	(*?*?s{%t 	444u{% {% {%v 	;;w{%x 	222y{%z 	666{{%| 	777}{%~ 	888{%@ 	
...A{%B 	==C{%D 	>>E{%F 	LLLG{%H 	BBBI{%J 	==K{%L 	\\\M{%N 	))O{%P 	
GGGQ{%R 	&&S{%T 	''U{%V 	222W{% {% {%X 	222Y{%Z 	tt]{%` 	((a{%b 	111c{%d 	444e{%f 	KKKg{%h 	**i{%j 	''k{%l 	&&m{%n 	..o{%p 	,,q{%r 	!1!1s{%t 	**u{%v 	33w{%x 	))y{%z 	WWW{{%| 	%%}{% {% {%~ 	 d  d  dA	{%D	 	rrrE	{%F	 	
++G	{%H	 	NNNI	{%J	 	"$c$c$cK	{%L	 	!L!L!LM	{%N	 	 S S SO	{%P	 	sssQ	{%R	 		444S	{%T	 	666U	{%V	 	333W	{%X	 	;;;Y	{%Z	 	
;;;[	{%\	 	000]	{%^	 	  K  K  K_	{%`	 		---a	{%b	 	<<<c	{% {% {%d	 	///e	{%f	 	///g	{%h	 	
...i	{%j	 	:::k	{%l	 	;;;m	{%n	 	&&o	{%p	 	...q	{%r	 	<<<s	{%t	 	555u	{%v	 	;;;w	{%x	 	<<<y	{%z	 	///{	{%|	 	III}	{%~	 	
ss	{%@
 	OOOA
{%B
 	333C
{%D
 	
555E
{% {% {%F
 	EEEG
{%H
 	BBBI
{%J
 	!  $A  $A  $AK
{%L
 	88M
{%N
 	<<O
{%P
 	===Q
{%R
 	777S
{%T
 	???U
{%V
 	 F F FW
{%X
 	!#]#]#]Y
{%Z
 	[[[[
{%\
 	>>>]
{%^
 	//_
{%`
 	!#@#@#@a
{%b
 	 "M"M"Mc
{%d
 	 < < <e
{%f
 	**g
{% {% {%h
 	!1!1i
{%j
 		---k
{%l
 	888m
{%n
 	
555o
{%p
 	lllq
{%r
 	llls
{%t
 	lllu
{%v
 	%ttty
{%|
 	
...}
{%~
 	VVV
{%@ 	000A{%B 	333C{%D 	55E{%F 		---G{%H 	888I{%J 	
555K{%L 	ttO{% {% {%R 	 "}"}S{%T 	)+v+vU{%V 	%'h'hW{%X 	*ww[{%^ 	*gga{%d 	 d  dg{%j 	BBBk{%l 	
EEEm{%n 	<<o{%p 	==q{%r 	AAAs{%t 		444u{%v 	999w{%x 	UUUy{%z 	111{{%| 	++}{%~ 	::{% {% {%@ 	WWWA{%B 	!s!sC{%D 	&(_(_E{%F 	88G{%H 	!f!f!fI{%J 	YYYK{%L 	!V!VM{%N 	UUUO{%P 	$&>&>Q{%R 	333S{%T 	:::U{%V 		---W{%X 	222Y{%Z 	:::[{%\ 	/1N1N]{%^ 	/1N1N_{%` 	/1d1d1da{% {% {%b 	<>q>q>qc{%d 	/1d1d1de{%f 	<>q>q>qg{%h 	/1d1d1di{%j 	<>q>q>qk{%l 	')S)S)Sm{%n 	)+a+a+ao{%p 	& B  B  Bs{%v 	& B  B  By{%| 	&xxx{%B 	$&R&R&RC{%D 	0cccG{%J 	<tttM{%P 	 "L"L"LQ{%R 	1iiiU{%X 	) L  L  L[{%^ 	$xxxa{% {% {%d 	#%Z%Z%Ze{%f 	%'\'\'\g{%h 	%'\'\'\i{%j 	%'\'\'\k{%l 	!K!K!Km{%n 	%|||q{%t 	) J  J  Jw{%z 	13i3i3i{{%| 	 "m"m"m}{%~ 	1zzzA{%D 	>zzzG{%J 	1zzzM{%P 	>zzzS{%V 	-/u/u/uW{%X 	 "F"F"FY{%Z 	!9!9!9[{%\ 	')z)z)z]{% {% {%^ 	&(g(g(g_{%` 	*,c,c,ca{%b 	&(C(C(Cc{%d 	$&`&`&`e{%f 	0bbbi{%l 	) I  I  Io{%r 	' M  M  Mu{%x 	"  %A  %A  %Ay{%z 	#%|%|%|{{%| 	&(m(m(m}{%~ 	&(\(\(\{%@ 	"$G$G$GA{%B 	/1g1g1gC{%D 	')^)^)^E{%F 	&(8(8G{%H 	%'m'm'mI{%J 	%'m'm'mK{% {% {%L 	%'m'm'mM{%N 	/iiiQ{%T 	&tttW{%Z 	3ttt]{%` 	&tttc{%f 	3ttti{%l 	&ttto{%r 	3tttu{%x 	(*z*z*zy{%z 	(*z*z*z{{%| 	(*z*z*z}{%~ 	$&}&}&}{%@ 	8 _  _  _C{%F 	-tttI{%L 	2VVVO{%R 	7cccU{%X 	$vvv[{% {% {%^ 	%'X'X'X_{%` 	#%F%F%Fa{%b 	!P!P!Pc{%d 	-/a/a/ae{%f 	,}}}i{%l 	!#;#;m{%n 	 "A"A"Ao{%p 	!#B#B#Bq{%r 	$&_&_&_s{%t 	!#y#y#yu{%v 	 "A"A"Aw{%x 	 "A"A"Ay{%z 	 "A"A"A{{%| 	8:u:u:u}{%~ 	*  -A  -A  -A{%@ 	&(j(j(jA{%B 	,.x.x.xC{% {% {%D 	#%Z%Z%ZE{%F 	#%Z%Z%ZG{%H 	$&L&L&LI{%J 	&(C(C(CK{%L 	$&6&6M{%N 	&(8(8O{%P 	%'X'X'XQ{%R 	/ L  L  LU{%X 	=DHQT[`lrvvvv[{%^ 	"$b$b$b_{%` 	 O O Oa{%b 	SSSc{%d 	!7!7e{%f 	&(x(x(xg{%h 	777i{%j 	FFFFk{%l 	((m{% {% {%n 	
\\\o{%p 	dddq{%r 	 "h"h"hs{%t 	  +/#3 #3 #3u{%| 	999}{%~ 	ddd{%@ 	%%A{%B 	**C{%D 	QQQE{%F 	!S!S!SG{%H 	++I{%J 	IIIK{%L 	***M{%N 	555O{%P 	IIIQ{%R 	==S{%T 	AAU{% {% {%V 	777W{%X 	 Y YY{%Z 	666[{%\ 	222]{%^ 	--_{%` 	ddda{%b 		777c{%d 	
000e{%f 		DDDg{%h 	 "2"2i{%j 	"$4$4k{%l 	')9)9m{%n 	''o{%p 	,,q{%r 	777s{%t 	CCCu{%v 	fffw{% {% {%x 	iiiy{%z 	"$V$V{{%| 	!#M#M}{%~ 	)+P+P{%@ 	"$s$sA{%B 	  a  aE{%H 	! a  aK{%N 	"J# # #Q{%X 	"L # # #[{%d 	"O 	# # #g{%r 	% a  au{%x 	% a  a{{%~ 	111{%@ 	%%A{%B 	
...C{%D 	
555E{%F 	FFFG{% {% {%H 	,,I{%J 	//K{%L 	444M{%N 	
333O{%P 	:::Q{%R 	AAAS{%T 	!;!;!;U{%V 	..W{%X 	QQQY{%Z 	xx[{%\ 	SSS]{%^ 	xx_{%` 	SSSa{%b 	
777c{%d 	777e{%f 	///g{%h 	555i{% {% {%j 	PPk{%l 	bbbm{%n 	///o{%p 	
444q{%r 	MMMs{%t 	88u{%v 	<<w{%x 	ZZZy{%z 	eee{{%| 	|||}{%~ 	22{%@ 	??A{%B 	WWWC{%D 	WWWE{%F 	
333G{%H 	111I{%J 	
...K{% {% {%L 	111M{%N 		---O{%P 		---Q{%R 	
...S{%T 	
...U{%V 	''W{%X 	..Y{%Z 		999[{%\ 	
:::]{%^ 	888_{%` 	@@@a{%b 	WWWc{%d 	
YeQUYYYYe{%f 	EEEg{%h 	 P P Pi{%j 	
...k{%l 	000m{% {% {%n 	;;;o{%p 	OOOq{%r 	888s{%t 		---u{%v 	222w{%x 	
 @  @  @{{%~ 		444{%@ 	999A{%B 		---C{%D 	))E{%F 	''G{%H 	I{%J 	K{%L 	''M{%N 	))O{%P 	Q{%R 	))S{% {% {%T 	((U{%V 	))W{%X 	((Y{%Z 	))[{%\ 	((]{%^ 	))_{%` 	))a{%b 	))c{%d 	))e{%f 	000g{%h 		IIIi{%j 	AAAk{%l 	HHHm{%n 	888o{%p 	44q{%r 	66s{%t 	//u{% {% {%v 	!1!1w{%x 	!1!1y{%z 	!1!1{{%| 	!1!1}{%~ 	,.K.K.K{%@ 	,.K.K.KA{%B 	,.K.K.KC{%D 	,.K.K.KE{%F 	//G{%H 	,,I{%J 	++K{%L 	,,M{%N 	--O{%P 	..Q{%R 	,,S{%T 	--U{%V 	--W{% {% {%X 	 A A AY{%Z 	!B!B!B[{%\ 	//]{%^ 	*,I,I,I_{%` 	+-J-J-Ja{%b 	**c{%d 	++e{%f 	**g{%h 	++i{%j 	+-J-J-Jk{%l 	+-J-J-Jm{%n 	--o{%p 	 0 0q{%r 	!#D#D#Ds{%t 	--u{%v 	!O!O!Ow{%x 	(*:*:y{% {% {%z 	(*:*:{{%| 	(*:*:}{%~ 	(*:*:{%@ 	"$7$7A{%B 	,,C{%D 	--E{%F 	!>!>!>G{%H 	++I{%J 	--K{%L 	/1A1AM{%N 	/1A1AO{%P 	46S6S6SQ{%R 	46S6S6SS{%T 	46S6S6SU{%V 	46S6S6SW{%X 	,,Y{%Z 	@@@[{% {% {%\ 	)+;+;]{%^ 	@@@_{%` 	>>>a{%b 	<<<c{%d 	!!e{%f 	
++g{%h 	KKKi{%j 		---k{%l 	
...m{%n 	 3 3 3o{%p 	 "<"<"<q{%r 	:::s{%t 	HHHu{%v 	JJJw{%x 	
**y{%z 	
KKK{{%| 	%%}{% {% {%~ 	55{%@ 	111A{%B 	555C{%D 	 e e eE{%F 	%'a'a'aG{%H 	
:::I{%J 	! L  L  LM{%P 	
:::Q{%R 	22S{%T 	///U{%V 	---W{%X 	<<Y{%Z 	hhh[{%\ 	 "g"g"g]{%^ 	66_{%` 	;;;a{%b 	LLLc{% {% {%d 	%'W'W'We{%f 	888g{%h 	111i{%j 		---k{%l 	222m{%n 	;;o{%p 	222q{%r 	999s{%t 	%'`'`u{%v 	1ooy{%| 	eee}{%~ 	55{%@ 	@@A{%B 	C{%D 	"OOE{%F 	//G{%H 	ooI{% {% {%J 	QQQK{%L 	')>)>M{%N 	FFFO{%P 	%CCCCQ{%R 	>>S{%T 	11U{%V 	!#@#@W{%X 	66Y{%Z 	??[{%\ 	NNN]{%^ 	<<<_{%` 	#%H%H%Ha{%b 	00c{%d 	ooe{%f 	99g{%h 	22i{%j 	__k{% {% {%l 	OOm{%n 	OOo{%p 	??q{%r 	s{%t 	u{%v 	w{%x 	y{%z 	111{{%| 	//}{%~ 	AA{%@ 	//A{%B 	33C{%D 	44E{%F 	44G{%H 	22I{%J 	33K{%L 	33M{% {% {%N 	11O{%P 	22Q{%R 	22S{%T 	11U{%V 	22W{%X 	22Y{%Z 	..[{%\ 	--]{%^ 	.._{%` 	//a{%b 	OOc{%d 	00e{%f 	g{%h 	33i{%j 	k{%l 	??m{%n 	..o{% {% {%p 	//q{%r 	//s{%t 	55u{%v 	00w{%x 	22y{%z 	{{%| 	}{%~ 	//{%@ 	A{%B 	77C{%D 	44E{%F 	__G{%H 	AAAAI{%J 	11K{%L 	//M{%N 	//O{%P 	//Q{% {% {%R 		??S{%T 		??U{%V 	&W{%X 	*2OOY{%Z 	oo[{%\ 	]{%^ 	___{%` 	ooa{%b 	c{%d 	e{%f 	!??g{%h 	i{%j 	-/p/pk{%l 	*,V,Vm{%n 	2OOo{%p 	__q{%r 	s{% {% {%t 	oou{%v 	w{%x 	y{%z 	{{%| 	}{%~ 	{%@ 	A{%B 	#__C{%D 	E{%F 	G{%H 	I{%J 	 //K{%L 	M{%N 	 //O{%P 	#__Q{%R 	 //S{%T 	$ooU{% {% {%V 	 //W{%X 	Y{%Z 	[{%\ 	__]{%^ 	oo_{%` 	a{%b 	__c{%d 	 //e{%f 	$oog{%h 	ooi{%j 	k{%l 	__m{%n 	__o{%p 	'/q{%r 	NNNs{%t 	oou{%v 	oow{% {% {%x 	y{%z 	{{%| 	__}{%~ 	__{%@ 	OOA{%B 	__C{%D 	OOE{%F 	==G{%H 	I{%J 	K{%L 	//M{%N 	==O{%P 	00Q{%R 	88S{%T 	99U{%V 	kkkW{%X 	E4IMMMY{% {% {%Z 	0EIII[{%\ 	0EIII]{%^ 	0EIII_{%` 	MTMMMMa{%b 	c{%d 	66e{%f 	e6MQQQg{%h 	>>>i{%j 	
u/DHHHk{%l 	0EIIIm{%n 	0EIIIo{%p 	
u/DHHHq{%r 	
u/DHHHs{%t 	u{%v 	//w{%x 	!O!O!Oy{%z 	
OO{{% {% {%| 	@@@}{%~ 	%2GKKK{%@ 	53HLLLA{%B 	__C{%D 	,,E{%F 	00G{%H 	HHHHHI{%J 	,,K{%L 	55M{%N 	1FJJJO{%P 	%2GKKKQ{%R 	@@@@S{%T 	??U{%V 	0EIIIW{%X 	1FJJJY{%Z 	//[{%\ 	]{% {% {%^ 	
u/DHHH_{%` 	__a{%b 	ooc{%d 	__e{%f 	//g{%h 	11i{%j 	//k{%l 	__m{%n 	MTMMMMo{%p 	00q{%r 	0EIIIs{%t 	66u{%v 	55w{%x 		888y{%z 	@@@{{%| 	EE}{%~ 	??{% {% {%@  	A {%B  	"OOC {%D  	-E {%F  	%G {%H  	I {%J  	ooK {%L  	,,M {%N  	??O {%P  	GGGQ {%R  	S {%T  	LDLLLLU {%V  	55W {%X  	33Y {%Z  	33[ {%\  	13H3H] {%^  	,,_ {%`  	--a {% {% {%b  	BBBc {%d  	11e {%f  	--g {%h  	--i {%j  	00k {%l  	 "8"8m {%n  	OOo {%p  	[[[q {%r  	??s {%t  	oou {%v  	1FJJJw {%x  	__y {%z  	WWW{ {%|  	??} {%~  	11 {%@! 	&(W(W(WA!{%B! 	GGC!{% {% {%D! 	')Q)QE!{%F! 	OOOG!{%H! 	I!{%J! 	K!{%L! 	M!{%N! 	__O!{%P! 	11Q!{%R! 	++S!{%T! 		EUZUjnnnU!{%V! 	IIIIIW!{%X! 	GGGY!{%Z! 	//[!{%\! 	]!{%^! 	//_!{%`! 	..a!{%b! 	==c!{%d! 	777e!{% {%f! 	++..ooddd //++FFFu!{% {%C|! 	(B  v/00 YFFF 	GF4556 JYGF<":<<==EFD#%%G		  1 JJ1:$AJ%AJ%
 :  ,, 	 jZ!2!23GLL$&$(>RV@VW    	 	D64..D~~ $c//d'6I6IT
	
 JJt
 %%'' 
((((((

 g g g!W!W!W mmm )+b+b+b	
 ![![![ +-o-o-o )+x+x+x vvv xxx #%g%g%g *,i,i,i &  )N  )N  )N !m!m!m 
ZZZ 
ZZZ  	YYY!" 	YYY#	
 	
 	
, Jr3   
dispatcherc                 x     dt           t          t          f         dt           t          t          f         f 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
    r!   r$   c                      t          j                   dt          j        dt          j        dt
          f fd            t          t          t          t
          f                   S )Nr&   r'   r$   c                       | i |}t          |          r6t          t          t          t          t
          f                   |g| R i |S  | i |S r  )r   r   r   r   r   r   )r&   r'   relevant_argsr  r!   wrappeds      r0   r#  z3wrap_torch_function.<locals>.inner.<locals>.wrapped]  s    &J777M!-00 ,"b&)733]EI  MS   4((((r3   )	functoolsr   r   r&   r'   r   r   r   )r!   r#  r  s   `@r0   r  z"wrap_torch_function.<locals>.inner\  sv    				)27 	)bi 	)B 	) 	) 	) 	) 	) 	) 	) 
		) HRV$g...r3   )r   r   r   )r  r  s   ` r0   r   r   B  sH    4/HRV$ /"b&)9 / / / / / / Lr3   r"  get_type_fnc                    |t           }t          j                                        sg S t	                      }g }| D ]} ||          }||vrt          |d          r|j        t          j        j        uru|rm|                    |           t          |          }t          |          D ]"\  }}t          | ||                    r|} n#|                    ||           |h}|g}|S )a
  Returns a list of arguments on which to call __torch_function__.

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

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

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

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

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

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

    .. _NEP-0018:
       https://numpy.org/neps/nep-0018-array-function-protocol.html
    Nr  )r
  r    _C_is_torch_function_enabledsetr   r  _disabled_torch_function_implr=  r  	enumerate
issubclassinsert)	r"  r%  overloaded_typesoverloaded_argsargarg_typer?  iold_args	            r0   _get_overloaded_argsr4  l  s7   J  8..00 	"%%%!#O ( (;s## ,,,"677 -+89: :
   ( $$X... O,,"+O"<"<  JAw!(KK,@,@AA  !  &&uc2222$,: #&%r3   
public_apir&   r'   c                    t          |          }t          t          t          |                    }t	                      rIt                      5 }|                    | |||          }ddd           n# 1 swxY w Y   |t          ur|S |D ]l}|j        }	t          |	d          r8|	j	        |u r/|	t          j        j        urt          j        dt          d            |	| |||          }|t          ur|c S m| 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.rJ  
stacklevel.zno implementation found for 'z.' on types that implement __torch_function__: c                 ,    g | ]}t          |          S rI  )r
  ).0r0  s     r0   
<listcomp>z)handle_torch_function.<locals>.<listcomp>  s    EEEcS		EEEr3   z nor in mode )r4  tuplemapr
  r   _pop_mode_temporarilyr  NotImplementedr   r7  r    r'  r*  r,   warnDeprecationWarning
__module__r  _get_current_function_mode	TypeError)r5  r"  r&   r'   r/  typesr  resultoverloaded_argtorch_func_method	func_namer  s               r0   r   r     s   T +=99O#dO,,--E '((  #$$ 	N,,ZfMMF	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N 	N''M *   +=%z22
	!*n<<!)OOOMQ"	    #":udFCC''MMM ( (@@:+>@@I	H	 	H 	HEE_EEE	H 	H  '(( >=9;;===
C..s   A33A7:A7a  Check for __torch_function__ implementations in the elements of an iterable
    or if a __torch_function__ mode is enabled.  Considers exact ``Tensor`` s
    and ``Parameter`` s non-dispatchable.  Use this to guard a call to
    :func:`handle_torch_function`; don't use it to test if something
    is Tensor-like, use :func:`is_tensor_like` instead.
    Arguments
    ---------
    relevant_args : iterable
        Iterable or arguments to check for __torch_function__ methods.
    Returns
    -------
    bool
        True if any of the elements of relevant_args have __torch_function__
        implementations, False otherwise.
    See Also
    ________
    torch.is_tensor_like
        Checks if something is a Tensor-like, including an exact ``Tensor``.
    zSpecial case of `has_torch_function` for single inputs.
    Instead of:
      `has_torch_function((t,))`
    call:
      `has_torch_function_unary(t)`
    which skips unnecessary packing and unpacking work.
    a'  Special case of `has_torch_function` that skips tuple creation.

    This uses the METH_FASTCALL protocol introduced in Python 3.7

    Instead of:
      `has_torch_function((a, b))`
    call:
      `has_torch_function_variadic(a, b)`
    which skips unnecessary packing and unpacking work.
    c                  j   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 ]m\  }}}|D ]b}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 rAd}	|j        t9                      v r(t;          |	                    ||j                            | |                              |j                   t/          |          s| d| ||<   |r|t7                      v r<d}	|t9                      v r(t;          |	                    ||j                            G| |                              |           do| |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   r7   r
  r   r	  r  endswithislowerr  object
isinstancerG  
ModuleType
__future___Featurer  r   r.  __set__r   r   AssertionErrorformatr  r  )
overridable_funcsr?  tested_namespacesnamespace_str	namespacens_funcsrK  r)   r!   r  s
             r0   _get_overridable_functionsrc  E  s    $/55E	%'	U-u/?/GH	 3S9L5M5MN	%(-UX]););<	s5<'8'89	s5<'8'89	eiUY0	%-U]););<	 /@ A6 A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#:#:::=  |'<'>'>>>,SZZ	4=-Q-QRRR%d+224<@@@D>> *88Y88E$K  ,....9  02222(It})M)MNNNi(//5555A@	6B e##r3   c                  *    t                      d         S )a  List functions that are overridable via __torch_function__

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    Independent subclasses of :class:`TorchFunctionMode` are compositional:
    modes can be pushed onto a stack using ``with MyMode():``.
    When you call functions in the PyTorch API inside your
    ``__torch_function__`` implementation, by default, they will forward on to
    the next mode on the mode stack.  If you want recursively call back into
    your current ``__torch_function__`` implementation, either explicitly
    invoke ``self.__torch_function__(...)``, or use the context manager
    ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
    API self-referential (beware of infinite loops, in this case!)
    r  r$   Nc                     d S r  rI  re  s    r0   r   zTorchFunctionMode.__init__%  s    r3   rI  c                     t           r  )NotImplementedErrorr  r!   rG  r&   r'   s        r0   r  z$TorchFunctionMode.__torch_function__(  s    !!r3   c                 $    t          |            | S r  )
_push_modere  s    r0   	__enter__zTorchFunctionMode.__enter__+  s    4r3   c                 "    t                       d S r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r0   __exit__zTorchFunctionMode.__exit__/  s    r3   c                 B    t          j        dd            | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`rJ  r8  )r,   rB  )clsr&   r'   instances       r0   pushzTorchFunctionMode.push2  s<    ^	
 	
 	
 	
 3'''r3   )r$   NrI  N)r  rD  __qualname____doc____annotations__r   r  rz  r  classmethodr  rI  r3   r0   rs  rs    s          <    " " " "       [  r3   rs  c                  R    t                      } | dk    rt          | dz
            nd S )Nr   rg  )r   r   	stack_lens    r0   rE  rE  <  s-    )++I4=MM!)a-000tKr3   c                  P    t                      } d t          |           D             S )Nc                 ,    g | ]}t          |          S rI  )r   )r<  r2  s     r0   r=  z4_get_current_function_mode_stack.<locals>.<listcomp>C  s!    @@@!"1%%@@@r3   )r   r   r  s    r0    _get_current_function_mode_stackr  A  s)    )++I@@uY/?/?@@@@r3   c                 $    t          |            d S r  )r   )r  s    r0   ry  ry  F  s    !$'''''r3   c                  "    t                      } | S r  )r   olds    r0   r|  r|  J  s    
#
%
%CJr3   c               #   v   K   t                      } 	 | V  t          |            d S # t          |            w xY wr  )r|  ry  r  s    r0   r@  r@  O  s<      
++C			3
3s   ' 8c                       e Zd ZddZdS )BaseTorchFunctionModerI  Nc                     |i } ||i |S r  rI  rw  s        r0   r  z(BaseTorchFunctionMode.__torch_function__Y  s!    >FtT$V$$$r3   r  )r  rD  r  r  rI  r3   r0   r  r  X  s(        % % % % % %r3   r  c               #   <  K   t           j                                        } 	 t           j                            t           j        j        j                   d V  t           j                            |            d S # t           j                            |            w xY wr  )r    r'  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r0   _enable_torch_functionr  _  s}      2244I6**58+G+OPPP**955555**95555s   7A: :!Bc               #      K   t           j                                        5  	 d V  n# w xY w	 d d d            d S # 1 swxY w Y   d S r  )r    r'  _RestorePythonTLSSnapshotrI  r3   r0   r   r   i  s       
	+	+	-	-  	EEEDDDDD	                 s   ;(;*;??)r   r    r  )Dr  rY  rO  
contextlibr$  r;  rG  r,   collections.abcr   r   r   typingr   r   r   typing_extensionsr	   r    torch._Cr
   r   r   r   r   r   r   r   r   rR  r   r   ri  r2   cacher)  r   rD  dictr   r   r
  rQ  r4  r&   r'   r   r   r   r   r>  rc  r   r   rn  r
  r   r   rs  rE  r  ry  r|  contextmanagerr@  r  r  r   rI  r3   r0   <module>r     s   ,                 



   . . . . . . . .       % % % % % % % % % % ' ' ' ' ' ' 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
   Yt__WT]]
 1   
2r6
     b"f	       F `s8} ` ` `  `F	 c(m    2 XtHh$67 X X X  Xv$'Xc]*+'xB (2r6"223' ' ' 'X 15L LC=L3%+&-L 
#YL L L L^VR VC=V 7V i	V
 V V V Vr ![  . ';	 	  *k 	   Q$Ed8n	tHcM22% Q$ Q$ Q$ Q$h 	+4T(^(;#< 	+ 	+ 	+ 	+ 2 2 2( S]     Gx GD G G G G2"K "K "KJ6 6 6 6 6 6 6 6rL L L
A A A
( ( (  
   % % % % %- % % % 6 6 6     r3   