
    9jZ                    4   d Z ddlZddlZddlZddlZddlZddlZddlZddlm	Z	m
Z
 ddlmZ ddlmZmZmZ ddlmZ ddlZddlmZmZmZmZmZmZmZmZmZ g dZ ed	      Z ed
      Z	 	 d0de	eef   de de de	eef   fdZ!ejD                  e!de#e	   fd              Z$ejD                  de#e	   fd       Z%ejD                  e!de&e	e	f   fd              Z'de	ee
e   f   de	e	eef   ge	eef   f   fdZ(	 d1de
e   de	ege)f   dz  de*e   fdZ+de	eef   de
e   dejX                  dejZ                  def
dZ. eed      Z/ eed      Z0 eed      Z1ejD                  de2e&ee*e	   f   e&e	e f   f   fd       Z3e!de&ee*e	   f   fd        Z4e!d!        Z5ejD                  de#e	   fd"       Z6e!de	de7fd#       Z8d$ Z9 G d% d&      Z:d' Z;d( Z<d) Z=d* Z>ej~                  d+        Z@ G d, d-e:      ZAej~                  d.        ZBej~                  d/        ZCy)2aE  
Python implementation of ``__torch_function__``

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

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

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

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

    Returns
    -------
    function
        The wrapped function.
    argskwargsr"   c                      t        j                         5  t        j                  dt                | i |cd d d        S # 1 sw Y   y xY w)Nignore)categorymessager!   )warningscatch_warningsfilterwarningsUserWarning)r$   r%   r   r!   r    s     O/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapper[   sJ    $$& 	)##;f ((		) 	) 	)s   %AA)r   r   r$   r%   r   )r   r    r!   r/   s   ``` r.   _disable_user_warningsr0   C   s?    0 4[)rww )")) ) ) ) N    c                  %   t         j                  } h t         j                  t         j                  t         j                  t         j
                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                   t         j"                  t         j$                  t         j&                  t         j(                  t         j*                  t         j,                  t         j.                  t         j0                  t         j2                  t         j4                  t         j6                  t         j8                  t         j:                  t         j<                  t         j>                  t         j@                  t         jB                  t         jD                  t         jF                  t         jH                  t         jJ                  t         jL                  t         jN                  t         jP                  t         jR                  t         jT                  t         jV                  t         jX                  t         jZ                  t         j\                  t         j^                  t         j`                  t         jb                  t         jd                  t         jf                  t         jh                  t         jj                  t         jl                  t         jn                  t         jp                  t         jr                  t         jt                  t         jv                  t         jx                  t         jz                  t         j|                  t         j~                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  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                  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$                  j0                  t         j                   j$                  j2                  t         j                   j$                  j4                  t         j                   j$                  j6                  t         j                   j$                  j8                  t         j:                  j<                  t        t        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         j                   j                  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                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  j                  | j                  j                  | j                  | j                  | j                  | j                  | j                   | j                  | j                  | j                  | j<                  | j                  | j
                  | j                  }t        j                  dk\  r|j                  | 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thread_safe_generatorsaveloadset_printoptionsforkget_default_dtypeget_num_interop_threadsget_num_threadsinit_num_threadsimport_ir_moduleimport_ir_module_from_bufferis_anomaly_enabledis_anomaly_check_nan_enabledis_grad_enabledmerge_type_from_type_commentparse_irparse_schemaparse_type_commentset_anomaly_enabledset_flush_denormalset_num_interop_threadsset_num_threadswait	as_tensor
from_numpytensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutalign_tensorsarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptablemiopen_ctc_lossemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrand	rand_likerandn
randn_likerandintrandint_likerandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   
grouped_mmscaled_grouped_mm	scaled_mmsigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_dtensor__new___make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_count_philox_normal__philox_uniform_sysversion_infoadd__annotate__)r6   	functionss     r.   r   r   f   s   ( \\FHHH 	H 	%%	H
 	  H 	  H 	H 	H 	H 	H 	

H 	##H 	

H 	

H 	H  	

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

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

oHp 	

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

KHL 	MHN 	

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

IHJ 	

KHL 	MHN 	OHP 	--QHR 	SHT 	UHV 	WHX 	YHZ 	[H\ 	]H^ 	  _H` 	aHb 	cHd 	!!eHf 	!!gHh 	iHj 	kHl 	mHn 	oHp 	qHr 	

sHt 	uHv 	wHx 	

yHz 	{H| 	}H~ 	  H@ 	AHB 	CHD 	EHF 	GHH 	IHJ 	KHL 	MHN 	OHP 	QHR 	SHT 	UHV 	WHX 	YHZ 	[H\ 	]H^ 	_H` 	aHb 	cHd 	eHf 	""gHh 	iHj 	kHl 	  mHn 	$$oHp 	qHr 	%%sHt 	''uHv 	**22wHx 	yHz 	{H| 	}H~ 	H@ 	AHB 	CHD 	EHF 	  GHH 	IHJ 	KHL 	MHN 	OHIT 7"f))*r1   c                      t         j                  } | j                  j                  | j                  j                  | j
                  j                  hS )a  
    Return public functions that do not wrap in a subclass when invoked by
    the default ``Tensor.__torch_function__`` that preserves subclasses.  Typically,
    these functions represent field accesses (i.e., retrieving a Tensor that
    is stored somewhere on the Tensor) as opposed to computation.  Users of
    these functions expect object identity to be preserved over multiple accesses
    (e.g., ``a.grad is a.grad``) which cannot be upheld if we're wrapping on
    the fly every time (furthermore, the tensor stored here might already be
    the subclass, in which case wrapping really ought not to happen).

    Not ALL property accessors have this property; for example ``Tensor.T`` actually
    just creates a new transposed tensor on the fly, and so we SHOULD interpose on
    these calls (you need to check the implementation of the function to see if
    this is the case or not).  Additionally, if a property accessor doesn't return a Tensor,
    it doesn't have to be on this list (though it is harmless if it is).
    )r5   r6   _baser-  grad_grad)r6   s    r.   get_default_nowrap_functionsrC    s>    $ \\F r1   c                     t         j                  } i t         j                  ddt         j                  ddt         j                  d t         j
                  d t         j                  ddt         j                  d t         j                  ddt         j                  dd	t         j                  dd
