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

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

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

If changing this file in a way that can affect ``__torch_function__`` overhead,
please report the benchmarks in ``benchmarks/overrides_benchmark``. See the
instructions in the ``README.md`` in that directory.
    N)Iterable)wraps)AnyCallableOptionalTypeVar)	ParamSpec)	_add_docstr_get_function_stack_at_has_torch_function_has_torch_function_unary_has_torch_function_variadic_is_torch_function_mode_enabled_len_torch_function_stack_pop_torch_function_stack_push_on_torch_function_stack)
get_ignored_functionsget_overridable_functionsget_testing_overrideshandle_torch_functionhas_torch_functionresolve_nameis_tensor_likeis_tensor_method_or_propertywrap_torch_functionenable_reentrant_dispatch_P_R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 Y   y xY w)Nignore)categorymessager!   )warningscatch_warningsfilterwarningsUserWarning)r$   r%   r   r!   r    s     I/var/www/html/engine/venv/lib/python3.12/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperZ   sJ    $$& 	)##;f ((		) 	) 	)s   %AA)r   r   r$   r%   r   )r   r    r!   r/   s   ``` r.   _disable_user_warningsr0   B   s?    0 4[)rww )")) ) ) ) N    c                  #   t         j                  } h t         j                  t         j                  t         j                  t         j
                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                   t         j"                  t         j$                  t         j&                  t         j(                  t         j*                  t         j,                  t         j.                  t         j0                  t         j2                  t         j4                  t         j6                  t         j8                  t         j:                  t         j<                  t         j>                  t         j@                  t         jB                  t         jD                  t         jF                  t         jH                  t         jJ                  t         jL                  t         jN                  t         jP                  t         jR                  t         jT                  t         jV                  t         jX                  t         jZ                  t         j\                  t         j^                  t         j`                  t         jb                  t         jd                  t         jf                  t         jh                  t         jj                  t         jl                  t         jn                  t         jp                  t         jr                  t         jt                  t         jv                  t         jx                  t         jz                  t         j|                  t         j~                  t         j                  t         j                  t         j                  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                  t         j                  j                  t         j                  j                  t         j                  j                  t         j                  t         j                  j                  t         j                  j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  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                  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                  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         j0                  t         j2                  t         j4                  t         j6                  t         j8                  t         j:                  t         j<                  t         j>                  t         j@                  t         jB                  t         jD                  t         jF                  t         jH                  t         jJ                  t         jL                  t         jN                  t         jP                  t         jR                  t         jT                  t         jV                  t         jX                  t         jZ                  t         j                  j                  j\                  t         j^                  t         j`                  t         jb                  t         jd                  t         jf                  t         jh                  t         jj                  t         jl                  t         jn                  t         jp                  t         jr                  t         jt                  t         jv                  t         jx                  t         jz                  t         j|                  t         j~                  t         j                  t         j                  t         j                  t         j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  j                  | j                  j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j,                  | j                  S )a%  
    Return public functions that cannot be overridden by ``__torch_function__``.

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

    Examples
    --------
    >>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions()
    True
    >>> torch.add in torch.overrides.get_ignored_functions()
    False
    (   torchTensortypename	is_tensor
is_storageset_default_tensor_typeset_default_deviceget_default_deviceset_rng_stateget_rng_statemanual_seedinitial_seedseedsaveloadset_printoptionsforkget_default_dtypeget_num_interop_threadsget_num_threadsinit_num_threadsimport_ir_moduleimport_ir_module_from_bufferis_anomaly_enabledis_anomaly_check_nan_enabledis_grad_enabledmerge_type_from_type_commentparse_irparse_schemaparse_type_commentset_anomaly_enabledset_flush_denormalset_num_interop_threadsset_num_threadswait	as_tensor
from_numpytensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutalign_tensorsarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptableemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrandrandnrandintrandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   sigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvitals_enabled	set_vitalread_vitalsvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_make_dtensor_make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_countr4   s    r.   r   r   e   sK   ( \\FAAA 	A 	%%	A
 	  A 	  A 	A 	A 	A 	A 	

A 	

A 	

A 	A 	

A  	!A" 	%%#A$ 	%A& 	'A( 	)A* 	**+A, 	  -A. 	**/A0 	1A2 	**3A4 	5A6 	7A8 	  9A: 	!!;A< 	  =A> 	%%?A@ 	AAB 	

CAD 	EAF 	GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	_A` 	aAb 	cAd 	eAf 	gAh 	iAj 	kAl 	mAn 	''oAp 	qAr 	sAt 	uAv 	wAx 	yAz 	{A| 	}A~ 	A@ 	AAB 	))CAD 	EAF 	GAH 	))IAJ 	$$KAL 	((MAN 	  OAP 	!!QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	''_A` 	aAb 			cAd 			eAf 			gAh 	iAj 	

kAl 	

mAn 	oAp 	qAr 	sAt 	uAv 	wAx 	((yAz 	  {A| 	}A~ 	A@ 	,,AAB 	CAD 	EAF 	

GAH 	IAJ 	

KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	&&[A\ 	]A^ 	_A` 	aAb 	cAd 	!!eAf 	**gAh 	iAj 	kAl 	mAn 	oAp 	qAr 	,,sAt 	..uAv 	$$wAx 	--yAz 	,,{A| 	..}A~ 	44A@ 	77AAB 	11CAD 	##EAF 	''GAH 	  IAJ 	++KAL 	**MAP 	$$QAT 	UAV 	WAX 	YAZ 	[A\ 	]A^ 	$$_A` 	##aAb 	%%cAd 	$$eAf 	  gAh 	iAj 	%%kAl 	mAn 	oAp 	""qAr 	!!sAt 	  uAv 	  wAx 	""yAz 	&&{A| 	%%}A~ 	&&A@ 	%%AAB 	&&CAD 	%%EAF 	$$GAH 	$$IAJ 	$$KAL 	$$MAN 	$$OAP 	$$QAR 	$$SAT 	$$UAV 	((WAX 	((YAZ 	''[A\ 	((]A^ 	%%_A` 	!!aAb 	22cAd 	**eAf 	;;gAh 	**iAj 	kAl 	**mAn 	**oAp 	**qAr 	sAt 	$$uAv 	wAx 	yAz 	{A| 	}A~ 	

A@ 	

AAB 	CAD 	EAF 	--GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	  UAV 	WAX 	YAZ 	!![A\ 	!!]A^ 	_A` 	aAb 	cAd 	eAf 	gAh 	

iAj 	kAl 	mAn 	

oAp 	qAr 	sAt 	  uAv 	wAx 	yAz 	{A| 	}A~ 	A@ 	AAB 	CAD 	EAF 	GAH 	IAJ 	KAL 	MAN 	OAP 	QAR 	SAT 	UAV 	WAX 	YAZ 	[A\ 	""]A^ 	_A` 	aAb 	  cAd 	$$eAf 	gAh 	%%iAj 	''kAl 	**22mAn 	oAp 	qAr 	sAt 	uAv 	wAx 	yAz 	{A| 	  }A~ 	A@ 	AA Ar1   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).
    )r3   r4   _baser&  grad_gradr1  s    r.   get_default_nowrap_functionsr6  ~  s>    $ \\F r1   c                  ^   t         j                  } i t         j                  ddt         j                  ddt         j                  d t         j
                  d t         j                  ddt         j                  d t         j                  ddt         j                  dd	t         j                  dd
t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                   ddt         j"                  ddt         j$                  d i t         j&                  ddt         j(                  ddt         j*                  ddt         j,                  ddt         j.                  ddt         j0                  ddt         j2                  ddt         j4                  ddt         j6                  d t         j8                  d t         j:                  ddt         j<                  ddt         j>                  d  t         j@                  dd!t         jB                  dd"t         jD                  dd#t         jF                  dd$i t         jH                  dd%t         jJ                  dd&t         jL                  dd't         jN                  dd(t         jP                  dd)t         jR                  d* t         jT                  d+ t         jV                  d, t         jX                  dd-t         jZ                  dd.t         j\                  d/ t         j^                  d0 t         j`                  d1 t         jb                  d2 t         jd                  d3 t         jf                  d4 t         jh                  d5 i t         jj                  d6 t         jl                  dd7t         jn                  d8 t         jp                  dd:t         jr                  dd;t         jt                  dd<t         jv                  dd=t         jx                  dd>t         jz                  dd?t         j|                  dd@t         j~                  ddAt         j                  ddBt         j                  dC t         j                  ddDt         j                  dE t         j                  dF t         j                  ddGi t         j                  dH t         j                  ddIt         j                  ddJt         j                  ddKt         j                  ddLt         j                  ddMt         j                  ddOt         j                  ddPdQt         j                  dR t         j                  ddSt         j                  j                  ddTt         j                  j                  ddUt         j                  ddVt         j                  ddWt         j                  dX t         j                  ddYt         j                  ddZi t         j                  dd[t         j                  dd\t         j                  dd]t         j                  dd^t         j                  dd_t         j                  d` t         j                  ddat         j                  db t         j                  ddct         j                  dd t         j                  j                  ddet         j                  ddft         j                  ddgt         j                  ddht         j                  ddit         j                  ddjt         j                  ddki t         j                  ddlt         j                  ddmt         j                  dn t         j                  ddot         j                  ddpt         j                  ddqt         j                  ddrt         j                  ds t         j                  ddtt         j                  ddut         j                  ddvt         j                  ddwt         j                  dx t         j                  ddyt         j                  j                  dd{t         j                  dd|t         j                  dd}i t         j                  dd~t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  j                  d t         j                   d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j
                  ddt         j                  j
                  ddt         j                  ddi t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                   d t         j"                  ddt         j                  j$                  ddt         j                  j&                  ddt         j                  j(                  ddt         j                  j*                  ddt         j,                  d t         j.                  ddi t         j0                  ddt         j2                  ddt         j4                  ddt         j6                  d t         j8                  ddt         j:                  ddt         j<                  ddt         j>                  ddt         j@                  ddt         jB                  ddt         jD                  d t         jF                  d t         jH                  ddt         jJ                  d t         jL                  d t         jN                  d t         jP                  d i t         jR                  d t         jT                  d t         jV                  d t         jX                  d t         jZ                  d t         j\                  j^                  ddt         j\                  j`                  ddt         j\                  jb                  ddt         j\                  jd                  ddt         j\                  jf                  ddt         j\                  jh                  ddt         j\                  jj                  ddt         j\                  jl                  ddt         j\                  jn                  ddt         j\                  jp                  ddt         j\                  jr                  ddt         j\                  jt                  ddi t         j\                  jv                  ddt         j\                  jx                  ddÄt         j\                  jz                  ddĄt         j\                  j|                  ddńt         j\                  j~                  ddƄt         j\                  j                  ddǄt         j\                  j                  ddȄt         j\                  j\                  ddɄt         j                  ddʄt         j                  dd˄t         j                  d̄ t         j                  d̈́ t         j                  d΄ t         j                  ddτt         j                  ddЄt         j                  dф t         j                  dd҄i t         j                  ddӄt         j                  ddԄt         j                  ddՄt         j                  ddt         j                  ddfdքt         j                  dׄ t         j                  dd؄t         j                  d dلt         j                  ddڄt         j                  ddۄt         j                  d܄ t         j                  dd݄t         j                  ddބt         j                  dd߄t         j                  ddt         j                  ddt         j                  ddt         j                  ddi t         j                  d t         j                  d t         j                  d t         j                  ddt         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  j                  d t         j                  ddi t         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  d t         j                  d	d t         j                  d t         j                  ddt         j                  d t         j                  d t         j                  dڐdi t         j                  dڐdt         j                   d t         j                  d t         j                  dڐd	t         j                  j                  dڐd
t         j                  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                   d
dt         j"                  ddt         j$                  d t         j&                  dߐdt         j                  j(                  ddt         j                  j*                  ddt         j                  j,                  ddt         j.                  ddt         j0                  dڐdt         j2                  dڐd t         j4                  dڐd!t         j6                  dڐd"t         j8                  dڐd#t         j:                  dڐd$t         j<                  dd%t         j>                  dڐd&t         j@                  dڐd'i t         jB                  dڐd(t         jD                  dڐd)t         jF                  dڐd*t         jH                  dڐd+t         jJ                  dڐd,t         jL                  d- t         jN                  dڐd.t         jP                  dڐd/t         jR                  dڐd0t         jT                  dڐd1t         jV                  dڐd2t         jX                  dڐd3t         jZ                  dd4t         j\                  d5 t         j^                  dd6t         j`                  dڐd7t         jb                  dڐd8i t         jd                  dd9t         jf                  dڐd:t         jh                  dd;t         jj                  d< t         jl                  d= t         jn                  dڐd>t         jp                  dڐd?t         j                  jd                  dd@t         j                  jr                  ddAt         j                  jt                  ddBt         j                  jf                  ddCt         j                  jp                  dڐdDt         jv                  dE t         j                  jv                  dڐdFt         j                  jx                  d dGt         j                  jz                  dڐdHt         j|                  dI i t         j                  j|                  dJ t         j~                  dڐdKt         j                  dڐdLt         j                  dڐdMt         j                  ddNt         j                  ddOt         j                  ddPt         j                  ddQt         j                  dڐdRt         j                  ddSt         j                  dڐdTt         j                  dڐdUt         j                  dV t         j                  dڐdWt         j                  dڐdXt         j                  dڐdYt         j                  dZ i t         j                  d[ t         j                  d\ t         j                  d] t         j                  d^ t         j                  d_ t         j                  d` t         j                  ddat         j                  ddbt         j                  dc t         j                  dd t         j                  ddet         j                  dڐdft         j                  dڐdgt         j                  ddht         j                  dڐdit         j                  dj t         j                  dk i t         j                  ddlt         j                  dm t         j                  dn t         j                  do t         j                  ddpt         j                  ddqt         j                  dr t         j                  ddst         j                  dt t         j                  dڐdut         j                  dڐdvt         j                  dڐdwt         j                  dڐdxt         j                  dڐdyt         j                  j                  j                  dz t         j                  j                  j                  d{ t         j                  j                  j
                  dސd|i t         j                  j                  j                  dސd}t         j                  j                  j                  dސd~t         j                  j                  j                  dސdt         j                  j                  j                  dސdt         j                  j                  j                  dސdt         j                  j                  j                  dڐdt         j                  j                  j*                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j\                  ddt         j                  j                  jn                  dڐdt         j                  j                  j                  ddt         j                  j                  jp                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddi t         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j.                  ddt         j                  j                  j0                  ddt         j                  j                  jX                  ddt         j                  j                  j                   ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j
                  d dt         j                  j                  j                  d!dt         j                  j                  j                  d"dt         j                  j                  j                  d#di t         j                  j                  j                  ddt         j                  j                  j                  d$dt         j                  j                  j                  ddt         j                  j                  j                  d%dt         j                  j                  j                  ddt         j                  j                  j                   d&dt         j                  j                  j                  d'dt         j                  j                  j"                  ddt         j                  j                  j                  d(dt         j                  j                  j.                  ddt         j                  j                  j                  d)dt         j                  j                  j                  dڐdt         j                  j                  j                  d*dt         j                  j                  j@                  d+dt         j                  j                  j                   d t         j                  j                  j"                  d dt         j                  j                  j$                  d di t         j                  j                  j&                  d dt         j                  j                  jh                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j(                  ddt         j                  j                  j                  ddt         j                  j                  j*                  ddt         j                  j                  j,                  d,dt         j                  j                  j.                  d,dt         j                  j                  j0                  d,dt         j                  j                  j2                  d(dt         j                  j                  j4                  d-dt         j                  j                  j6                  d.dt         j                  j                  j8                  d/dt         j                  j                  j:                  ddt         j                  j                  j<                  d0di t         j                  j                  j>                  d1dt         j                  j                  j@                  d"dt         j                  j                  jB                  d2dÄt         j                  j                  jD                  d3dĄt         j                  j                  jF                  d4dńt         j                  j                  jH                  dƄ t         j                  j                  jJ                  dސdǄt         j                  j                  jL                  dސdȄt         j                  j                  jN                  d5dɄt         j                  j                  jP                  d6dʄt         j                  j                  jR                  dސd˄t         j                  j                  jT                  dސd̄t         j                  j                  jV                  dސd̈́t         j                  j                  jX                  d7d΄t         j                  j                  jZ                  d8dτt         j                  j                  j\                  d9dЄt         j                  j                  j^                  d/dфi t         j                  j                  j`                  d+d҄t         j                  j                  jb                  d+dӄt         j                  j                  jd                  d:dԄt         j                  j                  jf                  ddՄt         j                  j                  jh                  dք t         j                  j                  jj                  dׄ t         j                  j                  jl                  dސd؄t         j                  j                  jn                  d;dلt         j                  j                  jp                  ddNdd9dڜdۄt         j                  j                  jr                  dd܄t         j                  jt                  jv                  d<d݄t         j                  jt                  jx                  d<dބt         j                  jt                  jz                  d߄ t         j                  jt                  j|                  d=dt         j~                  dސdt         j                  dzddt         j                  d i t         j                  d>dt         j                  j                  d?dt         j                  j                  d@dt         j                  j                  	 dAdt         j                  dBdt         j                  d>dt         j                  d t         j                  d t         j                  dCdt         jD                  d3dt         j                  d t         j                  dDdt         j                  ddt         j                  dEdt         j                  j                  dFdt         j                  d t         j                  d i t         j                  dڐdt         jF                  d t         j                  dڐdt         j                  dڐdt         jH                  d t         j                  ddt         j                  dڐdt         j                  dڐdt         j                  dސdt         j                  d t         j                  d  t         j                  d t         j                  d t         j                  d t         j                  ddt         j                  j                  dGdt         j                  dHdi t         j                  dHdt         j                  d t         j                  d	 t         j                  d
 t         j                  d t         j                  d t         j                  d t         j                  	 	 dIdt         j                  	 	 dJdt         j                  	 	 dKdt         j                  d t         j                  d t         j                  dڐdt         j                  ddt         j                  dt         j                  ddfdt         j                  ddt         j                  d i t         j                  dڐdt         j                  dڐdt         j                  j                  ddt         j                  d t         j                  d t         j                  dڐdt         jJ                  dސd t         j                  dڐd!t         j                  dڐd"t         j                  dڐd#t         j                  d$ t         jN                  d5d%t         j                  d& t         j                  dd't         j                  d( t         j                  dd)t         j                  dڐd*i t         j                  dLd,t         j                   dڐd-t         j                  dڐd.t         j                  d/ t         jP                  d6d0t         j                  dڐd1t         j                  dd2t         j
                  dېd3t         j                  d4 t         j                  d5 t         j                  d	d6t         j                  dd7t         j                  dMd8t         j                  d9 t         j                  d: t         j                  dNd;t         j                  dNd<i t         jR                  dސd=t         j                  dڐd>t         j                   dڐd?t         j"                  dڐd@t         j$                  dڐdAt         j&                  dڐdBt         j(                  dڐdCt         j*                  dڐdDt         j,                  dE t         j                  j,                  dF t         j.                  dڐdGt         j0                  dڐdHt         j`                  dڐdIt         j                  j2                  ddJt         j                  j4                  ddKt         j6                  dOdddLdMt         j8                  ddNi t         j:                  ddOt         j<                  dڐdPt         j>                  dڐdQt         j@                  ddRt         jB                  dېdSt         jD                  ddTt         jF                  dڐdUt         jH                  dڐdVt         jJ                  dPdWt         jL                  dڐdXt         jN                  dڐdYt         jP                  dڐdZt         jR                  d[ t         jT                  d\ t         jV                  d] t         jX                  d^ t         jZ                  d_ i t         j\                  d` t         j^                  da t         j`                  db t         jb                  dc t         jd                  dd t         jf                  de t         jh                  df t         jj                  dg t         jl                  dh t         jn                  di t         jp                  dj t         jr                  dk t         jt                  dڐdlt         jv                  dQdmt         jx                  dRdnt         j                  jv                  ddot         j                  jz                  dڐdpi t         j|                  dq t         j~                  dr t         j                  j                  ds t         j                  j                  dt t         j                  j                  du t         j                  j                  dv t         j                  j                  dw t         j                  j                  dڐdxt         j                  j                  dڐdyt         j                  j                  dڐdzt         j                  j                  dڐd{t         j                  j                  d| t         j                  j                  d} t         j                  j8                  d~ t         j                  j:                  d t         j                  j                  d t         j                  j<                  d i t         j                  j@                  d t         j                  j                  d t         j                  jB                  d t         j                  j                  dڐdt         j                  j                  dڐdt         j                  j                  d t         j                  j                  dڐ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                  jD                  d t         j                  j                  d t         j                  j@                  dڐdi t         j                  jX                  d t         j                  jZ                  ddt         j                  j                  d t         j                  j                  d t         j                  j                  d t         j                  j                  d t         j                  j                  d t         j                  j                  d t         j                  j                  d t         j                  j                  dڐdt         j                  j                  d t         j                  j                   d t         j                  j                  d t         j                  j                  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 t         j                  j`                  dڐdt         j                  j                  d t         j                  j                  dڐdt         j                  jN                  dڐdt         j                  j                  dڐdt         j                  d t         j                  d t         j                  ddt         j                  dڐdt         j                  dڐdt         j                  j                  ddt         j                  j                  dڐdt         j                  dSdt         j                  ddt         jl                  dސdi t         j                  d t         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  dTdt         j                  j                  dCdt         j                  ddt         jn                  d;dt         j                  ddt         j                  d t         j                  dڐdt         j                  ddt         j                  dÄ t         j                   dU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                  j                  dڐd˄t         j                  dڐd̄t         j                  dڐd̈́t         j                  d΄ t         j                  dڐdτt         j                  ddЄt         j                  dф t         j                  d҄ t         j                  ddӄt         j                   dԄ t         j"                  dՄ t         j$                  dք i t         j&                  dׄ t         j(                  d؄ t         j*                  dل t         j,                  dڐdڄt         j.                  dۄ t         j0                  dd܄t         j2                  ddݜdބt         j4                  d߄ t         j6                  d t         j8                  d t         j:                  d t         j<                  d t         j>                  dNdt         j@                  ddt         jB                  ddt         jD                  d t         jF                  d i t         jH                  d t         jJ                  d t         jL                  d t         jN                  d t         jP                  d t         jR                  d t         jT                  d t         jV                  d t         jX                  d t         jZ                  d t         j\                  ddt         j^                  d t         j`                  d t         jb                  d | jd                  d | jf                  d | jh                  d i | jj                  d | jl                  d | jn                  d | jp                  d | jr                  d | jt                  d | jv                  d  | jx                  d | jz                  d | j|                  d | j~                  d | j                  d | j                  d | j                  d | j                  d | j                  d	 | j                  d
 i | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddd| j                  d i | j                  d | j                  j                  d | j                  j                  d | j                  j                  d  | j                  j                  d! | j                  j                  d" | j                  j                  d# | j                  j                  d$ | j                  j                  d% | j                  j                  d& | j                  j                  d' | j                  j                  d( | j                  j                  d) | j                  j                  d* | j                  d+ | j                  d, | j                  d- i | j                  j                  d. | j                  j                  d/ | j                  j                  d0 | j                  j                  d1 | j                  j                  d2 | j                  j                  d3 | j                  j                  d4 | j                  j                  d5 | j                  j                  d6 | j                  j                  d7 | j                  j                  d8 | j                  j                  d9 | j                  j                  d: | j                  j                  d; | j                  j                  d< | j                  j                  d= | j                  j                  d> i | 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 | j                  j                  dM | j                  j                  dN | j                  d dOi | j                  dP | j                  dQ | j                  dR | j                  dS | j                  dT | j                  dU | j                  dV | j                   dW | j"                  dX | j$                  dY | j&                  dZ | j                  d[ | j(                  d\ | j*                  d] | j,                  d^ | j.                  d_ | j0                  d` i | j2                  dda| j4                  t         j6                  fdb| j8                  t         j6                  fdc| j:                  t         j6                  fdd| j<                  t         j6                  fde| j>                  d+ddfdg| j@                  dh | jB                  di | jD                  t         jF                  fdj| jH                  dސdk| jJ                  t         j6                  fdl| jL                  t         j6                  fdm| jN                  t         j6                  fdn| jP                  t         j6                  fdo| jR                  t         j6                  fdp| jT                  dq | jV                  dr i | j                  dds| jX                  dt | jZ                  dސdu| j\                  t         j6                  fdv| j^                  t         j6                  fdw| j`                  dx | jb                  dy | jd                  dz | jf                  dddfd{| jh                  d| | jj                  d} | jl                  t         j6                  fd~| jn                  t         j6                  fd| jp                  ddfd| j                  d | jr                  t         j6                  fd| jt                  t         j6                  fdi | jv                  d | jx                  d | jz                  t         j6                  fd| j|                  d | j~                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dVddfd| j@                  d | j                  t         j6                  fd| j                  d | j                  d | j                  dڐd| j                  dސdi | j4                  d | j                  d | j                  d | j                  d | j                  d | j                  d | jx                  d | j                  d | j                  d | j                  d | j                  dސd| j                  d | j                  dddfd| j                  d | j                  d | j                  d | j                  d i | j                  d | j                  d | j                  d	d| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dWd| j                  d | j                  d | j                  t         j6                  fd| j                  d | j                  dNd| j                  d | j                  d i | j                  dސd| j                  d | j                  d | jB                  dېd| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddt         j6                  fd| j                  ddddÄ| j                  ddĄ| j                  dń | j                  dƄ | j                  dǄ | j                  dȄ | jr                  dɄ | jv                  d+dʄ| j                  d˄ | j                  d̄ | j                  d̈́ | j                  d΄ | j                  dXdτ| j                  dЄ t         j                  j                  ddфi	}t         j                  j                  j                  }t        | |      r5d d҄|t        | |      <   dӄ |t        | d|       j                  <   i }t               }|j                         D ]  \  }}|j                  |j                  dz   d|j                  z   dz   d|j                  z   dz   d|j                  z   dz   g}|j                  j	                  d٫      rH|j                  t        d٫      d }|j                  d|z   dz   d|z   dz   d|z   dz   g       |D ],  }	t        | |	d      }
