
    Pg;             	          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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
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 Z0 G d d       Z1d! Z2d" Z3d# Z4d$ Z5ejl                  d%        Z7 G d& d'e1      Z8ejl                  d(        Z9ejl                  d)        Z:y),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Optional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     V/var/www/html/suriana-translation/venv/lib/python3.12/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperU   sJ    $$& 	)##;f ((		) 	) 	)s   %AAr   )r!   r"   r#   r1   s   ``` r0   _disable_user_warningsr2   =   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                  j                  t         j                  j                  t         j                  j                  t         j                  j                  t         j                  t         j                  j                  t         j                  j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                   t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j
                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                   t         j                  j                  j"                  t         j                  j                  j$                  t         j                  j                  j&                  t         j                  j                  j(                  t         j*                  j,                  t        t        t         j.                  t         j0                  t         j2                  t         j4                  t         j6                  t         j8                  t         j:                  t         j<                  t         j>                  t         j@                  t         jB                  t         jD                  t         jF                  t         jH                  t         jJ                  t         jL                  t         jN                  t         jP                  t         jR                  t         jT                  t         jV                  t         jX                  t         jZ                  t         j                  j                  j\                  t         j^                  t         j`                  t         jb                  t         jd                  t         jf                  t         jh                  t         jj                  t         jl                  t         jn                  t         jp                  t         jr                  t         jt                  t         jv                  t         jx                  t         jz                  t         j|                  t         j~                  t         j                  t         j                  t         j                  t         j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  j                  | j                  j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j,                  | j                  S )a%  
    Return public functions that cannot be overridden by ``__torch_function__``.

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

    Examples
    --------
    >>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions()
    True
    >>> torch.add in torch.overrides.get_ignored_functions()
    False
    )torchTensortypename	is_tensor
is_storageset_default_tensor_typeset_default_deviceget_default_deviceset_rng_stateget_rng_statemanual_seedinitial_seedseedsaveloadset_printoptionsforkget_default_dtypeget_num_interop_threadsget_num_threadsinit_num_threadsimport_ir_moduleimport_ir_module_from_bufferis_anomaly_enabledis_anomaly_check_nan_enabledis_grad_enabledmerge_type_from_type_commentparse_irparse_schemaparse_type_commentset_anomaly_enabledset_flush_denormalset_num_interop_threadsset_num_threadswait	as_tensor
from_numpytensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutalign_tensorsarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptableemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrandrandnrandintrandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   sigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvitals_enabled	set_vitalread_vitalsvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_countr6   s    r0   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 	  {@| 	}@~ 	@ @r3   c                      t         j                  } | j                  j                  | j                  j                  | j
                  j                  hS )a  
    Return public functions that do not wrap in a subclass when invoked by
    the default ``Tensor.__torch_function__`` that preserves subclasses.  Typically,
    these functions represent field accesses (i.e., retrieving a Tensor that
    is stored somewhere on the Tensor) as opposed to computation.  Users of
    these functions expect object identity to be preserved over multiple accesses
    (e.g., ``a.grad is a.grad``) which cannot be upheld if we're wrapping on
    the fly every time (furthermore, the tensor stored here might already be
    the subclass, in which case wrapping really ought not to happen).

    Not ALL property accessors have this property; for example ``Tensor.T`` actually
    just creates a new transposed tensor on the fly, and so we SHOULD interpose on
    these calls (you need to check the implementation of the function to see if
    this is the case or not).  Additionally, if a property accessor doesn't return a Tensor,
    it doesn't have to be on this list (though it is harmless if it is).
    )r5   r6   _baser'  grad_gradr2  s    r0   get_default_nowrap_functionsr7  x  s>    $ \\F r3   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                  d܄ t         j                  dd݄t         j                  ddބt         j                  dd߄t         j                  ddt         j                  ddt         j                  ddt         j                  ddi t         j                  d t         j                  d t         j                  d t         j                  d dt         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  j                  d t         j                  ddt         j                  d i t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  d t         j                  ddt         j                  d  t         j                  ddt         j                  d t         j                  d t         j                  dؐdt         j                  dؐdi t         j                  d t         j                   d t         j                  dؐdt         j                  j                  dؐd	t         j                  j                  dd
t         j                  d t         j
                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  dېdt         j                  d t         j                  ddi t         j                   ddt         j"                  d t         j$                  dݐdt         j                  j&                  ddt         j                  j(                  ddt         j                  j*                  ddt         j,                  d dt         j.                  dؐdt         j0                  dؐdt         j2                  dؐd t         j4                  dؐd!t         j6                  dؐd"t         j8                  dؐd#t         j:                  d	d$t         j<                  dؐd%t         j>                  dؐd&t         j@                  dؐd'i t         jB                  dؐd(t         jD                  dؐd)t         jF                  dؐd*t         jH                  dؐd+t         jJ                  d, 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                  dd3t         jZ                  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d:t         jh                  d; t         jj                  d< t         jl                  dؐd=t         jn                  dؐd>t         j                  jb                  dd?t         j                  jp                  dd@t         j                  jr                  d
