
    sg1             	          d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm	Z	m
Z
mZmZmZmZmZmZ ddlZddlmZmZmZmZmZmZmZmZmZ g dZ	 	 d)de
ded	ed
e
fdZ ej>                  d      ed
ee
   fd              Z  ej>                  d      d
ee
   fd       Z! ej>                  d      ed
ee
e
f   fd              Z"de
fdZ#	 d*dee	   de
e	gef   d
ee	   fdZ$de
dee	   d
e	fdZ% eed      Z& eed      Z' eed      Z( ej>                  d      d
eee	ee
   f   ee
ef   f   fd       Z)ed
ee	ee
   f   fd       Z*ed        Z+ ej>                  d      d
ee
   fd       Z,ede
d
e-fd       Z.d Z/ G d d       Z0d! Z1d" Z2d# Z3d$ Z4ejj                  d%        Z6 G d& d'e0      Z7ejj                  d(        Z8y)+aE  
Python implementation of ``__torch_function__``

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

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

If changing this file in a way that can affect ``__torch_function__`` overhead,
please report the benchmarks in ``benchmarks/overrides_benchmark``. See the
instructions in the ``README.md`` in that directory.
    Nwraps)AnyCallableDictIterableListSetTupleType)	_add_docstr_get_function_stack_at_has_torch_function_has_torch_function_unary_has_torch_function_variadic_is_torch_function_mode_enabled_len_torch_function_stack_pop_torch_function_stack_push_on_torch_function_stack)
get_ignored_functionsget_overridable_functionsget_testing_overrideshandle_torch_functionhas_torch_functionresolve_nameis_tensor_likeis_tensor_method_or_propertywrap_torch_functionenable_reentrant_dispatchfuncregexmodulereturnc                 6     t                fd       }|S )a  
    Decorator that temporarily disables ``UserWarning``s for the given ``module`` if the warning message matches the
    given ``regex`` pattern.

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

    Returns
    -------
    function
        The wrapped function.
    c                      t        j                         5  t        j                  dt                | i |cd d d        S # 1 sw Y   y xY w)Nignore)categorymessager"   )warningscatch_warningsfilterwarningsUserWarning)argskwargsr    r"   r!   s     B/var/www/html/venv/lib/python3.12/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperU   sJ    $$& 	)##;f ((		) 	) 	)s   %AAr   )r    r!   r"   r0   s   ``` r/   _disable_user_warningsr1   =   s"    0 4[) ) N    c                  #   t         j                  } h t         j                  t         j                  t         j                  t         j
                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                   t         j"                  t         j$                  t         j&                  t         j(                  t         j*                  t         j,                  t         j.                  t         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                  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                  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                  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