t        |
      s|
|vs#|
|vs(|||
<   .  |j                  |       |S (Y  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                      yN inputouts     r.   <lambda>z'get_testing_overrides.<locals>.<lambda>      r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  output_sizes     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )inputsrD  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  otherr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  batch1batch2alphabetar>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  tensor1tensor2valuer>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rW  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  mat1mat2rU  rT  r>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  matvecrU  rT  r>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  vec1vec2rU  rT  r>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  thetasizealign_cornerss      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  dims     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   Fc                      yr9  r;  )r=  rO  trolatol	equal_nans        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  ptraininplaces       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rm  keepdimr>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rz  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  msgs     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  tensorss    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  kernel_sizer  padding	ceil_modecount_include_pads         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rQ  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  )	r=  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )grad_outr=  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r=  r  r  r  input_gweight_gbias_gs           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  r  counts           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  	generatorr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  input1input2r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   r  c                      yr9  r;  r=  targetr  size_averagereduce	reduction
pos_weights          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  weights	minlengths      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  probr  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r_  	out_dtyper>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  selfri  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  
boundaries	out_int32rightr>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  rm  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )x1x2rt  compute_modes       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1         ?c                      yr9  r;  r=  rT  rv  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )r>  c                      yr9  r;  )r>  matricess     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  groupss     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  upperr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  check_errorsr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  numeln_binsratio	bit_widths        r.   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r1   c                      yr9  r;  r=  chunksrm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r1   c                      yr9  r;  r=  minmaxr>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  r>  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  
correctionfweightsaweightss       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rwith_replacements      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )realimags     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )absangs     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  ords     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  padrZ  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  dilationr  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  )	r=  r  r  r  r  r  
transposedoutput_addingr  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1   c                      yr9  r;  r=  r  r  r  r  output_paddingr  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r1   c                      yr9  r;  r%  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r1   c                      yr9  r;  r%  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>#  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r1   c                      yr9  r;  r  r  r  marginr  r  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>&  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r1   c                      yr9  r;  )r  r  rm  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r1   c                      yr9  r;  r=  rO  rm  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r1   r:  c                      yr9  r;  r2  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r1   c                      yr9  r;  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r.   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r1   c                      yr9  r;  r=  rm  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r1   c                      yr9  r;  r=  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r1   c                      yr9  r;  r=  rm  r>  r^   s       r.   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r1   c                      yr9  r;  r@  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r1   c                      yr9  r;  yxrm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r1   c                      yr9  r;  r=  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r1   c                      yr9  r;  r=  diagonalr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r1   c                      yr9  r;  rM  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r1   c                      yr9  r;  )r=  offsets     r.   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r1   c                      yr9  r;  )r=  nrm  prependappendr>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r1   c                      yr9  r;  r=  rQ  dim1dim2s       r.   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r1   c                      yr9  r;  rW  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r1   c                      yr9  r;  )r=  srcrQ  rX  rY  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r1   c                      yr9  r;  )r  r\  ri  r  storage_offsets        r.   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r1   c                      yr9  r;  )r=  rO  rt  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r1   c                      yr9  r;  r=  rO  rounding_moder>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r1   c                      yr9  r;  rb  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r1   c                      yr9  r;  rs  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r1   c                      yr9  r;  r=  r_  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r1   c                      yr9  r;  )r^  r_  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r1   c                      yr9  r;  r=  indices_or_sectionss     r.   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r1   c                      yr9  r;  r=  UPLOr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r1   c                      yr9  r;  rq  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r1   c                      yr9  r;  )equationoperandss     r.   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r1   c                      yr9  r;  r=  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r.   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r1   c
                      yr9  r;  )