dAt         j                  jd                  d
dBt         j                  jn                  dؐdCt         jt                  dD t         j                  jt                  dؐdEt         j                  jv                  ddFt         j                  jx                  dؐdGt         jz                  dH t         j                  jz                  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                  dؐdTt         j                  dU t         j                  dؐdVt         j                  dؐdWt         j                  dؐdXt         j                  dY t         j                  dZ i t         j                  d[ t         j                  d\ t         j                  d] t         j                  d^ t         j                  d_ t         j                  dؐd`t         j                  ddat         j                  db t         j                  dc t         j                  dddt         j                  dؐdet         j                  dؐdft         j                  ddgt         j                  dؐdht         j                  di t         j                  dj t         j                  ddki t         j                  dl t         j                  dm t         j                  dn t         j                  ddot         j                  dp t         j                  ddqt         j                  dr t         j                  dؐdst         j                  dؐdtt         j                  dؐdut         j                  dؐdvt         j                  dؐdwt         j                  j                  j                  dx t         j                  j                  j                  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                  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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                  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i t         j                  j                  j                  ddt         j                  j                  j                  d!dt         j                  j                  j                  ddt         j                  j                  j                  d"dt         j                  j                  j                  d#dt         j                  j                  j                   ddt         j                  j                  j                  d$dt         j                  j                  j,                  ddt         j                  j                  j                  d%dt         j                  j                  j                  dؐdt         j                  j                  j                  d&dt         j                  j                  j>                  d'dt         j                  j                  j                  d t         j                  j                  j                  ddt         j                  j                  j                   ddt         j                  j                  j"                  ddt         j                  j                  jf                  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                  d0dÄt         j                  j                  jD                  dĄ t         j                  j                  jF                  dܐdńt         j                  j                  jH                  dܐdƄt         j                  j                  jJ                  d1dǄt         j                  j                  jL                  d2dȄt         j                  j                  jN                  dܐdɄt         j                  j                  jP                  dܐdʄt         j                  j                  jR                  dܐd˄t         j                  j                  jT                  d3d̄t         j                  j                  jV                  d4d̈́t         j                  j                  jX                  d5d΄t         j                  j                  jZ                  d+dτt         j                  j                  j\                  d'dЄt         j                  j                  j^                  d'dфi t         j                  j                  j`                  d6d҄t         j                  j                  jb                  ddӄt         j                  j                  jd                  dԄ t         j                  j                  jf                  dՄ t         j                  j                  jh                  dܐdքt         j                  j                  jj                  d7dׄt         j                  j                  jl                  ddNdd9d؜dلt         j                  j                  jn                  ddڄt         j                  jp                  jr                  d8dۄt         j                  jp                  jt                  d8d܄t         j                  jp                  jv                  d݄ t         j                  jp                  jx                  d9dބt         jz                  dܐd߄t         j|                  dzddt         j~                  d t         j                  d:dt         j                  j                  d;di t         j                  j                  d<dt         j                  j                  	 d=dt         j                  d>dt         j                  d:dt         j                  d t         j                  d t         j                  d?dt         j@                  d/dt         j                  d t         j                  d@dt         j                  ddt         j                  dAdt         j                  j                  dBdt         j                  d t         j                  d t         j                  dؐdt         jB                  d i t         j                  dؐdt         j                  dؐdt         jD                  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                  dCdt         j                  dDdt         j                  dDdt         j                  d i t         j                  d t         j                  d t         j                  d	 t         j                  d
 t         j                  d t         j                  	 	 dEdt         j                  	 	 dFdt         j                  	 	 dG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i t         j                  j                  ddt         j                  d t         j                  d t         j                  dؐdt         jF                  dܐdt         j                  dؐdt         j                  dؐd t         j                  dؐd!t         j                  d" t         jJ                  d1d#t         j                  d$ t         j                  dߐd%t         j                  d& t         j                  dߐd't         j                  dؐd(t         j                  dHd*t         j                  dؐd+i t         j                  dؐd,t         j                   d- t         jL                  d2d.t         j                  dؐd/t         j                  dd0t         j                  dِd1t         j                  d2 t         j
                  d3 t         j                  dd4t         j                  dd5t         j                  dId6t         j                  d7 t         j                  d8 t         j                  dJd9t         j                  dJd:t         jN                  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&                  dؐdBt         j(                  dC t         j                  j(                  dD t         j*                  dE t         j,                  dF t         j\                  dؐdGt         j                  j.                  ddHt         j                  j0                  d
dIt         j2                  dKdddJ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                  dؐdTt         jF                  dLdUt         jH                  dؐdVt         jJ                  dؐdWt         jL                  dؐ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                  dh t         jn                  di t         jp                  dؐdjt         jr                  dMdkt         jt                  dNdlt         j                  jr                  ddmt         j                  jv                  dؐdnt         jx                  do t         jz                  dp i t         j|                  j~                  dq t         j|                  j                  dr t         j|                  j                  ds t         j|                  j                  dt t         j|                  j                  du t         j|                  j                  dؐdvt         j|                  j                  dؐdwt         j|                  j                  dؐdxt         j|                  j                  dؐdyt         j|                  j                  dz t         j|                  j                  d{ t         j|                  j8                  d| t         j|                  j:                  d} t         j|                  j                  d~ t         j|                  j<                  d t         j|                  j@                  d t         j|                  j                  d i t         j|                  jB                  d t         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  j                  d t         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  jB                  d t         j|                  j                  d t         j|                  j>                  dؐdt         j|                  jV                  d t         j|                  jX                  ddi t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  dؐdt         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  d t         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  j                  dؐdt         j|                  j$                  d i t         j|                  j\                  dؐdt         j|                  j                  d t         j|                  j                  dؐdt         j|                  jL                  dؐdt         j|                  j                  dؐdt         j                  d t         j                  d t         j                  dߐdt         j                  dؐdt         j                  dؐdt         j                  j                  ddt         j                  j                  dؐdt         j                  dOdt         j                  ddt         jh                  dܐdt         j                  d t         j                  ddi t         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  dPdt         j                  j                  d?dt         j                  ddt         jj                  d7dt         j                  ddt         j                  d t         j                  dؐdt         j                  ddt         j                  d t         j                  dQdt         j                  ddÄt         j                   dĄ t         j                  ddńi t         j                  ddƄt         j                  ddǄt         j                  dؐdȄt         j                  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ք i t         j&                  dׄ t         j(                  dؐd؄t         j*                  dل t         j,                  ddڄt         j.                  ddۜd܄t         j0                  d݄ t         j2                  dބ t         j4                  d߄ t         j6                  d t         j8                  d t         j:                  dJdt         j<                  ddt         j>                  d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 t         jV                  d t         jX                  ddt         jZ                  d t         j\                  d t         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 | j                  d | j                  ddd| j                  d | 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                  j                  d' | j                  j                  d( | j                  d) | j                  d* | 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                  j                  dJ | j                  j                  dK | j
                  j                  dL | j                  d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\ | j*                  d] | j,                  d^ | j.                  dd_| j0                  t         j2                  fd`i | j4                  t         j2                  fda| j6                  t         j2                  fdb| j8                  t         j2                  fdc| j:                  d)dddde| j<                  df | j>                  dg | j@                  t         jB                  fdh| jD                  dܐdi| jF                  t         j2                  fdj| jH                  t         j2                  fdk| jJ                  t         j2                  fdl| jL                  t         j2                  fdm| jN                  t         j2                  fdn| jP                  do | jR                  dp | j                  ddq| jT                  dr i | jV                  dܐds| jX                  t         j2                  fdt| jZ                  t         j2                  fdu| j\                  dv | j^                  dw | j`                  dx | jb                  dddddy| jd                  dz | jf                  d{ | jh                  t         j2                  fd|| jj                  t         j2                  fd}| jl                  dddd~| j                  d | jn                  t         j2                  fd| jp                  t         j2                  fd| jr                  d | jt                  d i | jv                  t         j2                  fd| jx                  d | jz                  d | j                  d | j|                  d | j~                  d | j                  d | j                  d | j                  dRdddd| j>                  d | j                  t         j2                  fd| j                  d | j                  d | j                  d | j                  dܐd| j0                  d | j                  d i | j                  d | j                  d | j                  d | j                  d | jt                  d | j                  d | j                  d | j                  d | j                  dܐd| j                  d | j                  ddddd| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d i | j                  dd| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dSd| j                  d | j                  d | j                  t         j2                  fd| j                  d | j                  dJd| j                  d | j                  d | j                  dܐd| j                  d i | j                  d | j>                  dِd| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddt         j2                  fd| j                  dddd| j                  dߐd| j                  dÄ | j                  dĄ | j                  dń | j                  dƄ | jn                  dǄ | jr                  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 (T  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     r0   <lambda>z'get_testing_overrides.<locals>.<lambda>      r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  output_sizes     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )inputsrE  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  otherr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  batch1batch2alphabetar?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  tensor1tensor2valuer?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rX  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  mat1mat2rV  rU  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  matvecrV  rU  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  vec1vec2rV  rU  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  thetasizealign_cornerss      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  dims     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   Fc                      yr:  r<  )r>  rP  trolatol	equal_nans        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  ptraininplaces       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rn  keepdimr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r{  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  msgs     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  tensorss    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  kernel_sizer  padding	ceil_modecount_include_pads         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rR  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  )	r>  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )grad_outr>  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r>  r  r  r  input_gweight_gbias_gs           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  r  counts           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  	generatorr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  input1input2r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   r  c                      yr:  r<  r>  targetr  size_averagereduce	reduction
pos_weights          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  weights	minlengths      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  probr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r`  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  selfrj  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  
boundaries	out_int32rightr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  rn  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )x1x2ru  compute_modes       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3         ?c                      yr:  r<  r>  rU  rw  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   )r?  c                      yr:  r<  )r?  matricess     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  groupss     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  upperr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  check_errorsr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  numeln_binsratio	bit_widths        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  chunksrn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  minmaxr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  r?  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rA  r3   c                      yr:  r<  )r>  
correctionfweightsaweightss       r0   r@  z'get_testing_overrides.<locals>.<lambda>
  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rwith_replacements      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )realimags     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )absangs     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  ords     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  padr[  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  dilationr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  )	r>  r  r  r  r  r  
transposedoutput_addingr  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  output_paddingr  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r%  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r%  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  r  r  marginr  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>!  rA  r3   c                      yr:  r<  )r  r  rn  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>"  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>#  rA  r3   c                      yr:  r<  r>  rP  rn  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>$  rA  r3   r;  c                      yr:  r<  r2  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>%  rA  r3   c                      yr:  r<  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r0   r@  z'get_testing_overrides.<locals>.<lambda>'  rA  r3   c                      yr:  r<  r>  rn  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rA  r3   c                      yr:  r<  r=  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>*  rA  r3   c                      yr:  r<  r>  rn  r?  r`   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>+  rA  r3   c                      yr:  r<  r@  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rA  r3   c                      yr:  r<  yxrn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>-  rA  r3   c                      yr:  r<  r=  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>/  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>0  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rA  r3   c                      yr:  r<  r>  diagonalr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rA  r3   c                      yr:  r<  rM  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rA  r3   c                      yr:  r<  )r>  offsets     r0   r@  z'get_testing_overrides.<locals>.<lambda>6  rA  r3   c                      yr:  r<  )r>  nrn  prependappendr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>7  rA  r3   c                      yr:  r<  r>  rQ  dim1dim2s       r0   r@  z'get_testing_overrides.<locals>.<lambda>8  rA  r3   c                      yr:  r<  rW  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>9  rA  r3   c                      yr:  r<  )r>  srcrQ  rX  rY  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>:  rA  r3   c                      yr:  r<  )r  r\  rj  r  storage_offsets        r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda><  rA  r3   c                      yr:  r<  )r>  rP  ru  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rA  r3   c                      yr:  r<  r>  rP  rounding_moder?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>>  rA  r3   c                      yr:  r<  rb  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rA  r3   c                      yr:  r<  rt  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>A  rA  r3   c                      yr:  r<  r>  r`  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rA  r3   c                      yr:  r<  )r_  r`  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>C  rA  r3   c                      yr:  r<  r>  indices_or_sectionss     r0   r@  z'get_testing_overrides.<locals>.<lambda>D  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rA  r3   c                      yr:  r<  r>  UPLOr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rA  r3   c                      yr:  r<  rq  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>I  rA  r3   c                      yr:  r<  )equationoperandss     r0   r@  z'get_testing_overrides.<locals>.<lambda>J  rA  r3   c                      yr:  r<  r>  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r0   r@  z'get_testing_overrides.<locals>.<lambda>L  rA  r3   c
                      yr:  r<  )