t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                   ddt         j"                  ddt         j$                  d i t         j&                  ddt         j(                  ddt         j*                  ddt         j,                  ddt         j.                  ddt         j0                  ddt         j2                  ddt         j4                  ddt         j6                  d t         j8                  d t         j:                  ddt         j<                  ddt         j>                  d  t         j@                  dd!t         jB                  dd"t         jD                  dd#t         jF                  dd$i t         jH                  dd%t         jJ                  dd&t         jL                  dd't         jN                  dd(t         jP                  dd)t         jR                  d* t         jT                  d+ t         jV                  d, t         jX                  dd.t         jZ                  dd/t         j\                  d0 t         j^                  d1 t         j`                  d2 t         jb                  d3 t         jd                  d4 t         jf                  d5 t         jh                  d6 i t         jj                  d7 t         jl                  dd8t         jn                  d9 t         jp                  dd;t         jr                  dd<t         jt                  dd=t         jv                  dd>t         jx                  dd?t         jz                  dd@t         j|                  ddAt         j~                  ddBt         j                  ddCt         j                  dD t         j                  ddEt         j                  dF t         j                  dG t         j                  ddHi t         j                  dI t         j                  ddJt         j                  ddKt         j                  ddLt         j                  ddMt         j                  ddNt         j                  ddPt         j                  ddQdRt         j                  dS t         j                  ddTt         j                  j                  ddUt         j                  j                  ddVt         j                  ddWt         j                  ddXt         j                  dY t         j                  ddZt         j                  dd[i t         j                  dd\t         j                  dd]t         j                  dd^t         j                  dd_t         j                  dd`t         j                  da t         j                  ddbt         j                  dc t         j                  dddt         j                  de t         j                  j                  ddft         j                  ddgt         j                  ddht         j                  ddit         j                  ddjt         j                  ddkt         j                  ddli t         j                  ddmt         j                  ddnt         j                  do t         j                  ddpt         j                  ddqt         j                  ddrt         j                  ddst         j                  dt t         j                  ddut         j                  ddvt         j                  ddwt         j                  d dxt         j                  dy t         j                  ddzt         j                  j                  dd|t         j                  dd}t         j                  dd~i t         j                  ddt         j                  ddt         j                  ddt         j                  dd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                  ddt         j
                  ddt         j                  j
                  ddt         j                  ddi t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                   d t         j"                  ddt         j                  j$                  ddt         j                  j&                  ddt         j                  j(                  ddt         j                  j*                  ddt         j,                  d t         j.                  d	di t         j0                  d