r=  r  offsetsrz  r{  r|  moder   per_sample_weightsry  s
             r.   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r1   c                      yr9  r;  r=  r^   rl   r]   requires_grads        r.   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r1   c                      yr9  r;  r=  rO  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r1   c                      yr9  r;  )r=  scale
zero_pointaxis	quant_min	quant_maxs         r.   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r1   c                      yr9  r;  )r=  r  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r1   c                      yr9  r;  )rE  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r.   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r1   c                      yr9  r;  r=  packed_weightr  outputs       r.   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r1   c                      yr9  r;  r=  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r1   c                      yr9  r;  )r=  abs      r.   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r1   c                      yr9  r;  r=  rt  ru  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r1   c                      yr9  r;  r=  rS  rm  norms       r.   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>s  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r1   c                      yr9  r;  r=  srm  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  	start_dimend_dims      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  dimss     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rz  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  exponentr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  
fill_valuer>  r^   rl   r]   r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  	dep_tokens      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  indexr>  sparse_grads        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  re  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  spacingrm  
edge_orders       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  gridinterpolation_modepadding_moderj  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  
num_groupsr  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  	r=  hxparams
has_biases
num_layersdropoutru  bidirectionalbatch_firsts	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  w_ihw_hhb_ihb_hhs         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  lambds     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  r{  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  valuesr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r-  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  binsr  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  densityr>  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r   r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  taus     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r^  r_  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rk  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rm  r  sources       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r&  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  indicesr  
accumulates       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  r  rZ  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  r  r'  r  include_inputs         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rX   s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )eteassume_uniqueinverts       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r1  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r1  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  )	r=  r  r  r  r  use_input_statsr  r  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rO  rtolrp  rq  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c
                      yr9  r;  )