r>  r  offsetsrz  r{  r|  moder   per_sample_weightsry  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rA  r3   c                      yr:  r<  r>  r`   rn   r_   requires_grads        r0   r@  z'get_testing_overrides.<locals>.<lambda>Q  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>R  rA  r3   c                      yr:  r<  r>  rP  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>S  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>T  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>U  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>W  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>X  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rA  r3   c                      yr:  r<  )r>  scale
zero_pointaxis	quant_min	quant_maxs         r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  rA  r3   c                      yr:  r<  )r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rA  r3   c                      yr:  r<  )rE  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rA  r3   c                      yr:  r<  r>  packed_weightr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>`  rA  r3   c                      yr:  r<  r>  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>c  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>e  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>f  rA  r3   c                      yr:  r<  )r>  abs      r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rA  r3   c                      yr:  r<  r>  ru  rv  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>h  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>i  rA  r3   c                      yr:  r<  r>  rS  rn  norms       r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>k  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>m  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>n  rA  r3   c                      yr:  r<  r>  srn  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>o  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>p  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>q  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>s  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>t  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>v  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>x  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>y  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>{  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>|  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rA  r3   c                      yr:  r<  )r>  	start_dimend_dims      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  dimss     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r{  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  exponentr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  
fill_valuer?  r`   rn   r_   r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  	dep_tokens      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  indexr?  sparse_grads        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rf  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  spacingrn  
edge_orders       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  gridinterpolation_modepadding_moderk  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  
num_groupsr  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  	r>  hxparams
has_biases
num_layersdropoutrv  bidirectionalbatch_firsts	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  w_ihw_hhb_ihb_hhs         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  lambds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  valuesr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r-  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  binsr  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  densityr?  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r   r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  taus     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r_  r`  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rk  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rn  r  sources       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r$  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  indicesr  
accumulates       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  r  r[  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  r  r%  r  include_inputs         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rZ   s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )eteassume_uniqueinverts       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r/  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r/  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  )	r>  r  r  r  r  use_input_statsr  r  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rP  rtolrq  rr  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c
                      yr:  r<  )