dt         j2                  dd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                  dd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^                  ddt         j\                  j`                  ddt         j\                  jb                  ddt         j\                  jd                  ddt         j\                  jf                  ddt         j\                  jh                  ddt         j\                  jj                  ddt         j\                  jl                  ddt         j\                  jn                  ddt         j\                  jp                  ddt         j\                  jr                  ddt         j\                  jt                  ddi t         j\                  jv                  ddÄt         j\                  jx                  ddĄt         j\                  jz                  ddńt         j\                  j|                  ddƄt         j\                  j~                  ddǄt         j\                  j                  ddȄt         j\                  j                  ddɄt         j\                  j\                  ddʄt         j                  dd˄t         j                  dd̄t         j                  d̈́ t         j                  d΄ t         j                  dτ t         j                  ddЄt         j                  ddфt         j                  d҄ t         j                  ddӄi t         j                  ddԄt         j                  ddՄt         j                  ddքt         j                  ddt         j                  ddfdׄt         j                  d؄ t         j                  ddلt         j                  ddڄt         j                  ddۄt         j                  dd܄t         j                  d݄ t         j                  ddބt         j                  dd߄t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddi t         j                  d t         j                  d t         j                  d t         j                  ddt         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  j                  d t         j                  ddi t         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  d  t         j                  ddt         j                  d t         j                  ddt         j                  d t         j                  d t         j                  ddi t         j                  ddt         j                   d t         j                  d	 t         j                  dd
t         j                  j                  ddt         j                  j                  ddt         j
                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  ddt         j                  d i t         j                   ddt         j"                  ddt         j$                  d t         j&                  ddt         j                  j(                  ddt         j                  j*                  ddt         j                  j,                  ddt         j.                  ddt         j0                  dd t         j2                  dd!t         j4                  dd"t         j6                  dd#t         j8                  dd$t         j:                  dd%t         j<                  dd&t         j>                  dd't         j@                  dd(i t         jB                  dd)t         jD                  dd*t         jF                  dd+t         jH                  dd,t         jJ                  dd-t         jL                  d. t         jN                  dd/t         jP                  dd0t         jR                  dd1t         jT                  dd2t         jV                  dd3t         jX                  dd4t         jZ                  dd5t         j\                  d6 t         j^                  dd7t         j`                  dd8t         jb                  dd9i t         jd                  dd:t         jf                  dd;t         jh                  dd<t         jj                  d= t         jl                  d> t         jn                  dd?t         jp                  dd@t         j                  jd                  ddAt         j                  jr                  ddBt         j                  jt                  ddCt         j                  jf                  ddDt         j                  jp                  ddEt         jv                  dF t         j                  jv                  ddGt         j                  jx                  ddHt         j                  jz                  ddIt         j|                  dJ i t         j                  j|                  dK t         j~                  ddLt         j                  ddMt         j                  ddNt         j                  d dOt         j                  d dPt         j                  d dQt         j                  d!dRt         j                  ddSt         j                  d"dTt         j                  ddUt         j                  ddVt         j                  dW t         j                  ddXt         j                  ddYt         j                  ddZt         j                  d[ i t         j                  d\ t         j                  d] t         j                  d^ t         j                  d_ t         j                  d` t         j                  da t         j                  ddbt         j                  d#dct         j                  dd t         j                  de t         j                  ddft         j                  ddgt         j                  ddht         j                  ddit         j                  ddjt         j                  dk t         j                  dl i t         j                  d$dmt         j                  dn t         j                  do t         j                  dp t         j                  d%dqt         j                  d&drt         j                  ds t         j                  d'dtt         j                  du t         j                  ddvt         j                  ddwt         j                  ddxt         j                  ddyt         j                  ddzt         j                  j                  j                  d{ t         j                  j                  j                  d| t         j                  j                  j
                  dd}i t         j                  j                  j                  dd~t         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j*                  d(dt         j                  j                  j                  d)dt         j                  j                  j                  d)dt         j                  j                  j\                  d*dt         j                  j                  jn                  ddt         j                  j                  j                  d+dt         j                  j                  jp                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  d,dt         j                  j                  j                  ddi t         j                  j                  j                  d-dt         j                  j                  j                  d-dt         j                  j                  j                  d-dt         j                  j                  j                  d-dt         j                  j                  j                  ddt         j                  j                  j.                  d	dt         j                  j                  j0                  d.dt         j                  j                  jX                  d(dt         j                  j                  j                   d/dt         j                  j                  j                  d0dt         j                  j                  j                  d0dt         j                  j                  j                  d0dt         j                  j                  j                  d0dt         j                  j                  j
                  d1dt         j                  j                  j                  d2dt         j                  j                  j                  d3dt         j                  j                  j                  d4di t         j                  j                  j                  d%dt         j                  j                  j                  d5dt         j                  j                  j                  ddt         j                  j                  j                  d6dt         j                  j                  j                  ddt         j                  j                  j                   d7dt         j                  j                  j                  d8dt         j                  j                  j"                  ddt         j                  j                  j                  d9dt         j                  j                  j.                  d%dt         j                  j                  j                  d:dt         j                  j                  j                  ddt         j                  j                  j                  d;dt         j                  j                  j@                  d<dt         j                  j                  j                   d t         j                  j                  j"                  ddt         j                  j                  j$                  ddi t         j                  j                  j&                  ddt         j                  j                  jh                  ddt         j                  j                  j                  d!dt         j                  j                  j                  d!dt         j                  j                  j                  d!dt         j                  j                  j(                  d!dt         j                  j                  j                  d!dt         j                  j                  j*                  d!dt         j                  j                  j,                  d=dt         j                  j                  j.                  d=dt         j                  j                  j0                  d=dt         j                  j                  j2                  d9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<                  dAdi t         j                  j                  j>                  dBdt         j                  j                  j@                  d3dÄt         j                  j                  jB                  dCdĄt         j                  j                  jD                  dDdńt         j                  j                  jF                  dEdƄt         j                  j                  jH                  dǄ t         j                  j                  jJ                  ddȄt         j                  j                  jL                  ddɄt         j                  j                  jN                  dFdʄt         j                  j                  jP                  dGd˄t         j                  j                  jR                  dd̄t         j                  j                  jT                  dd̈́t         j                  j                  jV                  dd΄t         j                  j                  jX                  dHdτt         j                  j                  jZ                  dIdЄt         j                  j                  j\                  dJdфt         j                  j                  j^                  d@d҄i t         j                  j                  j`                  d<dӄt         j                  j                  jb                  d<dԄt         j                  j                  jd                  dKdՄt         j                  j                  jf                  ddքt         j                  j                  jh                  dׄ t         j                  j                  jj                  d؄ t         j                  j                  jl                  ddلt         j                  j                  jn                  dLdڄt         j                  j                  jp                  ddOdd:dۜd܄t         j                  j                  jr                  d/d݄t         j                  jt                  jv                  dMdބt         j                  jt                  jx                  dMd߄t         j                  jt                  jz                  d t         j                  jt                  j|                  dNdt         j~                  ddt         j                  d{ddt         j                  d i t         j                  dOdt         j                  j                  dPdt         j                  j                  dQdt         j                  j                  	 dRdt         j                  dSdt         j                  dOdt         j                  d t         j                  d t         j                  dTdt         jD                  dDdt         j                  d t         j                  dUdt         j                  ddt         j                  dVdt         j                  j                  dWdt         j                  d t         j                  d i t         j                  ddt         jF                  d t         j                  ddt         j                  ddt         jH                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d  t         j                  d t         j                  d t         j                  d t         j                  d t         j                  ddt         j                  j                  dXdt         j                  dYdi t         j                  dYdt         j                  d	 t         j                  d
 t         j                  d t         j                  d t         j                  d t         j                  d t         j                  	 	 dZ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dt         j                  ddt         j                  j                  ddi t         j                  d t         j                  d t         j                  ddt         jJ                  ddt         j                  ddt         j                  dd t         j                  dd!t         j                  d" t         jN                  dFd#t         j                  d$ t         j                  dd%t         j                  d& t         j                  dd't         j                  dd(t         j                  d]d*t         j                  dd+t         j                  dd,i t         j                  d- t         jP                  dG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                  d^d6t         j                  d7 t         j                  d8 t         j                  d_d9t         j                  d_d:t         jR                  dd;t         j                  dd<t         j                  dd=i t         j                  dd>t         j                  dd?t         j                   dd@t         j"                  ddAt         j$                  ddBt         j&                  dC t         j                  j&                  dD t         j(                  ddEt         j*                  ddFt         j`                  ddGt         j                  j,                  ddHt         j                  j.                  ddIt         j0                  d`dddJdKt         j2                  ddLt         j4                  ddMt         j6                  ddNt         j8                  ddOi t         j:                  ddPt         j<                  ddQt         j>                  ddRt         j@                  ddSt         jB                  ddTt         jD                  dadUt         jF                  ddVt         jH                  ddWt         jJ                  ddXt         jL                  dY t         jN                  dZ t         jP                  d[ t         jR                  d\ t         jT                  d] t         jV                  d^ t         jX                  d_ t         jZ                  d` i t         j\                  da t         j^                  db t         j`                  dc t         jb                  dd t         jd                  de t         jf                  df t         jh                  dg t         jj                  dh t         jl                  di t         jn                  ddjt         jp                  dbdkt         jr                  dcdlt         j                  jp                  ddmt         j                  jt                  ddnt         jv                  do t         jx                  dp t         jz                  j|                  dq i t         jz                  j~                  dr t         jz                  j                  ds t         jz                  j                  dt t         jz                  j                  du t         jz                  j                  ddvt         jz                  j                  ddwt         jz                  j                  ddxt         jz                  j                  ddyt         jz                  j                  dz t         jz                  j                  d{ t         jz                  j8                  d| t         jz                  j:                  d} t         jz                  j                  d~ t         jz                  j<                  d t         jz                  j@                  d t         jz                  j                  d t         jz                  jB                  d i t         jz                  j                  ddt         jz                  j                  ddt         jz                  j                  d t         jz                  j                  ddt         jz                  j                  ddt         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  ddt         jz                  j                  ddt         jz                  jD                  d t         jz                  j                  d t         jz                  j@                  ddt         jz                  jX                  d t         jz                  jZ                  ddt         jz                  j                  d i t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  ddt         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  d t         jz                  j                  ddt         jz                  j                  ddt         jz                  j                  ddt         jz                  j                  ddt         jz                  j"                  d t         jz                  j`                  ddi t         jz                  j                  d t         jz                  j                  ddt         jz                  jN                  ddt         jz                  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                  dddt         j                  ddt         jl                  ddt         j                  d t         j                  d#dt         j                  d i t         j                  d t         j                  ddt         j                  ddt         j                  dedt         j                  j                  dTdt         j                  ddt         jn                  dLdt         j                  ddt         j                  d t         j                  ddt         j                  ddt         j                  d t         j                  dfdt         j                  ddÄt         j                  dĄ t         j                   ddńt         j                  ddƄi 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ք t         j$                  dׄ i t         j&                  dd؄t         j(                  dل t         j*                  ddڄt         j,                  ddۜd܄t         j.                  d݄ t         j0                  dބ t         j2                  d߄ t         j4                  d t         j6                  d t         j8                  d_dt         j:                  ddt         j<                  ddt         j>                  d t         j@                  d t         jB                  d t         jD                  d t         jF                  d i t         jH                  d t         jJ                  d t         jL                  d t         jN                  d t         jP                  d t         jR                  d t         jT                  d t         jV                  ddt         jX                  d t         jZ                  d t         j\                  d | j^                  d | j`                  d | jb                  d | jd                  d | jf                  d | jh                  d i | jj                  d | jl                  d | jn                  d | jp                  d | jr                  d | jt                  d  | jv                  d | jx                  d | jz                  d | j|                  d | j~                  d | j                  d | j                  d | j                  d | j                  d	 | j                  d
 | j                  d i | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddd| j                  d | j                  d | j                  j                  d | j                  j                  d i | j                  j                  d | j                  j                  d | j                  j                  d  | j                  j                  d! | j                  j                  d" | j                  j                  d# | j                  j                  d$ | j                  j                  d% | j                  j                  d& | j                  j                  d' | j                  j                  d( | j                  j                  d) | j                  d* | j                  d+ | j                  d, | j                  j                  d- | j                  j                  d. i | 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> | j                  j                  d? i | j                  j                  d@ | j                  j                  dA | j                  j                  dB | j                  j                  dC | j                  j                  dD | j                  j                  dE | j                   j                  dF | j                  j                  dG | j                  j                  dH | j                  j                  dI | j                  j                  dJ | j                  j                  dK | j                  j                  dL | j
                  j                  dM | j                  ddN| j                  dO | j                  dP i | 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_ | j.                  d0d`| j0                  t         j2                  fdai | j4                  t         j2                  fdb| j6                  t         j2                  fdc| j8                  t         j2                  fdd| j:                  d)ddedf| j<                  dg | j>                  dh | j@                  t         jB                  fdi| jD                  ddj| jF                  t         j2                  fdk| jH                  t         j2                  fdl| jJ                  t         j2                  fdm| jL                  t         j2                  fdn| jN                  t         j2                  fdo| jP                  dp | jR                  dq | j                  ddr| jT                  ds i | jV                  ddt| jX                  t         j2                  fdu| jZ                  t         j2                  fdv| j\                  dw | j^                  dx | j`                  dy | jb                  dddedz| jd                  d{ | jf                  d| | jh                  t         j2                  fd}| jj                  t         j2                  fd~| jl                  dde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                  dgdded| j@                  d | j                  t         j2                  fd| j                  d | j                  d | j                  dd| j                  dd| j.                  d | j                  d i | j                  d | j                  d | j                  d | j                  d | jx                  d | j                  d | j                  d | j                  d | j                  dd| j                  d | j                  ddde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                  dhd| j                  d | j                  d | j                  t         j2                  fd| j                  d | j                  d_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Ǆ | jr                  dȄ | jv                  d)dɄ| j                  dʄ | j                  d˄ | j                  d̄ | j                  d̈́ | j                  did΄| j                  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                  |       t         j                  j                         rLd-dlm} |j                  i |j                  d0dڄ|j                  ddۄ|j                  djd܄|j                  dd݄|j                  ddބ|j                   dd߄|j"                  dd|j                  djd|j                  djd|j$                  dd|j&                  dd|j(                  dd|j*                  dd|j,                  dkd|j.                  dkd|j0                  dkd|j2                  dkd       |S (l  ar  Return a dict containing dummy overrides for all overridable functions

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

    Examples
    --------
    >>> import inspect
    >>> my_add = torch.overrides.get_testing_overrides()[torch.add]
    >>> inspect.signature(my_add)
    <Signature (input, other, out=None)>
    Nc                      yN inputouts     r.   <lambda>z'get_testing_overrides.<locals>.<lambda>      r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  output_sizes     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )inputsrQ  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  otherrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  batch1batch2alphabetarK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  tensor1tensor2valuerK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rd  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  mat1mat2rb  ra  rK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  matvecrb  ra  rK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  vec1vec2rb  ra  rK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  thetasizealign_cornerss      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  dims     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   Fc                      yrF  rH  rJ  r\  rtolatol	equal_nans        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  ptraininplaces       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rz  keepdimrK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  msgs     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  tensorss    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   r   c                      yrF  rH  )rJ  kernel_sizer  padding	ceil_modecount_include_pads         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r^  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c	                      yrF  rH  )	rJ  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )grad_outrJ  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rJ  r  r  r  input_gweight_gbias_gs           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r  r  r  counts           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  	generatorrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  input1input2r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   r  c                      yrF  rH  rJ  targetr  size_averagereduce	reduction