r=  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  
log_targets         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  krm  r{  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  	hermitianr  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r^  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )LDpivotsBr^  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  normalized_shaper  r  espr  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  endr  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r\  rc  XrS  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rm  r^   s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rE  rD  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  namesr{  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  )	databatch_sizesr  r   r  r  r  ru  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )Apivot	get_infosr>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r,  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  maskrZ  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r'  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1   c                      yr9  r;  )LUrb  rc  leftadjointr>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rS  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rS  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rr  r^  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  return_indicesr  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  r{  r^   r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                       yr9  r;  )r  r%   s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  exponential_average_factorepsilons           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  	r=  r  r  r  r  r  r  	benchmarkdeterministics	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c	                      yr9  r;  )	r=  r  zrT  r  r  r  r  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c
                      yr9  r;  )
r=  r  r  r  r&  r  r  r  r  r  s
             r.   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r1   c	                      yr9  r;  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r1   c                      yr9  r;  )r=  r  weight_stride0r  cxr  hidden_sizer  r  r  ru  r  r  dropout_states                 r.   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r1   c                      yr9  r;  rz  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r1   c                      yr9  r;  r=  r'  destinations      r.   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r1   c                      yr9  r;  )r=  
descendingr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r1   c                      yr9  r;  )r=  num_samplesreplacementr>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r1   c                      yr9  r;  )r=  rb  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r1   c                      yr9  r;  r=  rt  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r1   c                      yr9  r;  )r=  rm  startrU  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r1   c                      yr9  r;  )r=  nanposinfneginfr>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  r  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r1   c                      yr9  r;  r=  re  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r1   c                      yr9  r;  r=  re  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r1   c                      yr9  r;  )r=  r  r  NCHxWgroupr  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r1   c                      yr9  r;  )r=  rt  rm  r{  r^   s        r.   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r1   c                      yr9  r;  rC  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r1   c                      yr9  r;  rC  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r1   c                      yr9  r;  r=  rD  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r1   c                      yr9  r;  rg  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r1   c                      yr9  r;  r=  rt  r  rv  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  divisor_overrides          r.   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  r  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r1   c                      yr9  r;  r,  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r1   c                      yr9  r;  )r=  r  r  r  ignore_indexr  r  label_smoothings           r.   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r1   c                      yr9  r;  r5  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>d  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>j  r@  r1   c                      yr9  r;  rx  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r1   c                      yr9  r;  )r=  r  r~  rz  r{  r|  r  r   r  include_last_offsetry  s              r.   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r1   c                      yr9  r;  )r=  rD  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r1   c                      yr9  r;  r=  r  rD  output_ratior  _random_sampless         r.   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r1   c                      yr9  r;  )r=  r  varr   r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  approximates     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  rj  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )logitsr  hardr  rm  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  min_valmax_valrv  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r<  r  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  ri  scale_factorr  rj  recompute_scale_factor	antialiass          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rX  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  negative_sloperv  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  ri  rT  rU  r\  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rm  _stacklevelr^   s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r{  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rD  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rD  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r,  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  r  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r*  r  r  r  rD  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rP  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rP  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r9  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )querykeyrZ  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                            r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  rt  r-  r  r  r  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r
  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  r  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rt  rm  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rX   num_classess     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  rZ  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  r  rt  r  r{  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  	log_inputr   r  r  r  r  s           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rv  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  lowerr  r  rv  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rU  rV  rZ  rc  r^  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  rU  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  deltar  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rn  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r@  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r@  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rU  	thresholds      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  rZ  rv  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c
                      yr9  r;  
anchorpositivenegativer-  rt  r  swapr  r  r  s
             r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )distance_functionr-  r  r  c                     yr9  r;  )r  r  r  r  r-  r  r  s          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rX   r  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rX   r  stdr  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rX   vals     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rX   r  r  nonlinearityr  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  as_tuples     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )r  c                     yr9  r;  )r=  ri  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  rt  rm  r{  r>  r^   s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  rm  r{  r>  r^   s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      	 yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  s     13 r1   c                      yr9  r;  )vpowrm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  input3r  	transposes        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rv  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  rm  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  qrR  rq  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rconds     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r^  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  upscale_factors     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  downscale_factors     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  rx  r   r  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rz  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r^   s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r'  r+  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  somer>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  rm  r{  interpolationr>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  scaleszero_pointsr  r^   s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r^   s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r^   reduce_ranges      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  r$  r  output_scaleoutput_zero_points           r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  r  r  r	  r
  r  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s                 r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   r      c                      	 yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  s
     " r1   c                      	 yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>!  s     !# r1   c                      	 yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>'  s     !#	 r1   c                      yr9  r;  r  s                 r.   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r1   c                      yr9  r;  r  s                 r.   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r1   c                      yr9  r;  )r=  highr^   rl   r]   r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r1   c                      yr9  r;  r2  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r1   c                      yr9  r;  )r=  rt  rm  maxnormr>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r1   c                      yr9  r;  )r=  shapes     r.   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r1   c	                      yr9  r;  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r1   c	                      yr9  r;  r  s	            r.   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r1   c                      yr9  r;  r  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r1   c                      yr9  r;  )r=  shiftsr  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r1   r   r  c                      yr9  r;  )r=  r\  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r1   c                      yr9  r;  )r  r  compressed_indices_dtypes      r.   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r1   c                      yr9  r;  )r=  rO  rT  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r1   c                      yr9  r;  r]  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r1   c                      yr9  r;  r=  rm  r  r\  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r1   c                      yr9  r;  )r=  rm  r  r\  r  include_selfs         r.   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r1   c                      yr9  r;  )sorted_sequencer=  r  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>T  r@  r1   c                      yr9  r;  )r  r  lengthsr*  r~  r  unsafes          r.   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r1   c                      yr9  r;  )r=  rm  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>V  r@  r1   c                      yr9  r;  )r=  r\  rm  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r1   c                      yr9  r;  r=  r\  rm  r  rl  steps         r.   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r1   c                      yr9  r;  r%  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r1   c                      yr9  r;  r|  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>b  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r1   c                      yr9  r;  rh  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>d  r@  r1   c                      yr9  r;  rh  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r1   c                      yr9  r;  r{  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r1   c                      yr9  r;  )r  rc  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r1   c                      yr9  r;  )r  rc  r  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r1   )stabler>  c                     yr9  r;  )r=  rm  r  r7  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r1   c                      yr9  r;  rX   split_size_or_sectionsrm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>j  r@  r1   c                      yr9  r;  r:  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r1   c                      yr9  r;  r=  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r1   c                      yr9  r;  r]  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r1   c                      yr9  r;  )r=  rN  rO  rP  rQ  rR  pad_moderS  rT  rV  align_to_windows              r.   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r1   c                      yr9  r;  r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r1   c                      yr9  r;  rM  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r1   c                      yr9  r;  )r  r  cs      r.   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r1   c                      yr9  r;  )r$   s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  
compute_uvr>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  rq  Ms       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  full_matricesr>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  dim0rX  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  axis0axis1s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r{  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rz  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r{  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rN  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rO  r>  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r*  rm  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  inds     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  r>  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rl  rm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r\  rm  r  r>  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rI  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rf  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rC  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rC  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  r  unitriangulars        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rc  r  r  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rM  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c
                      yr9  r;  r  s
             r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rM  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r<  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  rm  sizesr  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  sortedreturn_inversereturn_countsrm  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  r  r  rm  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r*  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r:  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r:  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r=  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )rE  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rl  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  rk  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )	conditionrE  rD  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r=  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  levels     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )primaltangentr  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  ri  r  r^  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rQ  rX  rY  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )implicitc                     yr9  r;  )r  ri  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rm  r  rU  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1   c                      yr9  r;  )r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  ri  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rm  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rm  r  rl  r&  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  