get_devicetensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutalign_tensorsarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptableemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrandrandnrandintrandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   sigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvitals_enabled	set_vitalread_vitalsvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_countr5   s    r/   r   r   `   s8   ( \\F@@@ 	@ 	%%	@
 	  @ 	  @ 	@ 	@ 	@ 	@ 	

@ 	

@ 	

@ 	@ 	

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

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

m@n 	

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

I@J 	K@L 	

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

@@ 	

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

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

o@p 	q@r 	s@t 	  u@v 	w@x 	y@z 	{@| 	}@~ 	@@ 	A@B 	C@D 	E@F 	G@H 	I@J 	K@L 	M@N 	O@P 	Q@R 	S@T 	U@V 	W@X 	Y@Z 	[@\ 	""]@^ 	_@` 	a@b 	  c@d 	$$e@f 	%%g@h 	''i@j 	**22k@l 	m@n 	o@p 	q@r 	s@t 	u@v 	w@x 	y@z 	  {@| 	}@~ 	@ @r2   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).
    )r4   r5   _baser&  grad_gradr1  s    r/   get_default_nowrap_functionsr6  x  s>    $ \\F r2   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                  d 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                  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                  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                  d dt         j                  ddt         j                  ddt         j                  j                  d t         j                  ddt         j                  d t         j                  ddi 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t         j                  dՐdt         j                  d i 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 t         j                  ddt         j                  ddi 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         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+ t         jJ                  dՐd,t         jL                  dՐd-t         jN                  dՐd.t         jP                  dՐd/t         jR                  dՐd0t         jT                  dՐd1t         jV                  dd2t         jX                  d3 t         jZ                  dܐd4t         j\                  dՐd5t         j^                  dՐd6t         j`                  dd7t         jb                  dՐd8i t         jd                  dd9t         jf                  d: t         jh                  d; t         jj                  dՐd<t         jl                  dՐd=t         j                  j`                  dd>t         j                  jn                  dd?t         j                  jp                  dd@t         j                  jb                  ddAt         j                  jl                  dՐdBt         jr                  dC t         j                  jr                  dՐdDt         j                  jt                  ddEt         j                  jv                  dՐdFt         jx                  dG t         j                  jx                  dH t         jz                  dՐdIi t         j|                  dՐ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                  dT t         j                  dՐdUt         j                  dՐdVt         j                  dՐdWt         j                  dX t         j                  dY t         j                  dZ i t         j                  d[ t         j                  d\ t         j                  d] t         j                  d^ t         j                  dՐd_t         j                  dd`t         j                  da t         j                  db t         j                  ddct         j                  dՐddt         j                  dՐdet         j                  ddft         j                  dՐdgt         j                  dh t         j                  di t         j                  ddjt         j                  dk i t         j                  dl t         j                  dm t         j                  ddnt         j                  do t         j                  ddpt         j                  dq t         j                  dՐdrt         j                  dՐdst         j                  dՐdtt         j                  dՐdut         j                  dՐdvt         j                  j                  j                  dw t         j                  j                  j                  dx t         j                  j                  j
                  dِdyt         j                  j                  j                  dِdz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                  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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                  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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 t         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                   ddt         j                  j                  jd                  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                  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<                  d*di t         j                  j                  j>                  d+dt         j                  j                  j@                  d,dt         j                  j                  jB                  dÄ t         j                  j                  jD                  dِdĄt         j                  j                  jF                  dِdńt         j                  j                  jH                  d-dƄt         j                  j                  jJ                  d.dǄt         j                  j                  jL                  dِdȄt         j                  j                  jN                  dِdɄt         j                  j                  jP                  dِdʄt         j                  j                  jR                  d/d˄t         j                  j                  jT                  d0d̄t         j                  j                  jV                  d1d̈́t         j                  j                  jX                  d!d΄t         j                  j                  jZ                  d$dτt         j                  j                  j\                  d$dЄt         j                  j                  j^                  d2dфi t         j                  j                  j`                  dd҄t         j                  j                  jb                  dӄ t         j                  j                  jd                  dԄ t         j                  j                  jf                  dِdՄt         j                  j                  jh                  d3dքt         j                  j                  jj                  ddNdd9dלd؄t         j                  j                  jl                  ddلt         j                  jn                  jp                  d4dڄt         j                  jn                  jr                  d4dۄt         j                  jn                  jt                  d܄ t         j                  jn                  jv                  d5d݄t         jx                  dِdބt         jz                  dzdߜdt         j|                  d t         j~                  d6dt         j                  j~                  d7dt         j                  j                  d8di t         j                  j                  	 d9dt         j                  d:dt         j                  d6dt         j                  d t         j                  d t         j                  d;dt         j>                  d+dt         j                  d t         j                  d<dt         j                  ddt         j                  d=dt         j                  j                  d>dt         j                  d t         j                  d t         j                  dՐdt         j@                  d t         j                  dՐdi t         j                  dՐdt         jB                  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                  d?dt         j                  d@dt         j                  d@dt         j                  d t         j                  d i t         j                  d t         j                  d t         j                  d	 t         j                  d
 t         j                  	 	 dAdt         j                  	 	 dBdt         j                  	 	 dC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 t         j                  dՐdt         j                  dՐdt         j                  j                  ddi t         j                  d t         j                  d t         j                  dՐdt         jD                  dِdt         j                  dՐdt         j                  dՐdt         j                  dՐd t         j                  d! t         jH                  d-d"t         j                  d# t         j                  dܐd$t         j                  d% t         j                  dܐd&t         j                  dՐd't         j                  dDd)t         j                  dՐd*t         j                  dՐd+i t         j                  d, t         jJ                  d.d-t         j                   dՐd.t         j                  dd/t         j                  d֐d0t         j                  d1 t         j                  d2 t         j
                  dd3t         j                  dߐd4t         j                  dEd5t         j                  d6 t         j                  d7 t         j                  dFd8t         j                  dFd9t         jL                  dِd:t         j                  dՐd;t         j                  dՐd<i t         j                  dՐd=t         j                  dՐd>t         j                   dՐd?t         j"                  dՐd@t         j$                  dՐdAt         j&                  dB t         j                  j&                  dC t         j(                  dD t         j*                  dE t         jZ                  dՐdFt         j                  j,                  ddGt         j                  j.                  ddHt         j0                  dGdddIdJt         j2                  ddKt         j4                  ddLt         j6                  dՐ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                  dHdTt         jF                  dՐdUt         jH                  dՐdVt         jJ                  dՐdWt         jL                  dX t         jN                  dY t         jP                  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                  dՐdht         jn                  dIdit         jp                  dJdjt         j                  jn                  ddkt         j                  jr                  dՐdlt         jt                  dm t         jv                  dn t         jx                  jz                  do t         jx                  j|                  dp i t         jx                  j~                  dq t         jx                  j                  dr t         jx                  j                  ds t         jx                  j                  dՐdtt         jx                  j                  dՐdut         jx                  j                  dՐdvt         jx                  j                  dՐdwt         jx                  j                  dx t         jx                  j                  dy t         jx                  j8                  dz t         jx                  j:                  d{ t         jx                  j                  d| t         jx                  j<                  d} t         jx                  j@                  d~ t         jx                  j                  d t         jx                  jB                  d t         jx                  j                  dՐdi t         jx                  j                  dՐdt         jx                  j                  d t         jx                  j                  dՐdt         jx                  j                  dՐdt         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dՐdt         jx                  j                  dՐdt         jx                  j@                  d t         jx                  j                  d t         jx                  j<                  dՐdt         jx                  jT                  d t         jx                  jV                  ddt         jx                  j                  d t         jx                  j                  d i t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dՐdt         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dՐdt         jx                  j                  dՐdt         jx                  j                  dՐdt         jx                  j                  dՐdt         jx                  j"                  d t         jx                  jZ                  dՐdt         jx                  j                  d i t         jx                  j                  dՐdt         jx                  jJ                  dՐdt         jx                  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                  dKdt         j                  ddt         jf                  dِdt         j                  d t         j                  ddt         j                  d t         j                  d i t         j                  ddt         j                  ddt         j                  dLdt         j                  j                  d;dt         j                  ddt         jh                  d3dt         j                  ddt         j                  d t         j                  dՐdt         j                  ddt         j                  d t         j                  dMdt         j                  dߐdt         j                  d t         j                  ddÄt         j                   ddĄt         j                  ddńi 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҄ t         j                  dӄ t         j                   dԄ t         j"                  dՄ t         j$                  dՐdքi t         j&                  dׄ t         j(                  dd؄t         j*                  ddٜdڄt         j,                  dۄ t         j.                  d܄ t         j0                  d݄ t         j2                  dބ t         j4                  d߄ t         j6                  dFdt         j8                  ddt         j:                  ddt         j<                  d t         j>                  d t         j@                  d t         jB                  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dt         jV                  d t         jX                  d t         jZ                  d | j\                  d | j^                  d | j`                  d | 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dd| j                  d | j                  d | j                  j                  d | j                  j                  d | j                  j                  d i | 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                  j                  d) | j                  j                  d* | j                  j                  d+ | j                  j                  d, | j                  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                  ddJ| j                  dK | j
                  dL | j                  dM | j                  dN | j                  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d\| j*                  t         j,                  fd]| j.                  t         j,                  fd^| j0                  t         j,                  fd_| j2                  t         j,                  fd`i | j4                  d(ddadb| j6                  dc | j8                  dd | j:                  t         j<                  fde| j>                  dِdf| j@                  t         j,                  fdg| jB                  t         j,                  fdh| jD                  t         j,                  fdi| jF                  t         j,                  fdj| jH                  t         j,                  fdk| jJ                  dl | jL                  dm | j                  ddn| jN                  do | jP                  dp | jR                  t         j,                  fdq| jT                  t         j,                  fdri | jV                  ds | jX                  dt | jZ                  du | j\                  dddadv| j^                  dw | j`                  dx | jb                  t         j,                  fdy| jd                  t         j,                  fdz| jf                  ddad{| jh                  d| | jj                  t         j,                  fd}| jl                  t         j,                  fd~| jn                  d | jp                  d | jr                  t         j,                  fd| jt                  d | jv                  d i | j                  d | jx                  d | jz                  d | j|                  d | j~                  d | j                  dNddad| j<                  d | j                  t         j,                  fd| j                  d | j                  d | j                  d | j                  dِd| j,                  d | j                  d | j                  d | j                  d | j                  d i | j                  d | jr                  d | j                  d | j                  d | j                  d | j                  dِd| j                  d | j                  dddad| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dd| j                  d | j                  d i | j                  d | j                  d | j                  d | j                  d | j                  dOd| j                  d | j                  d | j                  t         j,                  fd| j                  d | j                  dFd| j                  d | j                  d | j                  dِd| j                  d | j                  d | j<                  d֐d| j                  d i | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddt         j,                  fd| j                  dddd| j                  dܐd| j                  d | j                  d | j                  d | j                  dÄ | jl                  dĄ | jp                  d(dń| j                  dƄ | j                  dǄ | j                  dȄ | j                  dɄ | j                  dՐ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 (P  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>      r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  output_sizes     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )inputsrD  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  otherr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  batch1batch2alphabetar>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  tensor1tensor2valuer>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rW  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  mat1mat2rU  rT  r>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  matvecrU  rT  r>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  vec1vec2rU  rT  r>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  thetasizealign_cornerss      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  dims     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   Fc                      yr9  r;  )r=  rO  trolatol	equal_nans        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  ptraininplaces       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rm  keepdimr>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rz  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  msgs     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  tensorss    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  kernel_sizer  padding	ceil_modecount_include_pads         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rQ  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  )grad_outr=  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r=  r  r  r  input_gweight_gbias_gs           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  r  r  counts           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  	generatorr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  input1input2r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   r  c                      yr9  r;  r=  targetr  size_averagereduce	reduction