r>  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  
log_targets         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  krn  r|  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  	hermitianr  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r\  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )LDpivotsBr\  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  normalized_shaper  r  espr  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  endr  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rZ  ra  XrS  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rn  r`   s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rE  rD  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  namesr|  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  )	databatch_sizesr  r  r   r  r  rv  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )Apivot	get_infosr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r,  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  maskr[  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r%  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )LUr`  ra  leftadjointr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rS  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rS  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rp  r\  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  return_indicesr  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  r|  r`   r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                       yr:  r<  )r  r/   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  exponential_average_factorepsilons           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  	r>  r  r  r  r  r  r  	benchmarkdeterministics	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  )	r>  r  zrU  r  r  r  r  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c
                      yr:  r<  )
r>  r  r  r  r&  r  r  r  r  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c	                      yr:  r<  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  weight_stride0r  cxr  hidden_sizer  r  r  rv  r  r  dropout_states                 r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>"  rA  r3   c                      yr:  r<  r{  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>#  rA  r3   c                      yr:  r<  r>  r%  destinations      r0   r@  z'get_testing_overrides.<locals>.<lambda>$  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>%  rA  r3   c                      yr:  r<  )r>  
descendingr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>&  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>'  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>(  rA  r3   c                      yr:  r<  )r>  num_samplesreplacementr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rA  r3   c                      yr:  r<  )r>  rc  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>*  rA  r3   c                      yr:  r<  r>  ru  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>+  rA  r3   c                      yr:  r<  )r>  rn  startrS  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rA  r3   c                      yr:  r<  )r>  nanposinfneginfr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>-  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>/  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>0  rA  r3   c                      yr:  r<  r>  rc  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rA  r3   c                      yr:  r<  )r>  r  r  NCHxWgroupr  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rA  r3   c                      yr:  r<  )r>  ru  rn  r|  r`   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>6  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>7  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>8  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>9  rA  r3   c                      yr:  r<  rD  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>:  rA  r3   c                      yr:  r<  rD  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rA  r3   c                      yr:  r<  r>  rE  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda><  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>A  rA  r3   c                      yr:  r<  rh  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rA  r3   c                      yr:  r<  r>  ru  r  rw  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>C  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  divisor_overrides          r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rA  r3   c                      yr:  r<  r   s          r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>K  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>M  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>R  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>T  rA  r3   c                      yr:  r<  r,  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rA  r3   c                      yr:  r<  )r>  r  r  r  ignore_indexr  r  label_smoothings           r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rA  r3   c                      yr:  r<  r5  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>`  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>b  rA  r3   c                      yr:  r<  rx  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>d  rA  r3   c                      yr:  r<  )r>  r  r~  rz  r{  r|  r  r   r  include_last_offsetry  s              r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>i  rA  r3   c                      yr:  r<  )r>  rE  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rA  r3   c                      yr:  r<  r>  r  rE  output_ratior  _random_sampless         r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>o  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rA  r3   c                      yr:  r<  )r>  r  varr   r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rA  r3   c                      yr:  r<  )r>  approximates     r0   r@  z'get_testing_overrides.<locals>.<lambda>x  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>y  rA  r3   c                      yr:  r<  )r>  r  r  r  rk  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rA  r3   c                      yr:  r<  )r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>{  rA  r3   c                      yr:  r<  )logitsr  hardr  rn  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>|  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rA  r3   c                      yr:  r<  )r>  min_valmax_valrw  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r:  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rj  scale_factorr  rk  recompute_scale_factor	antialiass          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rV  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  negative_sloperw  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rj  rU  rV  rZ  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rn  _stacklevelr`   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r{  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r@  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r@  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r,  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rE  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r(  r  r  r  rE  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rL  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rL  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r5  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )querykeyr[  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  ru  r-  r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  ru  rn  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rZ   num_classess     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r[  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  r  ru  r  r|  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  	log_inputr   r  r  r  r  s           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rw  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  rc  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  lowerr  r  rw  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rQ  rR  r[  r_  rZ  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  rV  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  deltar  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rj  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r<  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r<  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rV  	thresholds      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  r[  rw  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c
                      yr:  r<  
anchorpositivenegativer-  ru  r  swapr  r  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   )distance_functionr-  r  r  c                     yr:  r<  )r  r  r  r  r-  r  r  s          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rZ   r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rZ   r  stdr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rZ   vals     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rZ   r  r  nonlinearityr  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  as_tuples     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   )r  c                     yr:  r<  )r>  rj  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  ru  rn  r|  r?  r`   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  rn  r|  r?  r`   s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      	 yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s     13 r3   c                      yr:  r<  )vpowrn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  input3r  	transposes        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rr  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  rn  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  qrP  ro  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rconds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r\  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  upscale_factors     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  downscale_factors     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  rt  r   r  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rv  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r`   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r%  r)  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  somer?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  r  rn  r|  interpolationr?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  scaleszero_pointsr  r`   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rA  r3   c                      yr:  r<  )r>  r  r  r`   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>
  rA  r3   c                      yr:  r<  )r>  r`   reduce_ranges      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  r   r  output_scaleoutput_zero_points           r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  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                 r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   r      c                      	 yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s
     " r3   c                      	 yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s     !# r3   c                      	 yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  s     !#	 r3   c                      yr:  r<  r  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>&  rA  r3   c                      yr:  r<  r  s                 r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>+  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rA  r3   c                      yr:  r<  )r>  highr`   rn   r_   r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>-  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>/  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>0  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rA  r3   c                      yr:  r<  r2  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>6  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>7  rA  r3   c                      yr:  r<  )r>  ru  rn  maxnormr?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>8  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>9  rA  r3   c                      yr:  r<  )r>  shapes     r0   r@  z'get_testing_overrides.<locals>.<lambda>:  rA  r3   c                      yr:  r<  r{  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rA  r3   c	                      yr:  r<  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda><  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rA  r3   c	                      yr:  r<  r  s	            r0   r@  z'get_testing_overrides.<locals>.<lambda>>  rA  r3   c                      yr:  r<  r  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rA  r3   c                      yr:  r<  )r>  shiftsr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rA  r3   r   r  c                      yr:  r<  )r>  rZ  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>A  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>C  rA  r3   c                      yr:  r<  )r  r  compressed_indices_dtypes      r0   r@  z'get_testing_overrides.<locals>.<lambda>D  rA  r3   c                      yr:  r<  r}  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rA  r3   c                      yr:  r<  )r>  rP  rU  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rA  r3   c                      yr:  r<  r^  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rA  r3   c                      yr:  r<  r>  rn  r  r\  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>I  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>J  rA  r3   c                      yr:  r<  )r>  rn  r  r\  r  include_selfs         r0   r@  z'get_testing_overrides.<locals>.<lambda>K  rA  r3   c                      yr:  r<  )sorted_sequencer>  r  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>L  rA  r3   c                      yr:  r<  )r  r  lengthsr(  r~  r  unsafes          r0   r@  z'get_testing_overrides.<locals>.<lambda>M  rA  r3   c                      yr:  r<  )r>  rn  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>N  rA  r3   c                      yr:  r<  )r>  r\  rn  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rA  r3   c                      yr:  r<  r>  r\  rn  r  rj  steps         r0   r@  z'get_testing_overrides.<locals>.<lambda>P  rA  r3   c                      yr:  r<  r"  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>Q  rA  r3   c                      yr:  r<  rx  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>R  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>S  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>T  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>U  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>W  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>X  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rA  r3   c                      yr:  r<  rh  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rA  r3   c                      yr:  r<  rh  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rA  r3   c                      yr:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rA  r3   c                      yr:  r<  )r  ra  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r3   c                      yr:  r<  )r  ra  r  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>`  rA  r3   )stabler?  c                     yr:  r<  )r>  rn  r  r4  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rA  r3   c                      yr:  r<  rZ   split_size_or_sectionsrn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>b  rA  r3   c                      yr:  r<  r7  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>c  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>d  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>e  rA  r3   c                      yr:  r<  r=  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>f  rA  r3   c                      yr:  r<  r^  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>h  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>i  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rA  r3   c
                      yr:  r<  )
r>  rL  rM  rN  rO  rP  pad_moderQ  rR  rT  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>n  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>o  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>p  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>q  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rA  r3   c                      yr:  r<  r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>s  rA  r3   c                      yr:  r<  rI  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>t  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rA  r3   c                      yr:  r<  )r  r  cs      r0   r@  z'get_testing_overrides.<locals>.<lambda>v  rA  r3   c                      yr:  r<  )r.   s    r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>x  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>y  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>{  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>|  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  
compute_uvr?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  ro  Ms       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  full_matricesr?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r>  dim0rX  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  axis0axis1s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r{  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  ry  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rO  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rP  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r(  rn  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  inds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  r?  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rl  rn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rZ  rn  r  r?  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rJ  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rb  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rC  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rC  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  r  unitriangulars        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  ra  r  r  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rM  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c
                      yr:  r<  r  s
             r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rM  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  rn  sizesr  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  sortedreturn_inversereturn_countsrn  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  r  r  rn  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r(  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r7  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r7  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r=  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )rE  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rm  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  rk  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )	conditionrE  rD  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r>  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  levels     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )primaltangentr  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rj  r  r^  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rQ  rX  rY  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   )implicitc                     yr:  r<  )r  rj  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rn  r  rS  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  rj  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rn  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rn  r  rj  r#  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  
split_sizern  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  split_sizesrn  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rc  rX  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>	  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>
  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  r`   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  	dimensionrj  r#  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  rP  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>   rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>!  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>"  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>#  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>$  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>%  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>&  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>'  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>(  rA  r3   c                      yr:  r<  )r  arrays     r0   r@  z'get_testing_overrides.<locals>.<lambda>)  rA  r3   c                      yr:  r<  )r  idxs     r0   r@  z'get_testing_overrides.<locals>.<lambda>*  rA  r3   c                      yr:  r<  )r  memos     r0   r@  z'get_testing_overrides.<locals>.<lambda>+  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>,  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>-  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>.  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>/  rA  r3   c                      yr:  r<  )r  format_specs     r0   r@  z'get_testing_overrides.<locals>.<lambda>0  rA  r3   c                      yr:  r<  )r  protos     r0   r@  z'get_testing_overrides.<locals>.<lambda>1  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>2  rA  r3   )tensor_contentsc                     yr:  r<  )r  r+  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>3  rA  r3   c                      yr:  r<  )r  rZ  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>4  rA  r3   c                      yr:  r<  )r  ds     r0   r@  z'get_testing_overrides.<locals>.<lambda>5  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>6  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>7  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>8  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>9  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>:  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>;  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda><  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>=  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>?  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>@  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>A  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>B  rA  r3   c                      yr:  r<  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r0   r@  z'get_testing_overrides.<locals>.<lambda>C  rA  r3   c                      yr:  r<  )r  r>  r?  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>D  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>E  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>F  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>G  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>H  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>I  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>J  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>K  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>L  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>M  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>N  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>O  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>P  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Q  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>R  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>S  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>T  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>U  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>V  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>W  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>X  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Y  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>Z  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>[  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>\  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>]  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>^  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>_  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>`  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>a  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>b  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>c  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>d  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>e  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>f  rA  r3   c                      yr:  r<  )r  r`   non_blockingr/   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>g  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>h  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>i  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>j  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>k  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>l  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>m  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>n  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>o  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>p  rA  r3   c                      yr:  r<  )r  r  rw  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>q  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>r  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>s  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>t  rA  r3   c                      yr:  r<  )r  orderellipsis_idxs      r0   r@  z'get_testing_overrides.<locals>.<lambda>u  rA  r3   c                      yr:  r<  )r  callables     r0   r@  z'get_testing_overrides.<locals>.<lambda>v  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>w  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>x  rA  r3   c                      yr:  r<  )r  gradientretain_graphcreate_graphrG  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>y  rA  r3   c                      yr:  r<  r  rg   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>z  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>{  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>|  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>}  rA  r3   )r  c                     yr:  r<  )r  mediansigmar  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>~  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  	coalesceds     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r\  rf  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r\  rQ  rX  rY  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  ambiguity_checks     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     yr:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  r[  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     yr:  r<  )r  ru  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rZ   s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     yr:  r<  )r  r  r  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rZ   rx  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rE  rD  rx  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r`  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  rP  assigns      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r   r  rS  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r(  rZ   r)  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                     yr:  r<  )r  from_tor  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  streams     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  hooks     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  names     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r%  r^  rj  r  s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r\  rn  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r\  rn  r  rj  r#  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  accumulate_matchess      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  size1size2	dense_dims       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r_  r`  rV  rU  r?  s         r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  repss     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r`   rf  copyrg   s        r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   )masked_gradc                     yr:  r<  r  r`   r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r  s      r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s     r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r  r   drivers       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  )r  r_   rf  r/   s       r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   c                      yr:  r<  r  s    r0   r@  z'get_testing_overrides.<locals>.<lambda>  rA  r3   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  Fr:  )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  N)g{Gz?F)g-C6?g      ?r  )N   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  )NNr  Nr  )r  r  g-q=N)r   r   )r  r  F)TFNr  Nr  )Nr  )g      ?gUUUUUU?FF)Nr  )NNr  r  )r  r  N)r     )r  r  r  FNNr  )r  r  N)r   fan_in