split_sizerm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  split_sizesrm  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r1   c                      yr9  r;  )r  rg  rX  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  r^   s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  	dimensionri  r&  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  rO  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>#  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>&  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r1   c                      yr9  r;  )r  arrays     r.   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r1   c                      yr9  r;  )r  idxs     r.   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r1   c                      yr9  r;  )r  memos     r.   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r1   c                      yr9  r;  )r  format_specs     r.   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r1   c                      yr9  r;  )r  protos     r.   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r1   )tensor_contentsc                     yr9  r;  )r  r/  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r1   c                      yr9  r;  )r  r\  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r1   c                      yr9  r;  )r  ds     r.   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r1   c                      yr9  r;  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r.   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r1   c                      yr9  r;  )r  rB  rC  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>T  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>V  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>b  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>d  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>j  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r1   c                      yr9  r;  )r  r^   non_blockingr%   s       r.   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>s  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r1   c                      yr9  r;  )r  r  rv  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r1   c                      yr9  r;  )r  orderellipsis_idxs      r.   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r1   c                      yr9  r;  )r  callables     r.   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  gradientretain_graphcreate_graphrF  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  re   s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )r  c                     yr9  r;  )r  mediansigmar  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  	coalesceds     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r\  rj  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r\  rQ  rX  rY  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  ambiguity_checks     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                     yr9  r;  )r  r  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  rZ  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                     yr9  r;  )r  rt  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rX   s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                     yr9  r;  )r  r  r  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rX   r|  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rE  rD  r|  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r_  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  rO  assigns      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  rU  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r*  rX   r+  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                     yr9  r;  )r  from_tor  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  streams     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  hooks     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  names     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r'  r^  ri  r  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r\  rm  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r\  rm  r  rl  r&  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  accumulate_matchess      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  size1size2	dense_dims       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r^  r_  rU  rT  r>  s         r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  repss     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r^   rj  copyre   s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   )masked_gradc                     yr9  r;  r  r^   r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r  s      r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s     r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  max_version	dl_devicer  s        r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r  r   drivers       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  )r  r]   rj  r%   s       r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   c                      yr9  r;  r  s    r.   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r1   is______i__rbitwise_N)r  r  N)r  N)h㈵>:0yE>F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)       @#use_mm_for_euclid_dist_if_necessary)r  F)FN)NNN)r  NN)   F)Nr  r   r  r  )Nr  r   r   r  r  )r   NNr  )r  r  )r:  N)r   r  Fr9  )r  r:  NNN)r   r   r  )r   r:  )r  )LN)NNr  FF)Nr  Fr  FNN)NNNF)FF)Nr:  N)Nr  r:  N)r   r:  )TT)NF)NNr  )NNr  T)      ?)NFr   N)r  NNr  )d   r   r   N)r  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   r  F)Nr   r  FF)NFNN)r:  FN)        NNN)NNr  )Nr  )r  NFN)r  FF)Nr   FTN)NNF皙?r  )NNNr  )NNNr  r  )r  TF)	NNr  Fr  FNFN)r  r   r  )NNFN)Fư>r  )none)r:  )bilinearr   N)r  Fg|=r:  )g      r  F)NNNNTr  r  )NNnearestNNF)NNr  N)g{Gz?F)g-C6?g      ?r  )N   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  )NNr  Nr  )r  r  g-q=N)r   r   )r  r   F)TFNr  Nr  )Nr   )g      ?gUUUUUU?FF)Nr  )NNr  r  )r  r  N)r     )r  r  r   FNNr  )r  r  N)r   fan_in
leaky_reluN)froNFNN)NNFNN)r  NFNN)r(  r  FNN)r  r   )TF)NTr  )V瞯<)r)  F)reducedN)NFlinearN)r;  r  r  F)r;  )r   r   )r  r  F)r;  )r   r   r   )r  r  r  F)r  r  )r  NNNr   F)r   NNr  )r:  F)	NNNTreflectFTNN)TTN)   r  N)r  N)TFF)TFFN)r  r  )Nr   NN)NNNN(  r3   r4   r  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshaddaddbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullir"   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffrN  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igammacr  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_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'  r+  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_normr  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar  	ones_liker  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2deg	rand_likerandint_like
randn_likeravelr  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_itesym_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r$  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr&  HmTmH_backward_hooks_post_accumulate_grad_hooksr3  _cdatar4  r5  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  r]   r^   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizerl   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_ro   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrr  rm  	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_rf   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_shortri  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager^  storage_typesum_to_sizer  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asr  viewview_aszero_
__dlpack____dlpack_device__r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrr   items__name__
startswithlenextendr|  update)r4   retprivateuse1_backend_nameret2ignoredr\  r  r  subnamer  r   s              r.   r   r     sX   6 \\F|%		-|%2|% 	!!#@|% 	!!#A	|%
 	

.|% 	'|% 	0|% 	/|% 	1|% 			4|% 	Q|% 	L|% 	L|% 	L|% 	J|%  	

K!|%" 	##%J#|%$ 			-%|%& 	X'|%( 	F)|%* 	

.+|%, 	

.-|%. 	J/|%0 	/1|%2 			F3|%4 	&5|%6 	&7|%8 	19|%: 	

.;|%< 	2=|%> 	0?|%@ 	/A|%B 	1C|%D 	