pos_weights          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  weights	minlengths      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  probr  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r_  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  selfri  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  
boundaries	out_int32rightr>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  rm  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )x1x2rt  compute_modes       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2         ?c                      yr9  r;  r=  rT  rv  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   )r>  c                      yr9  r;  )r>  matricess     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  groupss     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  upperr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  check_errorsr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  numeln_binsratio	bit_widths        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  chunksrm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  minmaxr>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  r>  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r2   c                      yr9  r;  )r=  
correctionfweightsaweightss       r/   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rwith_replacements      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )realimags     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )absangs     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  ords     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  padrZ  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  dilationr  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  )	r=  r  r  r  r  r  
transposedoutput_addingr  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  output_paddingr  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r$  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r$  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  r  r  marginr  r  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r2   c                      yr9  r;  )r  r  rm  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>#  r@  r2   c                      yr9  r;  r=  rO  rm  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r2   r:  c                      yr9  r;  r1  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r2   c                      yr9  r;  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r/   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r2   c                      yr9  r;  r=  rm  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r2   c                      yr9  r;  r<  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r2   c                      yr9  r;  r=  rm  r>  r`   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r2   c                      yr9  r;  r?  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r2   c                      yr9  r;  yxrm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r2   c                      yr9  r;  r<  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r2   c                      yr9  r;  r=  diagonalr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r2   c                      yr9  r;  rL  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r2   c                      yr9  r;  )r=  offsets     r/   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r2   c                      yr9  r;  )r=  nrm  prependappendr>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r2   c                      yr9  r;  r=  rP  dim1dim2s       r/   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r2   c                      yr9  r;  rV  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r2   c                      yr9  r;  )r=  srcrP  rW  rX  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r2   c                      yr9  r;  )r  r[  ri  r  storage_offsets        r/   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r2   c                      yr9  r;  )r=  rO  rt  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r2   c                      yr9  r;  r=  rO  rounding_moder>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r2   c                      yr9  r;  ra  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r2   c                      yr9  r;  rs  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r2   c                      yr9  r;  r=  r_  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r2   c                      yr9  r;  )r^  r_  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r2   c                      yr9  r;  r=  indices_or_sectionss     r/   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r2   c                      yr9  r;  r=  UPLOr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r2   c                      yr9  r;  rp  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r2   c                      yr9  r;  )equationoperandss     r/   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r2   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>L  r@  r2   c
                      yr9  r;  )
r=  r  offsetsry  rz  r{  moder   per_sample_weightsrx  s
             r/   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r2   c                      yr9  r;  r=  r`   rn   r_   requires_grads        r/   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r2   c                      yr9  r;  r=  rO  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>T  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>V  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r2   c                      yr9  r;  )r=  scale
zero_pointaxis	quant_min	quant_maxs         r/   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r2   c                      yr9  r;  )r=  r  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r2   c                      yr9  r;  )rD  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>]  r@  r2   c                      yr9  r;  r=  packed_weightr  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r2   c                      yr9  r;  r=  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r2   c                      yr9  r;  )r=  abs      r/   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r2   c                      yr9  r;  r=  rt  ru  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r2   c                      yr9  r;  r=  rR  rm  norms       r/   r?  z'get_testing_overrides.<locals>.<lambda>j  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r2   c                      yr9  r;  r=  srm  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>s  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r2   c                      yr9  r;  )r=  	start_dimend_dims      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  dimss     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rz  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  exponentr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  
fill_valuer>  r`   rn   r_   r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  	dep_tokens      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rm  indexr>  sparse_grads        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  re  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  spacingrm  
edge_orders       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  gridinterpolation_modepadding_moderj  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  
num_groupsr  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  	r=  hxparams
has_biases
num_layersdropoutru  bidirectionalbatch_firsts	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  w_ihw_hhb_ihb_hhs         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  lambds     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  valuesr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r,  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  binsr  r  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r  densityr>  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r   r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  taus     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r^  r_  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rj  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rm  r  sources       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r"  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  indicesr  
accumulates       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rm  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rm  r  rZ  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rm  r  r#  r  include_inputs         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rZ   s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )eteassume_uniqueinverts       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r-  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r-  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  )	r=  r  r  r  r  use_input_statsr  r  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rO  rtolrp  rq  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  r=  r  r  r  r  
log_targets         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  krm  r{  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  	hermitianr  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rZ  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )LDpivotsBrZ  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  normalized_shaper  r  espr  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  endr  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rX  r_  XrR  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rm  r`   s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rD  rC  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  namesr{  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  )	databatch_sizesr  r  r  r  r   ru  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )Apivot	get_infosr>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r+  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  maskrZ  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r#  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )LUr^  r_  leftadjointr>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rR  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rR  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rn  rZ  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  return_indicesr  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r2   c                      yr9  r;  )r=  rm  r{  r`   r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                       yr9  r;  )r  r.   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  exponential_average_factorepsilons           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  	r=  r  r  r  r  r  r  	benchmarkdeterministics	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  )	r=  r  zrT  r  r  r  r  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c
                      yr9  r;  )