leaky_reluN)froNFNN)NNFNN)r  NFNN)r#  r  FNN)r  r   )TF)NTr  )V瞯<)r$  F)reducedN)NFlinearN)r<  r  r  F)r<  )r   r   )r  r  F)r<  )r   r   r   )r  r  r  F)r  r  )r  NNNr   F)r   NNr  )r;  F)NNNTreflectFTN)TTN)   r  N)r  N)TFF)TFFN)r  r  )Nr   NN(  r5   r6   r  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshaddaddbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullir   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffrN  diagonal_scatteras_strided_scatterdigammadistdivdividedotr  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdge
get_devicegreater_equalgeqrfi0inneroutergerr|  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrink	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_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r   var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr'  HmTmH_backward_hooks_post_accumulate_grad_hooksr4  _cdatar5  r6  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  r_   r`   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkan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  rn  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesr(  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_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_shortrj  
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extendrx  update)r6   retprivateuse1_backend_nameret2ignoredrZ  r  r  subnamer  r!   s              r0   r   r     sW   6 \\Fz%		-z%2z% 	!!#@z% 	!!#A	z%
 	

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

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

.+z%, 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

.z%@ 	=Az%B 	>Cz%D 	LEz%F 	BGz%H 	=Iz%J 	\Kz%L 	)Mz%N 	