.E|%F 	0G|%H 	6I|%J 	8K|%L 	/M|%N 	1O|%P 	-Q|%R 	-S|%T 	-U|%V 	xW|%X 	RY|%Z 	{[|%\ 	''){]|%^ 	((*u_|%` 	 Qa|%b 	%%'vc|%d 	11  4Ce|%f 	 5g|%h 	%%'\i|%j 	Ck|%l 	?m|%n 	..tq|%t 	Cu|%v 	>w|%x 	<y|%z 	5{|%| 	;}|%~ 	<|%@ 	  "CA|%B 	!!#DC|%D 	-E|%F 			CG|%H 	!4I|%J 	1K|%L 	]M|%N 	1O|%P 			6Q|%R 	9S|%T 	>U|%V 	aW|%X 	

.Y|%Z 	

>[|%\ 	$:]|%^ 	7_|%` 	?a|%b 	9c|%d 	  "Pe|%f 	 Gg|%h 	Ni|%j 	&&(Yk|%l 	4m|%n 	Co|%p 	

Bq|%r 	8s|%t 	8u|%v 	8w|%x 			Oy|%z 	%{|%| 	I}|%~ 	,|%@ 	9A|%B 	(C|%D 	5E|%F 	

.G|%H 	7I|%J 	6K|%L 	5M|%N 	=O|%P 	dQ|%R 	dS|%T 	dU|%V 	wW|%X 	=Y|%Z 	  !A[|%\ 	  !A]|%^ 	  !A_|%` 	(a|%b 			-c|%d 	##  &Ce|%f 	

.g|%h 	!Ci|%j 	-k|%l 	@m|%n 	Eo|%p 	xs|%v 	5w|%x 	5y|%z 	B{|%| 	A}|%~ 	""$@|%@ 	;A|%B 	1C|%D 	*E|%F 			#G|%H 	*I|%J 	&K|%L 	

:M|%N 	@O|%P 	2Q|%R 	

VS|%T 	BU|%V 	KW|%X 	 OY|%Z 	  "Y[|%\ 	1]|%^ 	

0_|%` 			Ha|%b 	Kc|%d 			4e|%f 	@g|%h 	

:i|%j 	

)k|%l 	;m|%n 	2o|%p 	4q|%r 	8s|%t 	?u|%v 	Cw|%x 	4y|%z 	|}|%@ 	 jC|%F 	eG|%H 	3I|%J 	,K|%L 			-M|%N 	

.O|%P 	0Q|%R 			-S|%T 	

.U|%V 	/W|%X 	..0oY|%Z 	--/h[|%\ 	-- C_|%b 	'')Vc|%d 	779fe|%f 	'')}g|%h 	77`k|%n 	++-=o|%p 	**,<q|%r 	**,Bs|%t 	##%?u|%v 	9w|%x 			Cy|%z 			C{|%| 			D}|%~ 			C|%@ 			DA|%B 			JC|%D 			KE|%F 			DG|%H 			EI|%J 			EK|%L 			FM|%N 			FO|%P 			GQ|%R 			IS|%T 			JU|%V 			JW|%X 			KY|%Z 			6[|%\ 			7]|%^ 			B_|%` 			-a|%b 	@c|%d 	

*e|%f 	&g|%h 	&i|%j 	Qk|%l 	/m|%n 	3o|%p 	?q|%r 	

5s|%t 	