r=  r  r  r  r%  r  r  r  r  r  s
             r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c	                      yr9  r;  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r2   c                      yr9  r;  rz  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r2   c                      yr9  r;  r=  r#  destinations      r/   r?  z'get_testing_overrides.<locals>.<lambda>#  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r2   c                      yr9  r;  )r=  
descendingr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>&  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r2   c                      yr9  r;  )r=  num_samplesreplacementr>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r2   c                      yr9  r;  )r=  rb  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r2   c                      yr9  r;  r=  rt  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r2   c                      yr9  r;  )r=  rm  startrQ  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r2   c                      yr9  r;  )r=  nanposinfneginfr>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  r  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r2   c                      yr9  r;  r=  ra  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r2   c                      yr9  r;  )r=  r  r  NCHxWgroupr  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r2   c                      yr9  r;  )r=  rt  rm  r{  r`   s        r/   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r2   c                      yr9  r;  rC  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r2   c                      yr9  r;  rC  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r2   c                      yr9  r;  r=  rD  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r2   c                      yr9  r;  rg  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r2   c                      yr9  r;  r=  rt  r  rv  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r2   c                      yr9  r;  r=  r  r  r  r  r  divisor_overrides          r/   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  r  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r2   c                      yr9  r;  r=  r  r  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r2   c                      yr9  r;  r+  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r2   c                      yr9  r;  )r=  r  r  r  ignore_indexr  r  label_smoothings           r/   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r2   c                      yr9  r;  r4  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r2   c                      yr9  r;  rw  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r2   c                      yr9  r;  )r=  r  r}  ry  rz  r{  r~  r   r  include_last_offsetrx  s              r/   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r2   c                      yr9  r;  )r=  rD  r  r  r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r2   c                      yr9  r;  r=  r  rD  output_ratior  _random_sampless         r/   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r2   c                      yr9  r;  )r=  r  varr   r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r2   c                      yr9  r;  )r=  approximates     r/   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r2   c                      yr9  r;  )r=  r  r~  r  rj  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r2   c                      yr9  r;  )r=  r  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r2   c                      yr9  r;  )logitsr  hardr  rm  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r2   c                      yr9  r;  )r=  min_valmax_valrv  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r8  r  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  ri  scale_factorr~  rj  recompute_scale_factor	antialiass          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rT  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  negative_sloperv  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  ri  rT  rU  rX  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rm  _stacklevelr`   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rz  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r+  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  r  r  r  r  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rC  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r&  r  r  r  rD  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rJ  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rJ  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r3  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  )r=  r  rt  r,  r  r  r  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r3  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r	  r  r  s          r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rt  rm  r  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rZ   num_classess     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r~  rZ  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  r  rt  r  r{  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  	log_inputr   r  r  r  r  s           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rv  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  ra  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  lowerr  r  rv  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rO  rP  rZ  r]  rX  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  r  rU  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  deltas       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r3  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r:  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r:  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rU  	thresholds      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  rZ  rv  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c
                      yr9  r;  
anchorpositivenegativer,  rt  r  swapr  r  r  s
             r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   )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@  r2   c                      yr9  r;  )r=  r  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rZ   r  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rZ   r  stdr  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rZ   vals     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rZ   r  r~  nonlinearityr  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  as_tuples     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   )r  c                     yr9  r;  )r=  ri  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  rt  rm  r{  r>  r`   s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  rm  r{  r>  r`   s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      	 yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  s     13 r2   c                      yr9  r;  )vpowrm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  input3r  	transposes        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  ro  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  rm  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  qrN  rm  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rconds     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  rZ  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  upscale_factors     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  downscale_factors     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  rq  r   r  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rs  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r`   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r#  r'  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  somer>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r~  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  r  rm  r{  interpolationr>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  scaleszero_pointsr  r`   s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r`   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r2   c                      yr9  r;  )r=  r`   reduce_ranges      r/   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r2   c                      yr9  r;  )r=  r  r  r  r  r  output_scaleoutput_zero_points           r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  r  s                 r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   r      c                      	 yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  s
     " r2   c                      	 yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  s     !# r2   c                      	 yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  s     !#	 r2   c                      yr9  r;  r  s                 r/   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r2   c                      yr9  r;  r  s                 r/   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r2   c                      yr9  r;  )r=  highr`   rn   r_   r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r2   c                      yr9  r;  r1  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r2   c                      yr9  r;  )r=  rt  rm  maxnormr>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r2   c                      yr9  r;  )r=  shapes     r/   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r2   c                      yr9  r;  rx  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r2   c	                      yr9  r;  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r2   c	                      yr9  r;  r  s	            r/   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r2   c                      yr9  r;  )r=  shiftsr  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r2   r   r  c                      yr9  r;  )r=  rX  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r2   c                      yr9  r;  )r  r  compressed_indices_dtypes      r/   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r2   c                      yr9  r;  rz  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r2   c                      yr9  r;  )r=  rO  rT  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r2   c                      yr9  r;  r]  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r2   c                      yr9  r;  r=  rm  r  r[  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r2   c                      yr9  r;  )r=  rm  r  r[  r  include_selfs         r/   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r2   c                      yr9  r;  )sorted_sequencer=  r  r  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r2   c                      yr9  r;  )r  r  lengthsr&  r}  r  unsafes          r/   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r2   c                      yr9  r;  )r=  rm  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r2   c                      yr9  r;  )r=  r[  rm  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r2   c                      yr9  r;  r=  r[  rm  r  rh  steps         r/   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r2   c                      yr9  r;  r  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r2   c                      yr9  r;  ru  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>T  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>V  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r2   c                      yr9  r;  rg  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r2   c                      yr9  r;  rg  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r2   c                      yr9  r;  rw  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r2   c                      yr9  r;  )r  r_  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r2   c                      yr9  r;  )r  r_  r  r  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r2   )stabler>  c                     yr9  r;  )r=  rm  r  r1  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r2   c                      yr9  r;  rZ   split_size_or_sectionsrm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r2   c                      yr9  r;  r4  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>b  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>d  r@  r2   c                      yr9  r;  r<  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r2   c                      yr9  r;  r]  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r2   c
                      yr9  r;  )