pos_weights          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  weights	minlengths      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  probr  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rl  	out_dtyperK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  selfrv  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  
boundaries	out_int32rightrK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                       yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rz  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )x1x2r  compute_modes       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1         ?c                      yrF  rH  rJ  ra  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )rK  c                      yrF  rH  )rK  matricess     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  groupss     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  upperrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  check_errorsrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  numeln_binsratio	bit_widths        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  chunksrz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  minmaxrK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r
  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rK  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  
correctionfweightsaweightss       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rwith_replacements      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )realimags     r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r1   c                      yrF  rH  )absangs     r.   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r1   c                      yrF  rH  )rJ  ords     r.   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>'  rM  r1   c                      yrF  rH  )rJ  padrg  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r1   c                      yrF  rH  rJ  r  r  r  r  dilationr  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r1   c                      yrF  rH  r*  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r1   c                      yrF  rH  r*  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r1   c	                      yrF  rH  )	rJ  r  r  r  r  r+  
transposedoutput_addingr  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r1   c                      yrF  rH  )rJ  r  r  r(  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r1   c                      yrF  rH  rJ  r  r  r  r  output_paddingr  r+  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r1   c                      yrF  rH  r3  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r1   c                      yrF  rH  r3  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r1   c                      yrF  rH  r  r  r  marginr  r  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r1   c                      yrF  rH  )r  r  rz  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r1   c                      yrF  rH  rJ  r\  rz  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r1   rG  c                      yrF  rH  r@  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r1   c                      yrF  rH  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r.   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r1   c                      yrF  rH  rJ  rz  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r1   c                      yrF  rH  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r1   c                      yrF  rH  rJ  rz  rK  ra   s       r.   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r1   c                      yrF  rH  rN  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r1   c                      yrF  rH  yxrz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r1   c                      yrF  rH  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r1   c                      yrF  rH  rJ  diagonalrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r1   c                      yrF  rH  )rJ  offsets     r.   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r1   c                      yrF  rH  )rJ  nrz  prependappendrK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r1   c                      yrF  rH  rJ  r_  dim1dim2s       r.   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r1   c                      yrF  rH  re  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r1   c                      yrF  rH  )rJ  srcr_  rf  rg  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r1   c                      yrF  rH  )r  rj  rv  r  storage_offsets        r.   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r1   c                      yrF  rH  )rJ  r\  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r1   c                      yrF  rH  rJ  r\  rounding_moderK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r1   c                      yrF  rH  rp  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r1   c                      yrF  rH  rJ  rl  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r1   c                      yrF  rH  )rk  rl  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r1   c                      yrF  rH  rJ  indices_or_sectionss     r.   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r1   c                      yrF  rH  rJ  UPLOrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r1   c                      yrF  rH  )equationoperandss     r.   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r1   c                      yrF  rH  rJ  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r.   rL  z'get_testing_overrides.<locals>.<lambda>_  rM  r1   c
                      yrF  rH  )
rJ  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r.   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r1   c                      yrF  rH  rJ  ra   ro   r`   requires_grads        r.   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r1   c                      yrF  rH  rJ  r\  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r1   c                      yrF  rH  )rJ  scale
zero_pointaxis	quant_min	quant_maxs         r.   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r1   c                      yrF  rH  )rS  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r.   rL  z'get_testing_overrides.<locals>.<lambda>p  rM  r1   c                      yrF  rH  rJ  packed_weightr  outputs       r.   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r1   c                      yrF  rH  rJ  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r1   c                      yrF  rH  )rJ  abs      r.   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r1   c                      yrF  rH  rJ  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r1   c                      yrF  rH  rJ  ra  rz  norms       r.   rL  z'get_testing_overrides.<locals>.<lambda>}  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  srz  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  	start_dimend_dims      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  dimss     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  exponentrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  
fill_valuerK  ra   ro   r`   r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  	dep_tokens      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  indexrK  sparse_grads        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rr  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  spacingrz  
edge_orders       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  gridinterpolation_modepadding_moderw  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  
num_groupsr  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c	                      yrF  rH  	rJ  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  w_ihw_hhb_ihb_hhs         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  lambds     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  r  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  valuesrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r;  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  binsr  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r%  r  r  r  densityrK  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r%  r   r  r'  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  taus     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rk  rl  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rz  r  sources       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r4  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  indicesr!  
accumulates       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  r  rg  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  r  r5  r  include_inputs         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[   s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )eteassume_uniqueinverts       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r?  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r?  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c	                      yrF  rH  )	rJ  r  r  r  r  use_input_statsr  r  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r|  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c
                      yrF  rH  )