.u|%v 	/w|%x 	t4PUP]P]fjz  Dy|%z 	&&(H{|%| 	\}|%~ 	O|%@ 			4A|%B 	3C|%D 	*E|%F 	>G|%H 	/I|%J 	,K|%L 	6M|%N 	5O|%P 			3Q|%R 	NS|%T 	cU|%V 	fW|%X 	fY|%Z 	m[|%\ 			s]|%^ 	N_|%` 	3a|%b 	8c|%d 	5e|%f 	Vg|%h 	;i|%j 	""$zk|%l 	Gm|%n 	mo|%p 	Yq|%r 	((*?s|%t 	4u|%v 	;w|%x 	2y|%z 	6{|%| 	7}|%~ 	8|%@ 	

.A|%B 	=C|%D 	>E|%F 	LG|%H 	BI|%J 	=K|%L 	\M|%N 	)O|%P 	

GQ|%R 	&S|%T 	'U|%V 	2W|%X 	2Y|%Z 	t]|%` 	(a|%b 	1c|%d 	4e|%f 	Kg|%h 	*i|%j 	'k|%l 	&m|%n 	.o|%p 	,q|%r 	!1s|%t 	*u|%v 	3w|%x 	)y|%z 	W{|%| 	%}|%~ 	 dA	|%D	 	rE	|%F	 	

+G	|%H	 	NI	|%J	 	""$cK	|%L	 	!LM	|%N	 	 SO	|%P	 	sQ	|%R	 			4S	|%T	 	6U	|%V	 	3W	|%X	 	;Y	|%Z	 	

;[	|%\	 	0]	|%^	 	  K_	|%`	 			-a	|%b	 	<c	|%d	 	/e	|%f	 	/g	|%h	 	

.i	|%j	 	:k	|%l	 	;m	|%n	 	&o	|%p	 	.q	|%r	 	<s	|%t	 	5u	|%v	 	;w	|%x	 	<y	|%z	 	/{	|%|	 	I}	|%~	 	

s	|%@
 	OA
|%B
 	3C
|%D
 	

5E
|%F
 	EG
|%H
 	BI
|%J
 	!!  $AK
|%L
 	8M
|%N
 	<O
|%P
 	=Q
|%R
 	7S
|%T
 	?U
|%V
 	 FW
|%X
 	!!#]Y
|%Z
 	[[
|%\
 	>]
|%^
 	/_
|%`
 	!!#@a
|%b
 	  "Mc
|%d
 	 <e
|%f
 	*g
|%h
 	!1i
|%j
 			-k
|%l
 	8m
|%n
 	

5o
|%p
 	lq
|%r
 	ls
|%t
 	lu
|%v
 	%%ty
|%|
 	

.}
|%~
 	V
|%@ 	0A|%B 	3C|%D 	5E|%F 			-G|%H 	8I|%J 	

5K|%L 	tO|%R 	  "}S|%T 	))+vU|%V 	%%'hW|%X 	**w[|%^ 	**ga|%d 	 dg|%j 	Bk|%l 	

Em|%n 	<o|%p 	=q|%r 	As|%t 			4u|%v 	9w|%x 	Uy|%z 	1{|%| 	+}|%~ 	:|%@ 	WA|%B 	!sC|%D 	&&(_E|%F 	8G|%H 	!fI|%J 	YK|%L 	!VM|%N 	UO|%P 	$$&>Q|%R 	3S|%T 	:U|%V 			-W|%X 	2Y|%Z 	:[|%\ 	//1N]|%^ 	//1N_|%` 	//1da|%b 	<<>qc|%d 	//1de|%f 	<<>qg|%h 	//1di|%j 	<<>qk|%l 	'')Sm|%n 	))+ao|%p 	&& Bs|%v 	&& By|%| 	&&x|%B 	$$&RC|%D 	00cG|%J 	<<tM|%P 	  "LQ|%R 	11iU|%X 	)) L[|%^ 	$$xa|%d 	##%Ze|%f 	%%'\g|%h 	%%'\i|%j 	%%'\k|%l 	!Km|%n 	%%|q|%t 	)) Jw|%z 	113i{|%| 	  "m}|%~ 	11zA|%D 	>>zG|%J 	11zM|%P 	>>zS|%V 	--/uW|%X 	  "FY|%Z 	!9[|%\ 	'')z]|%^ 	&&(g_|%` 	**,ca|%b 	&&(Cc|%d 	$$&`e|%f 	00bi|%l 	)) Io|%r 	'' Mu|%x 	""  %Ay|%z 	##%|{|%| 	&&(m}|%~ 	&&(\|%@ 	""$GA|%B 	//1gC|%D 	'')^E|%F 	&&(8G|%H 	%%'mI|%J 	%%'mK|%L 	%%'mM|%N 	//iQ|%T 	&&tW|%Z 	33t]|%` 	&&tc|%f 	33ti|%l 	&&to|%r 	33tu|%x 	((*zy|%z 	((*z{|%| 	((*z}|%~ 	$$&}|%@ 	88 _C|%F 	--tI|%L 	22VO|%R 	77cU|%X 	$$v[|%^ 	%%'X_|%` 	##%Fa|%b 	!Pc|%d 	--/ae|%f 	,,}i|%l 	!!#;m|%n 	  "Ao|%p 	!!#Bq|%r 	$$&_s|%t 	!!#yu|%v 	  "Aw|%x 	  "Ay|%z 	  "A{|%| 	88:u}|%~ 	**  -A|%@ 	&&(jA|%B 	,,.xC|%D 	##%ZE|%F 	##%ZG|%H 	$$&LI|%J 	&&(CK|%L 	$$&6M|%N 	&&(8O|%P 	%%'XQ|%R 	// LU|%X 	==DHQT[`lrv[|%^ 	""$b_|%` 	 Oa|%b 	Sc|%d 	!7e|%f 	&&(xg|%h 	7i|%j 	Fk|%l 	(m|%n 	

\o|%p 	dq|%r 	  "hs|%t 	   +/#3u|%| 	9}|%~ 	d|%@ 	%A|%B 	*C|%D 	QE|%F 	!SG|%H 	+I|%J 	IK|%L 	*M|%N 	5O|%P 	IQ|%R 	=S|%T 	AU|%V 	7W|%X 	 YY|%Z 	6[|%\ 	2]|%^ 	-_|%` 	da|%b 			7c|%d 	

0e|%f 			Dg|%h 	  "2i|%j 	""$4k|%l 	'')9m|%n 	'o|%p 	,q|%r 	7s|%t 	Cu|%v 	fw|%x 	iy|%z 	""$V{|%| 	!!#M}|%~ 	))+P|%@ 	""$sA|%B 	   aE|%H 	!! aK|%N 	""J#Q|%X 	""L #[|%d 	""O 	#g|%r 	%% au|%x 	%% a{|%~ 	1|%@ 	dA|%B 	d5==Y]mrvC|%D 	eE|%F 	%G|%H 	

.I|%J 	

5K|%L 	FM|%N 	,O|%P 	/Q|%R 	4S|%T 	

3U|%V 	:W|%X 	AY|%Z 	!;[|%\ 	.]|%^ 	Q_|%` 	xa|%b 	Sc|%d 	xe|%f 	Sg|%h 	

7i|%j 	7k|%l 	/m|%n 	5o|%p 	Pq|%r 	bs|%t 	/u|%v 	

4w|%x 	My|%z 	8{|%| 	<}|%~ 	Z|%@ 	eA|%B 	|C|%D 	2E|%F 	?G|%H 	WI|%J 	WK|%L 	

3M|%N 	1O|%P 	

.Q|%R 	1S|%T 			-U|%V 			-W|%X 	

.Y|%Z 	

.[|%\ 	']|%^ 	._|%` 			9a|%b 	

:c|%d 	8e|%f 	@g|%h 	Wi|%j 	

YeQUYk|%l 	Em|%n 	 Po|%p 	

.q|%r 	0s|%t 	;u|%v 	Ow|%x 	8y|%z 			-{|%| 	2}|%~ 	

 @A|%D 			4E|%F 	9G|%H 			-I|%J 	)K|%L 	'M|%N 	O|%P 	Q|%R 	'S|%T 	)U|%V 	W|%X 	)Y|%Z 	([|%\ 	)]|%^ 	(_|%` 	)a|%b 	(c|%d 	)e|%f 	)g|%h 	)i|%j 	)k|%l 	0m|%n 			Io|%p 	Aq|%r 	Hs|%t 	8u|%v 	4w|%x 	6y|%z 	/{|%| 	!1}|%~ 	!1|%@ 	!1A|%B 	!1C|%D 	,,.KE|%F 	,,.KG|%H 	,,.KI|%J 	,,.KK|%L 	/M|%N 	,O|%P 	+Q|%R 	,S|%T 	-U|%V 	.W|%X 	,Y|%Z 	-[|%\ 	-]|%^ 	 A_|%` 	!Ba|%b 	/c|%d 	**,Ie|%f 	++-Jg|%h 	*i|%j 	+k|%l 	*m|%n 	+o|%p 	++-Jq|%r 	++-Js|%t 	-u|%v 	 0w|%x 	!!#Dy|%z 	-{|%| 	!O}|%~ 	((*:|%@ 	((*:A|%B 	((*:C|%D 	((*:E|%F 	""$7G|%H 	,I|%J 	-K|%L 	!>M|%N 	+O|%P 	-Q|%R 	//1AS|%T 	//1AU|%V 	446SW|%X 	446SY|%Z 	446S[|%\ 	446S]|%^ 	,_|%` 	@a|%b 	))+;c|%d 	@e|%f 	>g|%h 	<i|%j 	!k|%l 	

+m|%n 	Ko|%p 			-q|%r 	

.s|%t 	 3u|%v 	  "<w|%x 	:y|%z 	H{|%| 	J}|%~ 	

*|%@ 	

KA|%B 	%C|%D 	5E|%F 	1G|%H 	5I|%J 	 eK|%L 	%%'aM|%N 	

:O|%P 	!! LS|%V 	

:W|%X 	2Y|%Z 	/[|%\ 	-]|%^ 	<_|%` 	ha|%b 	  "gc|%d 	6e|%f 	;g|%h 	Li|%j 	%%'Wk|%l 	8m|%n 	1o|%p 			-q|%r 	2s|%t 	;u|%v 	2w|%x 	9y|%z 	%%'`{|%| 	11o|%B 	eC|%D 	5E|%F 	@G|%H 	I|%J 	""OK|%L 	/M|%N 	oO|%P 	QQ|%R 	'')>S|%T 	FU|%V 	%CW|%X 	>Y|%Z 	1[|%\ 	!!#@]|%^ 	6_|%` 	?a|%b 	Nc|%d 	<e|%f 	##%Hg|%h 	0i|%j 	ok|%l 	9m|%n 	2o|%p 	_q|%r 	Os|%t 	Ou|%v 	?w|%x 	y|%z 	{|%| 	}|%~ 	|%@ 	1A|%B 	/C|%D 	AE|%F 	/G|%H 	3I|%J 	4K|%L 	4M|%N 	2O|%P 	3Q|%R 	3S|%T 	1U|%V 	2W|%X 	2Y|%Z 	1[|%\ 	2]|%^ 	2_|%` 	.a|%b 	-c|%d 	.e|%f 	/g|%h 	Oi|%j 	0k|%l 	m|%n 	3o|%p 	q|%r 	?s|%t 	.u|%v 	/w|%x 	/y|%z 	5{|%| 	0}|%~ 	2|%@ 	A|%B 	C|%D 	/E|%F 	G|%H 	7I|%J 	4K|%L 	_M|%N 	AO|%P 	1Q|%R 	/S|%T 	/U|%V 	/W|%X 			?Y|%Z 			?[|%\ 	&&]|%^ 	**22O_|%` 	oa|%b 	c|%d 	_e|%f 	og|%h 	i|%j 	k|%l 	m|%n 	--/po|%p 	**,Vq|%r 	22Os|%t 	_u|%v 	w|%x 	oy|%z 	{|%| 	}|%~ 	|%@ 	A|%B 	C|%D 	E|%F 	##_G|%H 	I|%J 	K|%L 	M|%N 	  /O|%P 	Q|%R 	  /S|%T 	##_U|%V 	  /W|%X 	$$oY|%Z 	  /[|%\ 	]|%^ 	_|%` 	_a|%b 	oc|%d 	e|%f 	_g|%h 	  /i|%j 	$$ok|%l 	om|%n 	o|%p 	_q|%r 	_s|%t 	''//u|%v 	Nw|%x 	oy|%z 	o{|%| 	}|%~ 	|%@ 	_A|%B 	_C|%D 	OE|%F 	_G|%H 	OI|%J 	=K|%L 	M|%N 	O|%P 	/Q|%R 	=S|%T 	0U|%V 	8W|%X 	9Y|%Z 	k[|%\ 	E4I4IM]|%^ 	0E0EI_|%` 	0E0EIa|%b 	0E0EIc|%d 	MTMe|%f 	g|%h 	6i|%j 	e6M6MQk|%l 	>m|%n 	

u/D/DHo|%p 	0E0EIq|%r 	0E0EIs|%t 	

u/D/DHu|%v 	

u/D/DHw|%x 	y|%z 	/{|%| 	!O}|%~ 	

O|%@ 	@A|%B 	%2G2GKC|%D 	53H3HLE|%F 	_G|%H 	,I|%J 	0K|%L 	HHM|%N 	,O|%P 	5Q|%R 	1F1FJS|%T 	%2G2GKU|%V 	@W|%X 	?Y|%Z 	0E0EI[|%\ 	1F1FJ]|%^ 	/_|%` 	a|%b 	

u/D/DHc|%d 	_e|%f 	og|%h 	_i|%j 	/k|%l 	1m|%n 	/o|%p 	_q|%r 	MTMs|%t 	0u|%v 	0E0EIw|%x 	6y|%z 	5{|%| 			8}|%~ 	@|%@  	EA |%B  	?C |%D  	E |%F  	""OG |%H  	--I |%J  	%%K |%L  	M |%N  	oO |%P  	,Q |%R  	?S |%T  	GU |%V  	W |%X  	LDLY |%Z  	5[ |%\  	3] |%^  	3_ |%`  	113Ha |%b  	,c |%d  	-e |%f  	Bg |%h  	1i |%j  	-k |%l  	-m |%n  	0o |%p  	  "8q |%r  	Os |%t  	[u |%v  	?w |%x  	oy |%z  	1F1FJ{ |%|  	_} |%~  	W |%@! 	?A!|%B! 	1C!|%D! 	&&(WE!|%F! 	GG!|%H! 	'')QI!|%J! 	OK!|%L! 	M!|%N! 	O!|%P! 	Q!|%R! 	_S!|%T! 	1U!|%V! 	+W!|%X! 			EUZUjUjnY!|%Z! 	II[!|%\! 	G]!|%^! 	/_!|%`! 	a!|%b! 	/c!|%d! 	.e!|%f! 	=7+.od  /Fw!|%C~! 	((BB  v/0F 	GF456 JYGFc":!;<=EEFD#%G		 1 JJJJ1::$AJJ%AJJ%
 ::  , jjZ!23GLL$&$(>RV@VW  	D64.D~$c/d'6IT
	%. JJtJr1   
dispatcherc                       fd}|S )a  Wraps a given function with ``__torch_function__`` -related functionality.

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

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

    Examples
    --------
    >>> def dispatcher(a):  # Must have the same signature as func
    ...     return (a,)
    >>> @torch.overrides.wrap_torch_function(dispatcher)
    >>> def func(a):  # This will make func dispatchable by __torch_function__
    ...     return a + 0
    c                 J     t        j                          fd       S )Nc                  Z     | i |}t        |      rt        |g| i |S  | i |S r  )r   r   )r$   r%   relevant_argsr~
  r   wrappeds      r.   r
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped2  sA    &77M!-0,WmUdUfUU(((r1   )	functoolsr   )r   r
  r~
  s   `@r.   r  z"wrap_torch_function.<locals>.inner1  s%    			) 
	) r1   r;  )r~
  r  s   ` r.   r   r     s    0	 Lr1   r
  get_type_fnc                    |t         }t        j                  j                         sg S t	               }g }| D ]  } ||      }||vst        |d      s|j                  t        j                  j                  k7  sE|rX|j                  |       t        |      }t        |      D ]  \  }}t        | ||            s|} n |j                  ||       |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
    r   )r
  r3   _C_is_torch_function_enabledsetrq
  r   _disabled_torch_function_implr5  rv
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer  iold_args	            r.   _get_overloaded_argsr
  ?  s    J  88..0	"%%!#O (s# ,,"67++uxx/U/UU   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%9(: r1   
public_apic           	         t        |      }t        t        t        |            }t	               r1t               5 }|j                  | |||      }ddd       t        ur|S |D ]u  }|j                  }	t        |	d      rD|	j                  |u r6|	t        j                  j                  urt        j                  dt                |	| |||      }|t        uss|c S  | j                    d| j"                   }
d|
 d|D cg c]  }t        |       c} }t	               r|dt%                z  }t'        |      # 1 sw Y   xY wc c}w )a=  Implement a function with checks for ``__torch_function__`` overrides.

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

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

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

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

    Example
    -------
    >>> def func(a):
    ...     if has_torch_function_unary(a):
    ...         return handle_torch_function(func, (a,), a)
    ...     return a + 0
    N__self__zDefining your `__torch_function__ as a plain method is deprecated and will be an error in future, please define it as a classmethod..zno implementation found for 'z.' on types that implement __torch_function__: z nor in mode )r
  tuplemapr
  r   _pop_mode_temporarilyr   NotImplementedrq
  r
  r3   r
  r
  r*   warnDeprecationWarning
__module__rt
  _get_current_function_mode	TypeError)r
  r
  r$   r%   r
  typesr  resultoverloaded_argtorch_func_method	func_namer
  r  s                r.   r   r     sf   T +=9O#dO,-E '( #$ 	N,,ZfMF	N'M *  +==%z2!**n<!)O)OOMMQ" #:udFC'M), (():+>+>*?@I
'	{ 35DEcS	EF	H  '(9;<==
C.G	N 	N>  Fs   D9=E
9E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                  d   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 ],  \  }}}|D ]  }d	}|t        j                  urV|j                  d
      r*|j                  d      rd}nW|j                  d      rd}nC|d   j                         sd}n-|dk(  r(nt!        ||      }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||j2                  <   | d| d||j4                  <   |rC|j2                  t7               v r;d}	|j2                  t9               vs!J |	j;                  ||j<                               | |   j?                  |j2                         t/        |      s| d| ||<   |r|t7               v r1d}	|t9               vs!J |	j;                  ||j<                               | |   j?                  |       " / | |fS )Nr3   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listr3   __all__r   r   dirr   r4   ry  r   rM	  ru
  endswithislowerrr
  object
isinstancer
  
ModuleType
__future___Featurer|  rq
  r&  __set__r   r   formatrt
  rU  )
overridable_funcsr  tested_namespacesnamespace_str	namespacens_funcsr
  r'   r   r  s
             r.   _get_overridable_functionsr
    sE    $//5E	%'	U--u/?/?/G/GH	 3 3S9L9L5MN	%((--UXX]]);<	s5<<'89	s5<<'89	eiiUYY0	%--U]]);<	 /@ C6*y(! B	6IF,''-))#.!F'',!F"1--/!F,.y)469d3t;-9i0DELL(WVY-MQU-U$ 0 01$
 3 34D>gdI&>)6q8&Ldll#)6q8&Ldll#<<#8#::=   <</D/FF 

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

    Returns
    -------
    Dict[Any, List[Callable]]
        A dictionary that maps namespaces that contain overridable functions
        to functions in that namespace that can be overridden.
    r   )r
  r;  r1   r.   r   r   m  s     &'**r1   c                     t        | t        j                  j                  t        j                  j                  f      rt        |       S t               d   j                  |       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.
    r  )r
  r3   _ops
OpOverloadOpOverloadPacketstrr
  get)fs    r.   r   r   z  sI      !ejj++UZZ-H-HIJ1v%'*..q11r1   c                  R    t               } t        | t        j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r
  r3   r4   )r
  methodss     r.   _get_tensor_methodsr
    s&     23#ELL12GNr1   c                 <    | t               v xs | j                  dk(  S )aw  
    Returns True if the function passed in is a handler for a
    method or property belonging to ``torch.Tensor``, as passed
    into ``__torch_function__``.

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

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

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

    Examples
    --------
    >>> is_tensor_method_or_property(torch.Tensor.add)
    True
    >>> is_tensor_method_or_property(torch.add)
    False
    r&  )r
  rt
  )r   s    r.   r   r     s!    . &((FDMMY,FFr1   c                 T    t        |       t        j                  u xs 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
  r3   r4   rq
  )inps    r.   r   r     s%    D 9$J5I(JJr1   c                   H    e Zd ZU dZd ed<   d	dZd
dZd Zd Ze	d        Z
y)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  Nc                      y r  r;  r  s    r.   r   zTorchFunctionMode.__init__  s    r1   c                     t         r  )NotImplementedErrorr  r   r
  r$   r%   s        r.   r   z$TorchFunctionMode.__torch_function__  s    !!r1   c                     t        |        | S r  )
_push_moder  s    r.   	__enter__zTorchFunctionMode.__enter__  s    4r1   c                     t                y r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r.   __exit__zTorchFunctionMode.__exit__  s    r1   c                 @    t        j                  d        | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`)r*   r
  )clsr$   r%   instances       r.   pushzTorchFunctionMode.push  s'    ^	
 ''r1   )r"   Nr;  N)rt
  r
  __qualname____doc____annotations__r   r   r
  r
  classmethodr
  r;  r1   r.   r
  r
    s7    < "  r1   r
  c                  @    t               } | dkD  rt        | dz
        S d S )Nr   r  )r   r   )	stack_lens    r.   r
  r
    s%    )+I4=M!)a-0KtKr1   c                  d    t               } t        |       D cg c]  }t        |       c}S c c}w r  )r   r   r   )r
  r
  s     r.    _get_current_function_mode_stackr
    s)    )+I/4Y/?@!"1%@@@s   -c                     t        |        y r  )r   )r  s    r.   r
  r
    s
    !$'r1   c                      t               } | S r  )r   olds    r.   r
  r
    s    