r=  rJ  rK  rL  rM  rN  pad_moderO  rP  rR  s
             r/   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r2   c                      yr9  r;  r  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r2   c                      yr9  r;  rF  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>s  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r2   c                      yr9  r;  )r  r  cs      r/   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  
compute_uvr>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  rm  Ms       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  full_matricesr>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r=  dim0rW  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  axis0axis1s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rw  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rz  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rw  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rN  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rO  r>  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r&  rm  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  inds     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  r>  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rk  rm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rX  rm  r  r>  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rI  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r^  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rB  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rB  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  r  unitriangulars        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r_  r  r  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rL  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c
                      yr9  r;  r  s
             r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rL  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  rm  sizesr  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  sortedreturn_inversereturn_countsrm  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r=  r  r  rm  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r&  r   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r4  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r4  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r<  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )rD  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rl  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  rj  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )	conditionrD  rC  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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@  r2   c                      yr9  r;  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  levels     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )primaltangentr  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  ri  r  r]  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rP  rW  rX  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   )implicitc                     yr9  r;  )r  ri  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rm  r  rQ  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  ri  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rm  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rm  r  rh  r   s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  
split_sizerm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  split_sizesrm  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r_  rW  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>	  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>
  r@  r2   c                      yr9  r;  r  r`   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  	dimensionri  r   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  rO  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>   r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>!  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>"  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>#  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>$  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>%  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>&  r@  r2   c                      yr9  r;  )r  arrays     r/   r?  z'get_testing_overrides.<locals>.<lambda>'  r@  r2   c                      yr9  r;  )r  idxs     r/   r?  z'get_testing_overrides.<locals>.<lambda>(  r@  r2   c                      yr9  r;  )r  memos     r/   r?  z'get_testing_overrides.<locals>.<lambda>)  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>*  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>+  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>,  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>-  r@  r2   c                      yr9  r;  )r  format_specs     r/   r?  z'get_testing_overrides.<locals>.<lambda>.  r@  r2   c                      yr9  r;  )r  protos     r/   r?  z'get_testing_overrides.<locals>.<lambda>/  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>0  r@  r2   )tensor_contentsc                     yr9  r;  )r  r'  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>1  r@  r2   c                      yr9  r;  )r  rX  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>2  r@  r2   c                      yr9  r;  )r  ds     r/   r?  z'get_testing_overrides.<locals>.<lambda>3  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>4  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>5  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>6  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>7  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>8  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>9  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>:  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>;  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda><  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>=  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>?  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>@  r@  r2   c                      yr9  r;  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r/   r?  z'get_testing_overrides.<locals>.<lambda>A  r@  r2   c                      yr9  r;  )r  r:  r;  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>B  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>C  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>D  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>E  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>F  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>G  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>H  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>I  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>J  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>K  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>L  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>M  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>N  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>O  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>P  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>Q  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>R  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>S  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>T  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>U  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>V  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>W  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>X  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>Y  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>Z  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>[  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>\  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>]  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>^  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>_  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>`  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>a  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>b  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>c  r@  r2   c                      yr9  r;  )r  r`   non_blockingr.   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>d  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>e  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>f  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>g  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>h  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>i  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>j  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>k  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>l  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>m  r@  r2   c                      yr9  r;  )r  r  rv  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>n  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>o  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>p  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>q  r@  r2   c                      yr9  r;  )r  orderellipsis_idxs      r/   r?  z'get_testing_overrides.<locals>.<lambda>r  r@  r2   c                      yr9  r;  )r  callables     r/   r?  z'get_testing_overrides.<locals>.<lambda>s  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>t  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>u  r@  r2   c                      yr9  r;  )r  gradientretain_graphcreate_graphrF  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>v  r@  r2   c                      yr9  r;  r  rg   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>w  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>x  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>y  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>z  r@  r2   )r  c                     yr9  r;  )r  mediansigmar  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>{  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>|  r@  r2   c                      yr9  r;  )r  	coalesceds     r/   r?  z'get_testing_overrides.<locals>.<lambda>}  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>~  r@  r2   c                      yr9  r;  )r  r[  ra  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r[  rP  rW  rX  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                     yr9  r;  )r  r  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  rZ  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                     yr9  r;  )r  rt  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rZ   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                     yr9  r;  )r  r  r  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rZ   rs  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rD  rC  rs  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r_  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  rO  assigns      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  rQ  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r&  rZ   r'  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                     yr9  r;  )r  from_tor  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  streams     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  hooks     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  names     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r#  r]  ri  r  s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r[  rm  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r{  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r[  rm  r  rh  r   s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  accumulate_matchess      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  size1size2	dense_dims       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r^  r_  rU  rT  r>  s         r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  repss     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r`   ra  copyrg   s        r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   )masked_gradc                     yr9  r;  r  r`   r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r  s      r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r   s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s     r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r  r   drivers       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  )r  r_   ra  r.   s       r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   c                      yr9  r;  r  s    r/   r?  z'get_testing_overrides.<locals>.<lambda>  r@  r2   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)      ?)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  )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  )g{Gz?F)g-C6?g      ?r  )N   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  Nr  )r  r  g-q=N)r   r   )r
  r  F)TFNr	  Nr  )Nr  )g      ?gUUUUUU?FF)Nr  )NNr  r  )r  r  )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reflectFTN)TTN)   r  N)r  N)TFF)TFFN)r  r  )Nr   NN(  r4   r5   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diffrM  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greater_equalgeqrfi0inneroutergerrw  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrink	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacr  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr~  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_lossr  r   local_response_norm
logsigmoid	lp_pool1d	lp_pool2d	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_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_precisionr  r_   r`   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizern   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_rq   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_rY   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_rh   random_record_streamrefine_namesregister_hook"register_post_accumulate_grad_hookrenamerepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_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extendrs  update)r5   retprivateuse1_backend_nameret2ignoredrX  r  r  subnamer  r    s              r/   r   r     sW   6 \\Fw%		-w%2w% 	!!#@w% 	!!#A	w%
 	

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

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

.+w%, 	

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

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

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

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

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

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

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

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

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

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

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

*iw%j 	

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

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

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

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

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

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