rJ  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r  
log_targets         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  krz  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  	hermitianr  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rk  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )LDpivotsBrk  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  normalized_shaper  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  endr  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  ri  rp  Xra  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rz  ra   s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rS  rR  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  namesr  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c	                      yrF  rH  )	databatch_sizesr  r  r  r  r  r  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )Apivot	get_infosrK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r:  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  maskrg  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r5  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r1   c                      yrF  rH  rJ  r  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )LUro  rp  leftadjointrK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  ra  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  ra  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r~  rk  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r+  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r+  return_indicesr  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  r  ra   rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r1   c                       yrF  rH  )r  r%   s     r.   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  exponential_average_factorepsilons           r.   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r1   c	                      yrF  rH  	rJ  r  r  r  r  r+  r  	benchmarkdeterministics	            r.   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r1   c	                      yrF  rH  )	rJ  r  zra  r  r  r  r+  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r1   c                      yrF  rH  r*  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r1   c
                      yrF  rH  )
rJ  r  r  r  r4  r  r+  r  r  r  s
             r.   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r1   c	                      yrF  rH  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r1   c                      yrF  rH  )rJ  r  weight_stride0r  cxr  hidden_sizer  r  r  r  r  r  dropout_states                 r.   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r1   c                      yrF  rH  rJ  r5  destinations      r.   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>9  rM  r1   c                      yrF  rH  )rJ  
descendingrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r1   c                      yrF  rH  )rJ  num_samplesreplacementrK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r1   c                      yrF  rH  )rJ  ro  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r1   c                      yrF  rH  rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r1   c                      yrF  rH  )rJ  rz  startrb  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r1   c                      yrF  rH  )rJ  nanposinfneginfrK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  r  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r1   c                      yrF  rH  rJ  rr  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r1   c                      yrF  rH  rJ  rr  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r1   c                      yrF  rH  )rJ  r  r  NCHxWgroupr  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r1   c                      yrF  rH  )rJ  r  rz  r  ra   s        r.   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r1   c                      yrF  rH  rP  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r1   c                      yrF  rH  rP  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r1   c                      yrF  rH  rJ  rQ  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r1   c                      yrF  rH  rt  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r1   c                      yrF  rH  rJ  r  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r1   c                      yrF  rH  rJ  r  r  r  r  r  divisor_overrides          r.   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  r  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r1   c                      yrF  rH  rJ  r  r  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r1   c                      yrF  rH  r:  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r1   c                      yrF  rH  )rJ  r  r  r  ignore_indexr  r  label_smoothings           r.   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r1   c                      yrF  rH  rC  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>u  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  r  r   r  include_last_offsetr  s              r.   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r1   c                      yrF  rH  )rJ  rQ  r  r+  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  rQ  output_ratior  _random_sampless         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r)  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r)  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r)  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  varr   r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  approximates     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  rw  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r
  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )logitsr+  hardr  rz  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  min_valmax_valr  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r#  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  rJ  r  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rv  scale_factorr  rw  recompute_scale_factor	antialiass          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  re  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  negative_sloper  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rv  ra  rb  ri  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  rz  _stacklevelra   s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rP  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rP  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r:  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r+  r  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rU  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r8  r  r  r  rQ  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r\  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r\  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rE  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )querykeyrg  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r;  r  r  r  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  r  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rz  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r[   num_classess     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r(  r  rg  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  r  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  	log_inputr   r  r  r  r  s           r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  lowerr  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )ra  rb  rg  ro  rj  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  r  rb  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  deltar  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rz  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rL  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rL  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rb  	thresholds      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  rg  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c
                      yrF  rH  
anchorpositivenegativer;  r  r  swapr  r  r  s
             r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )distance_functionr;  r  r  c                     yrF  rH  )r  r  r  r  r;  r  r  s          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r+  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r[   r  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r[   r  stdr  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r[   vals     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r[   r  r  nonlinearityr  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  as_tuples     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )r  c                     yrF  rH  )rJ  rv  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  rz  r  rK  ra   s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r"  rz  r  rK  ra   s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      	 yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  s     13 r1   c                      yrF  rH  )vpowrz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r*  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  input3r  	transposes        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rz  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  qr_  r}  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rconds     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rk  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r1   c                      yrF  rH  )rJ  upscale_factors     r.   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r1   c                      yrF  rH  )rJ  downscale_factors     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r   r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  ra   s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r5  r9  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  somerK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  r  rz  r  interpolationrK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  scaleszero_pointsr  ra   s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  ra   s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  ra   reduce_ranges      r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  )rJ  r  r  r  r0  r  output_scaleoutput_zero_points           r.   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r1   c                      yrF  rH  rJ  r  r  r  r  r  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r.   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r1   c                      yrF  rH  r  s                 r.   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r1   r      c                      	 yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>)  s
     " r1   c                      	 yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>.  s     !# r1   c                      	 yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>4  s     !#	 r1   c                      yrF  rH  r  s                 r.   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r1   c                      yrF  rH  r  s                 r.   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r1   c                      yrF  rH  r@  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r1   c                      yrF  rH  )rJ  r  rz  maxnormrK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r1   c                      yrF  rH  )rJ  shapes     r.   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r1   c	                      yrF  rH  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r1   c	                      yrF  rH  r  s	            r.   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r1   c                      yrF  rH  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r1   c                      yrF  rH  )rJ  shiftsr  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r1   r   r  c                      yrF  rH  )rJ  ri  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r1   c                      yrF  rH  )r  r  compressed_indices_dtypes      r.   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r1   c                      yrF  rH  )rJ  r\  ra  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r1   c                      yrF  rH  rj  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r1   c                      yrF  rH  rJ  rz  r  rj  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r1   c                      yrF  rH  r!  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r1   c                      yrF  rH  )rJ  rz  r  rj  r  include_selfs         r.   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r1   c                      yrF  rH  )sorted_sequencerJ  r  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>^  rM  r1   c                      yrF  rH  )r  r  lengthsr8  r  r  unsafes          r.   rL  z'get_testing_overrides.<locals>.<lambda>_  rM  r1   c                      yrF  rH  )rJ  rz  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r1   c                      yrF  rH  )rJ  rj  rz  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>a  rM  r1   c                      yrF  rH  rJ  rj  rz  r  rx  steps         r.   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r1   c                      yrF  rH  r-  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>c  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r1   c                      yrF  rH  rv  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r1   c                      yrF  rH  rv  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>o  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>p  rM  r1   c                      yrF  rH  )r  rp  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r1   c                      yrF  rH  )r  rp  r  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r1   )stablerK  c                     yrF  rH  )rJ  rz  r  r?  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r1   c                      yrF  rH  r[   split_size_or_sectionsrz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r1   c                      yrF  rH  rB  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>u  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r1   c                      yrF  rH  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r1   c                      yrF  rH  rj  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r1   c                      yrF  rH  )rJ  r[  r\  r]  r^  r_  pad_moder`  ra  rc  align_to_windows              r.   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rU  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  cs      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r$   s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  
compute_uvrK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r}  Ms       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  full_matricesrK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rJ  dim0rf  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  axis0axis1s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r\  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r8  rz  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  inds     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  rK  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  rz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  ri  rz  r  rK  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rV  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rn  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rQ  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rQ  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  r  unitriangulars        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rp  r  r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c
                      yrF  rH  r  s
             r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r[  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rI  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  rz  sizesr  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  sortedreturn_inversereturn_countsrz  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  r  r  rz  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r8  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rB  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rB  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  rK  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rS  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  ry  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )	conditionrS  rR  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )rJ  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  levels     r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  )primaltangentr  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rv  r  rl  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r_  rf  rg  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )implicitc                     yrF  rH  )r  rv  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r1   c                      yrF  rH  )r  rz  r  rb  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r1   c                      yrF  rH  )r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rv  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rz  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rz  r  rx  r.  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  