GOz%P 	&Qz%R 	'Sz%T 	2Uz%V 	2Wz%X 	t[z%^ 	(_z%` 	1az%b 	4cz%d 	Kez%f 	*gz%h 	'iz%j 	&kz%l 	.mz%n 	,oz%p 	!1qz%r 	*sz%t 	3uz%v 	)wz%x 	Wyz%z 	%{z%| 	 dz%B	 	rC	z%D	 	

+E	z%F	 	NG	z%H	 	""$cI	z%J	 	!LK	z%L	 	 SM	z%N	 	sO	z%P	 			4Q	z%R	 	6S	z%T	 	3U	z%V	 	;W	z%X	 	

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

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

s}	z%~	 	O	z%@
 	3A
z%B
 	

5C
z%D
 	EE
z%F
 	BG
z%H
 	!!  $AI
z%J
 	8K
z%L
 	<M
z%N
 	=O
z%P
 	7Q
z%R
 	?S
z%T
 	 FU
z%V
 	!!#]W
z%X
 	[Y
z%Z
 	>[
z%\
 	/]
z%^
 	!!#@_
z%`
 	  "Ma
z%b
 	 <c
z%d
 	*e
z%f
 	!1g
z%h
 			-i
z%j
 	8k
z%l
 	

5m
z%n
 	lo
z%p
 	lq
z%r
 	ls
z%t
 	%%tw
z%z
 	

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

5Iz%J 	tMz%P 	  "}Qz%R 	))+vSz%T 	%%'hUz%V 	**wYz%\ 	**g_z%b 	 dez%h 	2iz%j 	

Ekz%l 	<mz%n 	=oz%p 	Aqz%r 			4sz%t 	9uz%v 	Uwz%x 	1yz%z 	+{z%| 	:}z%~ 	Wz%@ 	!sAz%B 	&&(_Cz%D 	8Ez%F 	!fGz%H 	!VIz%J 	UKz%L 	$$&>Mz%N 	3Oz%P 	:Qz%R 			-Sz%T 	2Uz%V 	:Wz%X 	//1NYz%Z 	//1N[z%\ 	//1d]z%^ 	<<>q_z%` 	//1daz%b 	<<>qcz%d 	//1dez%f 	<<>qgz%h 	'')Siz%j 	))+akz%l 	&& Boz%r 	&& Buz%x 	&&x{z%~ 	$$&Rz%@ 	00cCz%F 	<<tIz%L 	  "LMz%N 	11iQz%T 	)) LWz%Z 	$$x]z%` 	##%Zaz%b 	%%'\cz%d 	%%'\ez%f 	%%'\gz%h 	!Kiz%j 	%%|mz%p 	)) Jsz%v 	113iwz%x 	  "myz%z 	11z}z%@ 	>>zCz%F 	11zIz%L 	>>zOz%R 	--/uSz%T 	  "FUz%V 	!9Wz%X 	'')zYz%Z 	&&(g[z%\ 	**,c]z%^ 	&&(C_z%` 	$$&`az%b 	00bez%h 	)) Ikz%n 	'' Mqz%t 	""  %Auz%v 	##%|wz%x 	&&(myz%z 	&&(\{z%| 	""$G}z%~ 	//1gz%@ 	'')^Az%B 	&&(8Cz%D 	%%'mEz%F 	%%'mGz%H 	%%'mIz%J 	//iMz%P 	&&tSz%V 	33tYz%\ 	&&t_z%b 	33tez%h 	&&tkz%n 	33tqz%t 	((*zuz%v 	((*zwz%x 	((*zyz%z 	$$&}{z%| 	88 _z%B 	--tEz%H 	22VKz%N 	77cQz%T 	$$vWz%Z 	%%'X[z%\ 	##%F]z%^ 	!P_z%` 	--/aaz%b 	,,}ez%h 	!!#;iz%j 	  "Akz%l 	!!#Bmz%n 	$$&_oz%p 	!!#yqz%r 	  "Asz%t 	  "Auz%v 	  "Awz%x 	88:uyz%z 	**  -A{z%| 	&&(j}z%~ 	,,.xz%@ 	##%ZAz%B 	##%ZCz%D 	$$&LEz%F 	&&(CGz%H 	$$&6Iz%J 	&&(8Kz%L 	%%'XMz%N 	// LQz%T 	==DHQT[`lrvWz%Z 	""$b[z%\ 	 O]z%^ 	S_z%` 	!7az%b 	&&(xcz%d 	7ez%f 	Fgz%h 	(iz%j 	

\kz%l 	dmz%n 	  "hoz%p 	   +/#3qz%x 	9yz%z 	d{z%| 	%}z%~ 	*z%@ 	QAz%B 	!SCz%D 	+Ez%F 	IGz%H 	*Iz%J 	5Kz%L 	IMz%N 	=Oz%P 	AQz%R 	7Sz%T 	 YUz%V 	6Wz%X 	2Yz%Z 	-[z%\ 	d]z%^ 			7_z%` 	

0az%b 			Dcz%d 	  "2ez%f 	""$4gz%h 	'')9iz%j 	'kz%l 	,mz%n 	7oz%p 	Cqz%r 	fsz%t 	iuz%v 	""$Vwz%x 	!!#Myz%z 	))+P{z%| 	""$s}z%~ 	   aAz%D 	!! aGz%J 	""J#Mz%T 	""L #Wz%` 	""O 	#cz%n 	%% aqz%t 	%% awz%z 	1{z%| 	d}z%~ 	d5==Y]mrvz%@ 	eAz%B 	%Cz%D 	

.Ez%F 	

5Gz%H 	FIz%J 	,Kz%L 	/Mz%N 	4Oz%P 	

3Qz%R 	:Sz%T 	AUz%V 	!;Wz%X 	.Yz%Z 	Q[z%\ 	x]z%^ 	S_z%` 	xaz%b 	Scz%d 	

7ez%f 	7gz%h 	/iz%j 	5kz%l 	Pmz%n 	boz%p 	/qz%r 	

4sz%t 	Muz%v 	8wz%x 	<yz%z 	Z{z%| 	e}z%~ 	|z%@ 	2Az%B 	?Cz%D 	WEz%F 	WGz%H 	

3Iz%J 	1Kz%L 	

.Mz%N 	1Oz%P 			-Qz%R 			-Sz%T 	

.Uz%V 	

.Wz%X 	'Yz%Z 	.[z%\ 			)]z%^ 	

*_z%` 	8az%b 	@cz%d 	Wez%f 	

YeQUYgz%h 	Eiz%j 	 Pkz%l 	

.mz%n 	0oz%p 	;qz%r 	Osz%t 	8uz%v 			-wz%x 	2yz%z 	

 j}z%@ 			4Az%B 	9Cz%D 			-Ez%F 	)Gz%H 	'Iz%J 	Kz%L 	Mz%N 	'Oz%P 	)Qz%R 	Sz%T 	)Uz%V 	(Wz%X 	)Yz%Z 	([z%\ 	)]z%^ 	(_z%` 	)az%b 	)cz%d 	)ez%f 	)gz%h 	0iz%j 			Ikz%l 	Amz%n 	Hoz%p 	8qz%r 	4sz%t 	6uz%v 	/wz%x 	!1yz%z 	!1{z%| 	!1}z%~ 	!1z%@ 	,,.KAz%B 	,,.KCz%D 	,,.KEz%F 	,,.KGz%H 	/Iz%J 	,Kz%L 	+Mz%N 	,Oz%P 	-Qz%R 	.Sz%T 	,Uz%V 	-Wz%X 	-Yz%Z 	 A[z%\ 	!B]z%^ 	/_z%` 	**,Iaz%b 	++-Jcz%d 	*ez%f 	+gz%h 	*iz%j 	+kz%l 	++-Jmz%n 	++-Joz%p 	-qz%r 	 0sz%t 	!!#Duz%v 	-wz%x 	!Oyz%z 	((*:{z%| 	((*:}z%~ 	((*:z%@ 	((*:Az%B 	""$7Cz%D 	,Ez%F 	-Gz%H 	!>Iz%J 	+Kz%L 	-Mz%N 	//1AOz%P 	//1AQz%R 	446SSz%T 	446SUz%V 	446SWz%X 	446SYz%Z 	,[z%\ 	@]z%^ 	))+;_z%` 	@az%b 	>cz%d 	<ez%f 	!gz%h 	

+iz%j 	Kkz%l 			-mz%n 	

.oz%p 	 3qz%r 	  "<sz%t 	:uz%v 	Hwz%x 	Jyz%z 	

*{z%| 	

K}z%~ 	%z%@ 	5Az%B 	1Cz%D 	5Ez%F 	 eGz%H 	%%'aIz%J 	

:Kz%L 	!! LOz%R 	

:Sz%T 	2Uz%V 	/Wz%X 	-Yz%Z 	<[z%\ 	h]z%^ 	  "g_z%` 	6az%b 	;cz%d 	Lez%f 	%%'Wgz%h 	8iz%j 	1kz%l 			-mz%n 	2oz%p 	;qz%r 	2sz%t 	9uz%v 	%%'`wz%x 	11o{z%~ 	ez%@ 	5Az%B 	@Cz%D 	Ez%F 	""OGz%H 	/Iz%J 	oKz%L 	QMz%N 	'')>Oz%P 	FQz%R 	%CSz%T 	>Uz%V 	1Wz%X 	!!#@Yz%Z 	6[z%\ 	?]z%^ 	N_z%` 	<az%b 	##%Hcz%d 	0ez%f 	ogz%h 	9iz%j 	2kz%l 	_mz%n 	Ooz%p 	Oqz%r 	?sz%t 	uz%v 	wz%x 	yz%z 	{z%| 	1}z%~ 	/z%@ 	AAz%B 	/Cz%D 	3Ez%F 	4Gz%H 	4Iz%J 	2Kz%L 	3Mz%N 	3Oz%P 	1Qz%R 	2Sz%T 	2Uz%V 	1Wz%X 	2Yz%Z 	2[z%\ 	.]z%^ 	-_z%` 	.az%b 	/cz%d 	Oez%f 	0gz%h 	iz%j 	3kz%l 	mz%n 	?oz%p 	.qz%r 	/sz%t 	/uz%v 	5wz%x 	0yz%z 	2{z%| 	}z%~ 	z%@ 	/Az%B 	Cz%D 	7Ez%F 	4Gz%H 	_Iz%J 	AKz%L 	1Mz%N 	/Oz%P 	/Qz%R 	/Sz%T 			?Uz%V 			?Wz%X 	&&Yz%Z 	**22O[z%\ 	o]z%^ 	_z%` 	_az%b 	ocz%d 	ez%f 	gz%h 	iz%j 	--/pkz%l 	**,Vmz%n 	22Ooz%p 	_qz%r 	sz%t 	ouz%v 	wz%x 	yz%z 	{z%| 	}z%~ 	z%@ 	Az%B 	##_Cz%D 	Ez%F 	Gz%H 	Iz%J 	  /Kz%L 	Mz%N 	  /Oz%P 	##_Qz%R 	  /Sz%T 	$$oUz%V 	  /Wz%X 	Yz%Z 	[z%\ 	_]z%^ 	o_z%` 	az%b 	_cz%d 	  /ez%f 	$$ogz%h 	oiz%j 	kz%l 	_mz%n 	_oz%p 	''//qz%r 	Nsz%t 	ouz%v 	owz%x 	yz%z 	{z%| 	_}z%~ 	_z%@ 	OAz%B 	_Cz%D 	OEz%F 	=Gz%H 	Iz%J 	Kz%L 	/Mz%N 	=Oz%P 	0Qz%R 	8Sz%T 	9Uz%V 	kWz%X 	E4I4IMYz%Z 	0E0EI[z%\ 	0E0EI]z%^ 	0E0EI_z%` 	MTMaz%b 	cz%d 	6ez%f 	e6M6MQgz%h 	>iz%j 	

u/D/DHkz%l 	0E0EImz%n 	0E0EIoz%p 	

u/D/DHqz%r 	

u/D/DHsz%t 	uz%v 	/wz%x 	!Oyz%z 	

O{z%| 	@}z%~ 	%2G2GKz%@ 	53H3HLAz%B 	_Cz%D 	,Ez%F 	0Gz%H 	HHIz%J 	,Kz%L 	5Mz%N 	1F1FJOz%P 	%2G2GKQz%R 	@Sz%T 	?Uz%V 	0E0EIWz%X 	1F1FJYz%Z 	/[z%\ 	]z%^ 	

u/D/DH_z%` 	_az%b 	ocz%d 	_ez%f 	/gz%h 	1iz%j 	/kz%l 	_mz%n 	MTMoz%p 	0qz%r 	0E0EIsz%t 	6uz%v 	5wz%x 			(yz%z 	@{z%| 	E}z%~ 	?z%@  	A z%B  	""OC z%D  	--E z%F  	%%G z%H  	I z%J  	oK z%L  	,M z%N  	?O z%P  	GQ z%R  	S z%T  	LDLU z%V  	5W z%X  	3Y z%Z  	3[ z%\  	113H] z%^  	,_ z%`  	-a z%b  	Bc z%d  	1e z%f  	-g z%h  	-i z%j  	0k z%l  	  "8m z%n  	Oo z%p  	[q z%r  	?s z%t  	ou z%v  	1F1FJw z%x  	_y z%z  	W{ z%|  	?} z%~  	1 z%@! 	&&(WA!z%B! 	GC!z%D! 	'')QE!z%F! 	OG!z%H! 	I!z%J! 	K!z%L! 	M!z%N! 	_O!z%P! 	1Q!z%R! 	+S!z%T! 			EUZUjUjnU!z%V! 	IIW!z%X! 	GY!z%Z! 	/[!z%\! 	]!z%^! 	/_!z%`! 	.a!z%b! 	=c!z%d! 	7e!z%f! 	+.o7  /Fs!z%Cz! 	((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Jr3   
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_argsrw
  r!   wrappeds      r0   r|
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped*  sA    &77M!-0,WmUdUfUU(((r3   )	functoolsr   )r!   r|
  rw
  s   `@r0   r  z"wrap_torch_function.<locals>.inner)  s%    			) 
	) r3   r<  )rw
  r  s   ` r0   r   r     s    0	 Lr3   r{
  get_type_fnc                    |t         }t        j                  j                         sg S t	               }g }| D ]  } ||      }||vst        |d      s|j                  t        j                  j                  k7  sE|rX|j                  |       t        |      }t        |      D ]  \  }}t        | ||            s|} n |j                  ||       |h}|g} |S )a
  Returns a list of arguments on which to call __torch_function__.

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

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

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

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

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

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

    .. _NEP-0018:
       https://numpy.org/neps/nep-0018-array-function-protocol.html
    r   )r	  r5   _C_is_torch_function_enabledsetrj
  r   _disabled_torch_function_implr0  ro
  	enumerate