.}w%~ 	=w%@ 	>Aw%B 	LCw%D 	BEw%F 	=Gw%H 	\Iw%J 	)Kw%L 	

GMw%N 	&Ow%P 	'Qw%R 	2Sw%T 	2Uw%V 	tYw%\ 	(]w%^ 	1_w%` 	4aw%b 	Kcw%d 	*ew%f 	'gw%h 	&iw%j 	.kw%l 	,mw%n 	!1ow%p 	*qw%r 	3sw%t 	)uw%v 	Www%x 	%yw%z 	 d}w%@	 	rA	w%B	 	

+C	w%D	 	NE	w%F	 	""$cG	w%H	 	!LI	w%J	 	 SK	w%L	 	sM	w%N	 			4O	w%P	 	6Q	w%R	 	3S	w%T	 	;U	w%V	 	

;W	w%X	 	0Y	w%Z	 	  K[	w%\	 			-]	w%^	 	<_	w%`	 	/a	w%b	 	/c	w%d	 	

.e	w%f	 	:g	w%h	 	;i	w%j	 	&k	w%l	 	.m	w%n	 	<o	w%p	 	5q	w%r	 	;s	w%t	 	<u	w%v	 	/w	w%x	 	Iy	w%z	 	

s{	w%|	 	O}	w%~	 	3	w%@
 	

5A
w%B
 	EC
w%D
 	BE
w%F
 	!!  $AG
w%H
 	8I
w%J
 	<K
w%L
 	=M
w%N
 	7O
w%P
 	?Q
w%R
 	 FS
w%T
 	!!#]U
w%V
 	[W
w%X
 	>Y
w%Z
 	/[
w%\
 	!!#@]
w%^
 	  "M_
w%`
 	 <a
w%b
 	*c
w%d
 	!1e
w%f
 			-g
w%h
 	8i
w%j
 	

5k
w%l
 	lm
w%n
 	lo
w%p
 	lq
w%r
 	%%tu
w%x
 	

.y
w%z
 	V{
w%|
 	0}
w%~
 	3
w%@ 	5Aw%B 			-Cw%D 	8Ew%F 	

5Gw%H 	tKw%N 	  "}Ow%P 	))+vQw%R 	%%'hSw%T 	**wWw%Z 	**g]w%` 	 dcw%f 	2gw%h 	

Eiw%j 	<kw%l 	=mw%n 	Aow%p 			4qw%r 	9sw%t 	Uuw%v 	1ww%x 	+yw%z 	:{w%| 	W}w%~ 	!sw%@ 	&&(_Aw%B 	8Cw%D 	!fEw%F 	!VGw%H 	UIw%J 	$$&>Kw%L 	3Mw%N 	:Ow%P 			-Qw%R 	2Sw%T 	:Uw%V 	//1NWw%X 	//1NYw%Z 	//1d[w%\ 	<<>q]w%^ 	//1d_w%` 	<<>qaw%b 	//1dcw%d 	<<>qew%f 	'')Sgw%h 	))+aiw%j 	&& Bmw%p 	&& Bsw%v 	&&xyw%| 	$$&R}w%~ 	00cAw%D 	<<tGw%J 	  "LKw%L 	11iOw%R 	)) LUw%X 	$$x[w%^ 	##%Z_w%` 	%%'\aw%b 	%%'\cw%d 	%%'\ew%f 	!Kgw%h 	%%|kw%n 	)) Jqw%t 	113iuw%v 	  "mww%x 	11z{w%~ 	>>zAw%D 	11zGw%J 	>>zMw%P 	--/uQw%R 	  "FSw%T 	!9Uw%V 	'')zWw%X 	&&(gYw%Z 	**,c[w%\ 	&&(C]w%^ 	$$&`_w%` 	00bcw%f 	)) Iiw%l 	'' Mow%r 	""  %Asw%t 	##%ouw%v 	&&(mww%x 	&&(\yw%z 	""$G{w%| 	//1g}w%~ 	'')^w%@ 	&&(8Aw%B 	%%'mCw%D 	%%'mEw%F 	%%'mGw%H 	//iKw%N 	&&tQw%T 	33tWw%Z 	&&t]w%` 	33tcw%f 	&&tiw%l 	33tow%r 	((*zsw%t 	((*zuw%v 	((*zww%x 	$$&pyw%z 	88 _}w%@ 	--tCw%F 	22VIw%L 	77cOw%R 	$$vUw%X 	%%'XYw%Z 	##%F[w%\ 	!P]w%^ 	--/a_w%` 	,,}cw%f 	!!#;gw%h 	  "Aiw%j 	!!#Bkw%l 	$$&_mw%n 	!!#yow%p 	  "Aqw%r 	  "Asw%t 	  "Auw%v 	88:uww%x 	**  -Ayw%z 	&&(]{w%| 	,,.x}w%~ 	##%Zw%@ 	##%ZAw%B 	$$&LCw%D 	&&(CEw%F 	$$&6Gw%H 	&&(8Iw%J 	%%'XKw%L 	// LOw%R 	==DHQT[`lrvUw%X 	""$bYw%Z 	 O[w%\ 	S]w%^ 	!7_w%` 	&&(xaw%b 	7cw%d 	Few%f 	(gw%h 	

\iw%j 	dkw%l 	  "hmw%n 	   +/#3ow%v 	9ww%x 	dyw%z 	%{w%| 	*}w%~ 	Qw%@ 	!SAw%B 	+Cw%D 	IEw%F 	*Gw%H 	5Iw%J 	IKw%L 	=Mw%N 	AOw%P 	7Qw%R 	 YSw%T 	6Uw%V 	2Ww%X 	-Yw%Z 	d[w%\ 			7]w%^ 	

0_w%` 			Daw%b 	  "2cw%d 	""$4ew%f 	'')9gw%h 	'iw%j 	,kw%l 	7mw%n 	Cow%p 	fqw%r 	isw%t 	""$Vuw%v 	!!#Mww%x 	))+Pyw%z 	""$s{w%| 	   aw%B 	!! aEw%H 	""J#Kw%R 	""L #Uw%^ 	""O 	#aw%l 	%% aow%r 	%% auw%x 	1yw%z 	d{w%| 	d5==Y]mrv}w%~ 	ew%@ 	%Aw%B 	

.Cw%D 	

5Ew%F 	FGw%H 	,Iw%J 	/Kw%L 	4Mw%N 	

3Ow%P 	:Qw%R 	ASw%T 	!;Uw%V 	.Ww%X 	QYw%Z 	x[w%\ 	S]w%^ 	x_w%` 	Saw%b 	

7cw%d 	7ew%f 	/gw%h 	5iw%j 	Pkw%l 	bmw%n 	/ow%p 	

4qw%r 	Msw%t 	8uw%v 	<ww%x 	Zyw%z 	e{w%| 	|}w%~ 	2w%@ 	?Aw%B 	WCw%D 	WEw%F 	

3Gw%H 	1Iw%J 	

.Kw%L 	1Mw%N 			-Ow%P 			-Qw%R 	

.Sw%T 	

.Uw%V 	'Ww%X 	.Yw%Z 			)[w%\ 	

*]w%^ 	8_w%` 	@aw%b 	Wcw%d 	

YeQUYew%f 	Egw%h 	 Piw%j 	

.kw%l 	0mw%n 	;ow%p 	Oqw%r 	8sw%t 			-uw%v 	2ww%x 	

 j{w%~ 			4w%@ 	9Aw%B 			-Cw%D 	)Ew%F 	'Gw%H 	Iw%J 	Kw%L 	'Mw%N 	)Ow%P 	)Qw%R 	(Sw%T 	)Uw%V 	(Ww%X 	)Yw%Z 	([w%\ 	)]w%^ 	)_w%` 	)aw%b 	)cw%d 	0ew%f 			Igw%h 	Aiw%j 	Hkw%l 	8mw%n 	4ow%p 	6qw%r 	/sw%t 	!1uw%v 	!1ww%x 	!1yw%z 	!1{w%| 	,,.K}w%~ 	,,.Kw%@ 	,,.KAw%B 	,,.KCw%D 	/Ew%F 	,Gw%H 	+Iw%J 	,Kw%L 	-Mw%N 	.Ow%P 	,Qw%R 	-Sw%T 	-Uw%V 	 AWw%X 	!BYw%Z 	/[w%\ 	**,I]w%^ 	++-J_w%` 	*aw%b 	+cw%d 	*ew%f 	+gw%h 	++-Jiw%j 	++-Jkw%l 	-mw%n 	 0ow%p 	!!#Dqw%r 	-sw%t 	!Ouw%v 	((*:ww%x 	((*:yw%z 	((*:{w%| 	((*:}w%~ 	""$7w%@ 	,Aw%B 	-Cw%D 	!>Ew%F 	+Gw%H 	-Iw%J 	//1AKw%L 	//1AMw%N 	446SOw%P 	446SQw%R 	446SSw%T 	446SUw%V 	,Ww%X 	@Yw%Z 	))+;[w%\ 	@]w%^ 	>_w%` 	<aw%b 	!cw%d 	