split_sizerz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  split_sizesrz  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  ro  rf  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  ra   s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  	dimensionrv  r.  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r1   c                      yrF  rH  r  r\  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>'  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r1   c                      yrF  rH  r	  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>9  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r1   c                      yrF  rH  )r  arrays     r.   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r1   c                      yrF  rH  )r  idxs     r.   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r1   c                      yrF  rH  )r  memos     r.   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r1   c                      yrF  rH  )r  format_specs     r.   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r1   c                      yrF  rH  )r  protos     r.   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r1   )tensor_contentsc                     yrF  rH  )r  r7  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r1   c                      yrF  rH  )r  ri  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r1   c                      yrF  rH  )r  ds     r.   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r1   c                      yrF  rH  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r.   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r1   c                      yrF  rH  )r  rK  rL  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>^  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>_  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>a  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>c  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>o  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>p  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>u  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r1   c                      yrF  rH  )r  ra   non_blockingr%   s       r.   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>}  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  orderellipsis_idxs      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  callables     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  gradientretain_graphcreate_graphrS  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rh   s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )r  c                     yrF  rH  )r  mediansigmar  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  	coalesceds     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rj  rs  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rj  r_  rf  rg  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  ambiguity_checks     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                     yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  rg  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                     yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r[   s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                     yrF  rH  )r  r  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r[   r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rS  rR  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rl  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r\  assigns      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  rb  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r8  r[   r9  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                     yrF  rH  )r  from_tor  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  streams     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  hooks     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  names     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r5  rl  rv  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rj  rz  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rj  rz  r  rx  r.  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  accumulate_matchess      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  size1size2	dense_dims       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  rk  rl  rb  ra  rK  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  repss     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  ra   rs  copyrh   s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   )masked_gradc                     yrF  rH  r  ra   r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s     r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  max_version	dl_devicer  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r  s      r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r  r   drivers       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  )r  r`   rs  r%   s       r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   c                      yrF  rH  r  s    r.   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r1   is______i__rbitwise_c                      yrF  rH  )r[   rj  r  async_op	group_srcs        r.   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r1   c                      yrF  rH  )r[   opr  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r1   c                      yrF  rH  )r[   dstr   r  r  	group_dsts         r.   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r1   c                      yrF  rH  )r  r   r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r1   c                      yrF  rH  )tensor_listr[   r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r1   c                      yrF  rH  )output_tensorinput_tensorr  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r1   c                      yrF  rH  )output_tensor_listsinput_tensor_listr  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r1   c                      yrF  rH  )r[   gather_listr"  r  r  r#  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r1   c                      yrF  rH  )r[   scatter_listrj  r  r  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r1   c                      yrF  rH  )r  
input_listr   r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r1   c                      yrF  rH  )r  rJ  r   r  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r1   c                      yrF  rH  )r  rJ  output_split_sizesinput_split_sizesr  r  s         r.   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r1   c                      yrF  rH  )output_tensor_listr,  r  r  s       r.   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r1   c                      yrF  rH  r[   r"  r  tagr#  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r1   c                      yrF  rH  r[   rj  r  r;  r  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>9  rM  r1   c                      yrF  rH  r:  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r1   c                      yrF  rH  r=  s        r.   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r1   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  rB  )rG  N)r   r  FrF  )r  rG  NNN)r   r   r  )r   rG  )rE  )LN)NNrC  FF)NrE  Fr  FNN)NNNF)FF)NrG  N)NrF  rG  N)r   rG  )TT)NF)NNr  )NNrA  T)      ?)NFr   N)r  NNr  )d   r   r   N)rJ  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   r  F)Nr   r  FF)NFNN)rG  FN)        NNN)NNrA  )NrA  )rE  NFN)rI  FF)Nr   FTN)NNF皙?rA  )NNNr  )NNNr  rK  )rI  TF)	NNrE  Fr  FNFN)r  r   r  )NNFN)Fư>r  )none)rG  )bilinearr   N)r  Fg|=rG  )g      r  F)NNNNTrL  rA  )NNnearestNNF)NNr  N)g{Gz?F)g-C6?g      ?r  )Nr3   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  )NNrM  Nr  )rE  r  g-q=N)r   r   )rC  rN  F)TFNrB  Nr  )NrN  )g      ?gUUUUUU?FF)NrK  )NNr  r  )r  r  N)r     )r  rE  rN  FNNr  )rK  r  N)r   fan_in
leaky_reluN)froNFNN)NNFNN)rE  NFNN)rU  rH  FNN)rE  r   )TF)NTrE  )V瞯<)rV  F)reducedN)NFlinearN)rH  r  r  F)rH  )r   r   )r  r  F)rH  )r   r   r   )r  r  r  F)r  r  )r  NNNr   F)r   NNr  )rG  F)	NNNTreflectFTNN)TTN)   rE  N)rE  N)TFF)TFFN)r  rE  )Nr   NN)NNNN)NNNFN)NNr   N(  r5   r6   r  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshr<  addbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullirP   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffr\  diagonal_scatteras_strided_scatterdigammadistdivdividedotr  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdge
get_devicegreater_equalgeqrfi0inneroutergerr  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrinkhash_tensor	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacr  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_fused_rms_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_lossrT  rX  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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   r0  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_hooksr@  _cdatarA  rB  _grad_fngrad_fn
grad_dtype_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  r`   ra   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itemsizero   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_rr   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrr  rz  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesr8  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_ri   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_shortrv  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storagerl  storage_typesum_to_sizer  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asr!  viewview_aszero_
__dlpack____dlpack_device__r  r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrr   items__name__
startswithlenextendr  updatedistributedis_availabletorch.distributed	broadcast
all_reducer  all_reduce_coalesced
all_gatherall_gather_into_tensorall_gather_coalescedreduce_scatterreduce_scatter_tensorall_to_all_single
all_to_allisendirecvsendrecv)r6   retprivateuse1_backend_nameret2ignoredri  r  r  subnamer  r   r  s               r.   r   r     sDY   6 \\F{%		-{%2{% 	!!#@{% 	!!#A	{%
 	

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

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

.+{%, 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

+g{%h 	Ki{%j 			-k{%l 	

.m{%n 	 3o{%p 	  "<q{%r 	:s{%t 	Hu{%v 	Jw{%x 	

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

K{{%| 	%}{%~ 	5{%@ 	1A{%B 	5C{%D 	 eE{%F 	%%'aG{%H 	

:I{%J 	!! LM{%P 	

:Q{%R 	2S{%T 	/U{%V 	-W{%X 	<Y{%Z 	h[{%\ 	  "g]{%^ 	6_{%` 	;a{%b 	Lc{%d 	%%'We{%f 	8g{%h 	1i{%j 			-k{%l 	2m{%n 	;o{%p 	2q{%r 	9s{%t 	%%'`u{%v 	11oy{%| 	e}{%~ 	5{%@ 	@A{%B 	C{%D 	""OE{%F 	/G{%H 	oI{%J 	QK{%L 	'')>M{%N 	FO{%P 	%CQ{%R 	>S{%T 	1U{%V 	!!#@W{%X 	6Y{%Z 	?[{%\ 	N]{%^ 	<_{%` 	##%Ha{%b 	0c{%d 	oe{%f 	9g{%h 	2i{%j 	_k{%l 	Om{%n 	Oo{%p 	?q{%r 	s{%t 	u{%v 	w{%x 	y{%z 	1{{%| 	/}{%~ 	A{%@ 	/A{%B 	3C{%D 	4E{%F 	4G{%H 	2I{%J 	3K{%L 	3M{%N 	1O{%P 	2Q{%R 	2S{%T 	1U{%V 	2W{%X 	2Y{%Z 	.[{%\ 	-]{%^ 	._{%` 	/a{%b 	Oc{%d 	0e{%f 	g{%h 	3i{%j 	k{%l 	?m{%n 	.o{%p 	/q{%r 	/s{%t 	5u{%v 	0w{%x 	2y{%z 	{{%| 	}{%~ 	/{%@ 	A{%B 	7C{%D 	4E{%F 	_G{%H 	AI{%J 	1K{%L 	/M{%N 	/O{%P 	/Q{%R 			?S{%T 			?U{%V 	&&W{%X 	**22OY{%Z 	o[{%\ 	]{%^ 	__{%` 	oa{%b 	c{%d 	e{%f 	!!?g{%h 	i{%j 	--/pk{%l 	**,Vm{%n 	22Oo{%p 	_q{%r 	s{%t 	ou{%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 	$$oU{%V 	  /W{%X 	Y{%Z 	[{%\ 	_]{%^ 	o_{%` 	a{%b 	_c{%d 	  /e{%f 	$$og{%h 	oi{%j 	k{%l 	_m{%n 	_o{%p 	''//q{%r 	Ns{%t 	ou{%v 	ow{%x 	y{%z 	{{%| 	_}{%~ 	_{%@ 	OA{%B 	_C{%D 	OE{%F 	=G{%H 	I{%J 	K{%L 	/M{%N 	=O{%P 	0Q{%R 	8S{%T 	9U{%V 	kW{%X 	E4I4IMY{%Z 	0E0EI[{%\ 	0E0EI]{%^ 	0E0EI_{%` 	MTMa{%b 	c{%d 	6e{%f 	e6M6MQg{%h 	>i{%j 	

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

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

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

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

u/D/DH_{%` 	_a{%b 	oc{%d 	_e{%f 	/g{%h 	1i{%j 	/k{%l 	_m{%n 	MTMo{%p 	0q{%r 	0E0EIs{%t 	6u{%v 	5w{%x 			8y{%z 	@{{%| 	E}{%~ 	?{%@  	A {%B  	""OC {%D  	--E {%F  	%%G {%H  	I {%J  	oK {%L  	,M {%N  	?O {%P  	GQ {%R  	S {%T  	LDLU {%V  	5W {%X  	3Y {%Z  	3[ {%\  	113H] {%^  	,_ {%`  	-a {%b  	Bc {%d  	1e {%f  	-g {%h  	-i {%j  	0k {%l  	  "8m {%n  	Oo {%p  	[q {%r  	?s {%t  	ou {%v  	1F1FJw {%x  	_y {%z  	W{ {%|  	?} {%~  	1 {%@! 	&&(WA!{%B! 	GC!{%D! 	'')QE!{%F! 	OG!{%H! 	I!{%J! 	K!{%L! 	M!{%N! 	_O!{%P! 	1Q!{%R! 	+S!{%T! 			EUZUjUjnU!{%V! 	IIW!{%X! 	GY!{%Z! 	/[!{%\! 	]!{%^! 	/_!{%`! 	.a!{%b! 	=c!{%d! 	7e!{%f! 	+.od  /+Fu!{%C|! 	((BB  v/0F 	GF456 JYGFc":!;<=EEFD#%G		 1 JJJJ1::$AJJ%AJJ%
 ::  , jjZ!23GLL$&$(>RV@VW  	D64.D~$c/d'6IT
	%. JJt
 %%'(

 g!W m ))+b	
 ![ ++-o ))+x v x ##%g **,i &&  )N !m 

Z 

Z  		Y!" 		Y#	
, Jr1   
dispatcherc                 `     dt         t        t        f   dt         t        t        f   f fd}|S )a  Wraps a given function with ``__torch_function__`` -related functionality.

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

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

    Examples
    --------
    >>> def dispatcher(a):  # Must have the same signature as func
    ...     return (a,)
    >>> @torch.overrides.wrap_torch_function(dispatcher)
    >>> def func(a):  # This will make func dispatchable by __torch_function__
    ...     return a + 0
    r   r"   c                      t        j                         dt        j                  dt        j                  dt
        f fd       t        t        t        t
        f         S )Nr$   r%   r"   c                       | i |}t        |      r+t        t        t        t        t
        f         |g| i |S  | i |S r@  )r   r   r   r   r   r   )r$   r%   relevant_argsr
  r   wrappeds      r.   r
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped]  s^    &77M!-0,"b&)73]EIMS  (((r1   )	functoolsr   r   r$   r%   r   r   r   )r   r
  r
  s   `@r.   r)  z"wrap_torch_function.<locals>.inner\  sS    			)277 	)bii 	)B 	) 
	) HRV$g..r1   )r   r   r   )r
  r)  s   ` r.   r   r   B  s/    4/HRV$ /"b&)9 / Lr1   r
  get_type_fnc                    |t         }t        j                  j                         sg S t	               }g }| D ]  } ||      }||vst        |d      s|j                  t        j                  j                  usD|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setr
  r  _disabled_torch_function_implr<  r
  	enumerate
issubclassinsert)	r
  r
  overloaded_typesoverloaded_argsargarg_typer  iold_args	            r.   _get_overloaded_argsr
  l  s    J  88..0	"%%!#O (s# ,,"67++8899:
   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%;(< r1   
public_apir$   r%   c           	         t        |      }t        t        t        |            }t	               r1t               5 }|j                  | |||      }ddd       t        ur|S |D ]w  }|j                  }	t        |	d      rF|	j                  |u r8|	t        j                  j                  urt        j                  dt        d        |	| |||      }|t        usu|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.rE  
stacklevel.zno implementation found for 'z.' on types that implement __torch_function__: z nor in mode )r
  tuplemapr.
  r   _pop_mode_temporarilyr  NotImplementedr
  r
  r5   r
  r
  r*   warnDeprecationWarning
__module__r
  _get_current_function_mode	TypeError)r
  r
  r$   r%   r
  typesr  resultoverloaded_argtorch_func_method	func_namer
  r  s                r.   r   r     sj   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.I	N 	N@  Fs   D;?E
;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                  t   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 ]4  \  }}}|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 r%t;        |	j=                  ||j>                              | |   jA                  |j2                         t/        |      s| d| ||<   |r|t7               v r5d}	|t9               v r%t;        |	j=                  ||j>                              | |   jA                  |       * 7 | |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   r  r   rv	  r
  endswithislowerr
  object
isinstancer
  
ModuleType
__future___Featurer  r
  r-  __set__r   r   AssertionErrorformatr
  rc  )
overridable_funcsr  tested_namespacesnamespace_str	namespacens_funcsr
  r'   r   r  s
             r.   _get_overridable_functionsr
  E  s2    $//5E	%'	U--u/?/?/G/GH	 3 3S9L9L5MN	%((--UXX]]);<	s5<<'89	s5<<'89	eiiUYY0	%--U]]);<	 /@ A6*y(! @	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#::=  ||'<'>>,SZZ	4==-QRR%d+224<<@D>*O1YK8E$K ,..9  022(It}})MNNi(//5A@	6A6D e##r1   c                      t               d   S )a  List functions that are overridable via __torch_function__

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

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

    Returns
    -------
    str
        Name of the function; if eval'ed it should give back the input
        function.
    r  )r
  r5   _ops
OpOverloadOpOverloadPacketstrr
  get)fs    r.   r   r     sI      !ejj++UZZ-H-HIJ1v%'*..q11r1   c                  R    t               } t        | t        j                           }|S )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r
  r5   r6   )r
  methodss     r.   _get_tensor_methodsr    s&     23#ELL12GNr1   c                 <    | t               v xs | j                  dk(  S )aw  
    Returns True if the function passed in is a handler for a
    method or property belonging to ``torch.Tensor``, as passed
    into ``__torch_function__``.

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

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

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

    Examples
    --------
    >>> is_tensor_method_or_property(torch.Tensor.add)
    True
    >>> is_tensor_method_or_property(torch.add)
    False
    r-  )r  r
  )r   s    r.   r   r     s!    . &((FDMMY,FFr1   c                 T    t        |       t        j                  u xs t        | d      S )a9  
    Returns ``True`` if the passed-in input is a Tensor-like.

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

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

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

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

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

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

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

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

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

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

    Independent subclasses of :class:`TorchFunctionMode` are compositional:
    modes can be pushed onto a stack using ``with MyMode():``.
    When you call functions in the PyTorch API inside your
    ``__torch_function__`` implementation, by default, they will forward on to
    the next mode on the mode stack.  If you want recursively call back into
    your current ``__torch_function__`` implementation, either explicitly
    invoke ``self.__torch_function__(...)``, or use the context manager
    ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
    API self-referential (beware of infinite loops, in this case!)
    r)  Nc                      y r@  rH  r  s    r.   r   zTorchFunctionMode.__init__%  s    r1   c                     t         r@  )NotImplementedErrorr  r   r
  r$   r%   s        r.   r  z$TorchFunctionMode.__torch_function__(  s    !!r1   c                     t        |        | S r@  )
_push_moder  s    r.   	__enter__zTorchFunctionMode.__enter__+  s    4r1   c                     t                y r@  )	_pop_mode)r  exc_typeexc_valexc_tbs       r.   __exit__zTorchFunctionMode.__exit__/  s    r1   c                 D    t        j                  dd        | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`rE  r
  )r*   r
  )clsr$   r%   instances       r.   pushzTorchFunctionMode.push2  s*    ^	
 ''r1   )r"   NrH  N)r
  r
  __qualname____doc____annotations__r   r  r  r  classmethodr  rH  r1   r.   r  r    s7    < "  r1   r  c                  @    t               } | dkD  rt        | dz
        S d S )Nr   r  )r   r   )	stack_lens    r.   r
  r
  <  s%    )+I4=M!)a-0KtKr1   c                  d    t               } t        |       D cg c]  }t        |       c}S c c}w r@  )r   r   r   )r#  r
  s     r.    _get_current_function_mode_stackr%  A  s)    )+I/4Y/?@!"1%@@@s   -c                     t        |        y r@  )r   )r  s    r.   r  r  F  s
    !$'r1   c                      t               } | S r@  )r   olds    r.   r  r  J  s    