issubclassinsert)	r{
  r~
  overloaded_typesoverloaded_argsargarg_typer  iold_args	            r0   _get_overloaded_argsr
  7  s    J  88..0	"%%!#O (s# ,,"67++uxx/U/UU   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%9(: r3   
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   NotImplementedrj
  r
  r5   r
  r
  r*   warnDeprecationWarning
__module__rm
  _get_current_function_mode	TypeError)r
  r{
  r.   r/   r
  typesr  resultoverloaded_argtorch_func_method	func_namer
  r  s                r0   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 )Nr5   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listr5   __all__r   r   dirr   r6   rt  r   rF	  rn
  endswithislowerrk
  object
isinstancer
  
ModuleType
__future___Featurerx  rj
  r'  __set__r   r   formatrm
  rU  )
overridable_funcsr  tested_namespacesnamespace_str	namespacens_funcsr
  r'   r!   r  s
             r0   _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##r3   c                      t               d   S )a  List functions that are overridable via __torch_function__

    Returns
    -------
    Dict[Any, List[Callable]]
        A dictionary that maps namespaces that contain overridable functions
        to functions in that namespace that can be overridden.
    r   )r
  r<  r3   r0   r   r   e  s     &'**r3   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
  r5   _ops
OpOverloadOpOverloadPacketstrr
  get)fs    r0   r   r   r  sI      !ejj++UZZ-H-HIJ1v%'*..q11r3   c                  R    t               } t        | t        j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r
  r5   r6   )r
  methodss     r0   _get_tensor_methodsr
    s&     23#ELL12GNr3   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
  rm
  )r!   s    r0   r   r     s!    . &((FDMMY,FFr3   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	  r5   r6   rj
  )inps    r0   r   r     s%    D 9$J5I(JJr3   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    r0   r   zTorchFunctionMode.__init__  s    r3   c                     t         r  )NotImplementedErrorr  r!   r
  r.   r/   s        r0   r   z$TorchFunctionMode.__torch_function__  s    !!r3   c                     t        |        | S r  )