+ew%f 	Kgw%h 			-iw%j 	

.kw%l 	 3mw%n 	  "<ow%p 	:qw%r 	Hsw%t 	Juw%v 	

*ww%x 	

Kyw%z 	%{w%| 	5}w%~ 	1w%@ 	5Aw%B 	 eCw%D 	%%'aEw%F 	

:Gw%H 	!! LKw%N 	

:Ow%P 	2Qw%R 	/Sw%T 	-Uw%V 	<Ww%X 	hYw%Z 	  "g[w%\ 	6]w%^ 	;_w%` 	Law%b 	%%'Wcw%d 	8ew%f 	1gw%h 			-iw%j 	2kw%l 	;mw%n 	2ow%p 	9qw%r 	%%'`sw%t 	11oww%z 	e{w%| 	5}w%~ 	@w%@ 	Aw%B 	""OCw%D 	/Ew%F 	oGw%H 	QIw%J 	'')>Kw%L 	FMw%N 	%COw%P 	>Qw%R 	1Sw%T 	!!#@Uw%V 	6Ww%X 	?Yw%Z 	N[w%\ 	<]w%^ 	##%H_w%` 	0aw%b 	ocw%d 	9ew%f 	2gw%h 	_iw%j 	Okw%l 	Omw%n 	?ow%p 	qw%r 	sw%t 	uw%v 	ww%x 	1yw%z 	/{w%| 	A}w%~ 	/w%@ 	3Aw%B 	4Cw%D 	4Ew%F 	2Gw%H 	3Iw%J 	3Kw%L 	1Mw%N 	2Ow%P 	2Qw%R 	1Sw%T 	2Uw%V 	2Ww%X 	.Yw%Z 	-[w%\ 	.]w%^ 	/_w%` 	Oaw%b 	0cw%d 	ew%f 	3gw%h 	iw%j 	?kw%l 	.mw%n 	/ow%p 	/qw%r 	5sw%t 	0uw%v 	2ww%x 	yw%z 	{w%| 	/}w%~ 	w%@ 	7Aw%B 	4Cw%D 	_Ew%F 	AGw%H 	1Iw%J 	/Kw%L 	/Mw%N 	/Ow%P 			?Qw%R 			?Sw%T 	&&Uw%V 	**22OWw%X 	oYw%Z 	[w%\ 	_]w%^ 	o_w%` 	aw%b 	cw%d 	ew%f 	--/pgw%h 	**,Viw%j 	_kw%l 	mw%n 	oow%p 	qw%r 	sw%t 	uw%v 	ww%x 	yw%z 	{w%| 	##_}w%~ 	w%@ 	Aw%B 	Cw%D 	  /Ew%F 	Gw%H 	  /Iw%J 	##_Kw%L 	  /Mw%N 	$$oOw%P 	  /Qw%R 	Sw%T 	Uw%V 	_Ww%X 	oYw%Z 	[w%\ 	_]w%^ 	  /_w%` 	$$oaw%b 	ocw%d 	ew%f 	_gw%h 	_iw%j 	''//kw%l 	Nmw%n 	oow%p 	oqw%r 	sw%t 	uw%v 	_ww%x 	_yw%z 	O{w%| 	_}w%~ 	Ow%@ 	=Aw%B 	Cw%D 	Ew%F 	/Gw%H 	=Iw%J 	0Kw%L 	8Mw%N 	9Ow%P 	kQw%R 	E4I4IMSw%T 	0E0EIUw%V 	0E0EIWw%X 	0E0EIYw%Z 	MTM[w%\ 	]w%^ 	6_w%` 	e6M6MQaw%b 	>cw%d 	

u/D/DHew%f 	0E0EIgw%h 	0E0EIiw%j 	

u/D/DHkw%l 	