#
%CJr1   c               #   `   K   t               } 	 |  t        |        y # t        |        w xY wwr@  )r  r  r(  s    r.   r
  r
  O  s%     
+C	3
3s   . .+.c                       e Zd ZddZy)BaseTorchFunctionModeNc                     |i } ||i |S r@  rH  r  s        r.   r  z(BaseTorchFunctionMode.__torch_function__Y  s    >FT$V$$r1   r  )r
  r
  r  r  rH  r1   r.   r,  r,  X  s    %r1   r,  c               #   Z  K   t         j                  j                         } 	 t         j                  j                  t         j                  j                  j
                         d  t         j                  j                  |        y # t         j                  j                  |        w xY wwr@  )r5   r
  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r.   _enable_torch_functionr4  _  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
  _RestorePythonTLSSnapshotrH  r1   r.   r   r   i  sC      
	+	+	- 		  	 s#   A61	A36?A)z.*is deprecated, please use.*r5   r@  )Dr  r
  r
  
contextlibr
  r:  r
  r*   collections.abcr   r   r   typingr   r   r   typing_extensionsr	   r5   torch._Cr
   r   r   r   r   r   r   r   r   r
  r   r   r  r0   cacher
  r   rC  dictr   r   r.
  r
  r
  r$   r%   r   r   r   r   r
  r
  r   r   r  rA
  r   r   r  r
  r%  r  r  contextmanagerr
  r,  r4  r   rH  r1   r.   <module>r?     s  ,     
   .  % % ' 
 
 
 t_T]
 1 
2r6
     b"f	 F `s8} `  `F	 c(m  2 XtHh$67 X  Xv$'Xc]*+'xB (2r6"223'X 15LC=L3%+&-L 
#YL^VR VC=V 77V ii	V
 Vr ! . '	  * 	  Q$Ed8n	tHcM22% Q$ Q$h 	+4T(^(;#< 	+ 	+ 2 2( S]   Gx GD G G2"KJ6 6rL
A
(
  %- % 6 6  r1   