#
%CJr1   c               #   `   K   t               } 	 |  t        |        y # t        |        w xY wwr  )r
  r
  r
  s    r.   r
  r
  !  s%     
+C	3
3s   . .+.c                       e Zd ZddZy)BaseTorchFunctionModeNc                     |i } ||i |S r  r;  r
  s        r.   r   z(BaseTorchFunctionMode.__torch_function__+  s    >FT$V$$r1   r
  )rt
  r
  r
  r   r;  r1   r.   r
  r
  *  s    %r1   r
  c               #   Z  K   t         j                  j                         } 	 t         j                  j                  t         j                  j                  j
                         d  t         j                  j                  |        y # t         j                  j                  |        w xY wwr  )r3   r
  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r.   _enable_torch_functionr
  1  sf     224I6**588+G+G+O+OP**95**95s   B+AB ' B+!B((B+c               #      K   t         j                  j                         5  	 d  	 d d d        y # w xY w# 1 sw Y   y xY wwr  )r3   r
  _RestorePythonTLSSnapshotr;  r1   r.   r   r   ;  sC      
	+	+	- 		  	 s#   A61	A36?A)z.*is deprecated, please use.*r3   r  )Ar
  r
  r
  
contextlibr
  r
  r*   collections.abcr   r   typingr   r   r   r   typing_extensionsr	   r3   torch._Cr
   r   r   r   r   r   r   r   r   r
  r   r   r
  r0   cacher
  r   r6  dictr   r   r
  r
  r
  r   r   r   r   r
  r
  r   r   r
  r
  r   r   r
  r
  r
  r
  r
  contextmanagerr
  r
  r
  r   r;  r1   r.   <module>r     s*  ,       $  3 3 ' 
 
 
 t_T]
 1 
2r6
     b"f	 F Ts8} T  Tn c(m  2 |tHh$67 |  |~##H #P 48KC=K(C5$;/0K 
#YK\UUC=U
 	Up ! . '	  * 	  S$Ed8n	tHcM22% S$ S$l 	+4T(^(;#< 	+ 	+ 2 2( S]   Gx GD G G2"KJ5 5pL
A
(
  %- % 6 6  r1   