u/D/DHmw%n 	ow%p 	/qw%r 	!Osw%t 	

Ouw%v 	/ww%x 	%2G2GKyw%z 	53H3HL{w%| 	_}w%~ 	,w%@ 	0Aw%B 	HHCw%D 	,Ew%F 	5Gw%H 	1F1FJIw%J 	%2G2GKKw%L 	@Mw%N 	?Ow%P 	0E0EIQw%R 	1F1FJSw%T 	/Uw%V 	Ww%X 	

u/D/DHYw%Z 	_[w%\ 	o]w%^ 	__w%` 	/aw%b 	1cw%d 	/ew%f 	_gw%h 	MTMiw%j 	0kw%l 	0E0EImw%n 	6ow%p 	5qw%r 			(sw%t 	@uw%v 	Eww%x 	?yw%z 	{w%| 	""O}w%~ 	--w%@  	%%A w%B  	C w%D  	oE w%F  	,G w%H  	?I w%J  	GK w%L  	M w%N  	LDLO w%P  	5Q w%R  	3S w%T  	3U w%V  	113HW w%X  	,Y w%Z  	-[ w%\  	B] w%^  	1_ w%`  	-a w%b  	-c w%d  	0e w%f  	  "8g w%h  	Oi w%j  	[k w%l  	?m w%n  	oo w%p  	1F1FJq w%r  	_s w%t  	Wu w%v  	?w w%x  	1y w%z  	&&(W{ w%|  	G} w%~  	'')Q w%@! 	OA!w%B! 	C!w%D! 	E!w%F! 	G!w%H! 	_I!w%J! 	1K!w%L! 	+M!w%N! 			EUZUjUjnO!w%P! 	IIQ!w%R! 	GS!w%T! 	/U!w%V! 	W!w%X! 	/Y!w%Z! 	.[!w%\! 	=]!w%^! 	7_!w%`! 	a!w%b! 	+c!w%d! 	.e!w%f! 	o7  /Fm!w%Ct! 	((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Jr2   
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_argsrn
  r    wrappeds      r/   rs
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped'  sA    &77M!-0,WmUdUfUU(((r2   )	functoolsr   )r    rs
  rn
  s   `@r/   r  z"wrap_torch_function.<locals>.inner&  s%    			) 
	) r2   r;  )rn
  r  s   ` r/   r   r     s    0	 Lr2   rr
  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	  r4   _C_is_torch_function_enabledsetra
  r   _disabled_torch_function_implr*  rf
  	enumerate
issubclassinsert)	rr
  ru
  overloaded_typesoverloaded_argsargarg_typer  iold_args	            r/   _get_overloaded_argsr
  4  s    J  88..0	"%%!#O (s# ,,"67++uxx/U/UU   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%9(: r2   
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   NotImplementedra
  r
  r4   rw
  rz
  r)   warnDeprecationWarning
__module__rd
  _get_current_function_mode	TypeError)r
  rr
  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 )Nr4   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listr4   __all__r   r   dirr   r5   rn  r   r>	  re
  endswithislowerrb
  object
isinstancer
  
ModuleType
__future___Featurers  ra
  r&  __set__r   r   formatrd
  rT  )
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##r2   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;  r2   r/   r   r   b  s     &'**r2   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
  r4   _ops
OpOverloadOpOverloadPacketstrr
  get)fs    r/   r   r   o  sI      !ejj++UZZ-H-HIJ1v%'*..q11r2   c                  R    t               } t        | t        j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   ry
  r4   r5   )r
  methodss     r/   _get_tensor_methodsr
    s&     23#ELL12GNr2   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
  rd
  )r    s    r/   r   r     s!    . &((FDMMY,FFr2   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	  r4   r5   ra
  )inps    r/   r   r     s%    D 9$J5I(JJr2   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__`` overrideable functions within a dynamic scope,
    without having to actually create a tensor subclass or manually
    monkey-patch functions in the PyTorch API.  Some common situations
    where you should use a mode:

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

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

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

    Independent subclasses of :class:`TorchFunctionMode` are compositional:
    modes can be pushed onto a stack using ``with MyMode():``.
    When you call functions in the PyTorch API inside your
    ``__torch_function__`` implementation, by default, they will forward on to
    the next mode on the mode stack.  If you want recursively call back into
    your current ``__torch_function__`` implementation, either explicitly
    invoke ``self.__torch_function__(...)``, or use the context manager
    ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
    API self-referential (beware of infinite loops, in this case!)
    r  Nc                      y r  r;  r  s    r/   r   zTorchFunctionMode.__init__  s    r2   c                     t         r  )NotImplementedErrorr  r    r
  r-   r.   s        r/   r   z$TorchFunctionMode.__torch_function__  s    !!r2   c                     t        |        | S r  )
_push_moder  s    r/   	__enter__zTorchFunctionMode.__enter__  s    4r2   c                     t                y r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r/   __exit__zTorchFunctionMode.__exit__  s    r2   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'    ^	
 ''r2   )r#   Nr;  N)rd
  r
  __qualname____doc____annotations__r   r   r
  r
  classmethodr
  r;  r2   r/   r
  r
    s7    < "  r2   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Kr2   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
    !$'r2   c                      t               } | S r  )r   olds    r/   r
  r
    s    
#
%CJr2   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$$r2   r
  )rd
  r
  r
  r   r;  r2   r/   r
  r
    s    %r2   r
  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  )r4   rw
  _RestorePythonTLSSnapshotr;  r2   r/   r   r   &  sC      
	+	+	- 		  	 s#   A61	A36?A)z.*is deprecated, please use.*r4   r  )9r
  r
  r
  
contextlibrt
  r
  r)   r   typingr   r   r   r   r	   r
   r   r   r4   torch._Cr   r   r   r   r   r   r   r   r   r
  r
  r1   	lru_cacher   r6  r   r   r
  r   r   r   r   r
  r   r   r
  r
  r   r   r
  r
  r
  r
  r
  contextmanagerr
  r
  r   r;  r2   r/   <module>r
     s  ,        H H H 
 
 
  1 
     	 F TSs8} S  Sl Tc(m  2 TwtHh$67 w  wt##H #P *.KC=K3%+&K 
#YK\UUC=U
 	Up ! . '	  * 	  TS$	$sDN"
#T(C-%8
89S$ S$l 	+4T(^(;#< 	+ 	+ 2 2( TS]   Gx GD G G2"KJ5 5pL
A
(
  %- %  r2   