_push_moder  s    r0   	__enter__zTorchFunctionMode.__enter__  s    4r3   c                     t                y r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r0   __exit__zTorchFunctionMode.__exit__  s    r3   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       r0   pushzTorchFunctionMode.push  s'    ^	
 ''r3   )r$   Nr<  N)rm
  r
  __qualname____doc____annotations__r   r   r
  r
  classmethodr
  r<  r3   r0   r
  r
    s7    < "  r3   r
  c                  @    t               } | dkD  rt        | dz
        S d S )Nr   r  )r   r   )	stack_lens    r0   r
  r
    s%    )+I4=M!)a-0KtKr3   c                  d    t               } t        |       D cg c]  }t        |       c}S c c}w r  )r   r   r   )r
  r
  s     r0    _get_current_function_mode_stackr
    s)    )+I/4Y/?@!"1%@@@s   -c                     t        |        y r  )r   )r  s    r0   r
  r
    s
    !$'r3   c                      t               } | S r  )r   olds    r0   r
  r
    s    
#
%CJr3   c               #   `   K   t               } 	 |  t        |        y # t        |        w xY wwr  )r
  r
  r
  s    r0   r
  r
    s%     
+C	3
3s   . .+.c                       e Zd ZddZy)BaseTorchFunctionModeNc                     |i } ||i |S r  r<  r
  s        r0   r   z(BaseTorchFunctionMode.__torch_function__#  s    >FT$V$$r3   r
  )rm
  r
  r
  r   r<  r3   r0   r
  r
  "  s    %r3   r
  c               #   Z  K   t         j                  j                         } 	 t         j                  j                  t         j                  j                  j
                         d  t         j                  j                  |        y # t         j                  j                  |        w xY wwr  )r5   r
  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r0   _enable_torch_functionr
  )  sf     224I6**588+G+G+O+OP**95**95s   B+AB ' B+!B((B+c               #      K   t         j                  j                         5  	 d  	 d d d        y # w xY w# 1 sw Y   y xY wwr  )r5   r
  _RestorePythonTLSSnapshotr<  r3   r0   r    r    3  sC      
	+	+	- 		  	 s#   A61	A36?A)z.*is deprecated, please use.*r5   r  );r
  r
  r
  
contextlibr}
  r
  r*   r   typingr   r   r   r   r	   r
   r   r   r   r5   torch._Cr   r   r   r   r   r   r   r   r   r
  r
  r2   	lru_cacher   r7  r   r   r
  r   r   r   r   r
  r   r   r
  r
  r   r   r
  r
  r
  r
  r
  contextmanagerr
  r
  r
  r    r<  r3   r0   <module>r
     s,  ,        R R R 
 
 
  1 
     	 F TSs8} S  Sl Tc(m  2 TztHh$67 z  zz##H #P 48KC=K(C5$;/0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
(
  %- % 6 6  r3   