
    ^j                    :   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	 	 d1d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(	 d2d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/        ZCd0 ZDy)3aE  
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redispatch_function_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     Z/var/www/ramen.bs-engineer-server.com/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"   r0   s   ``` r/   _disable_user_warningsr1   D   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                  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"                  j.                  t         j                  j"                  j0                  t         j                  j"                  j2                  t         j                  j"                  j4                  t         j                  j"                  j6                  t         j8                  j:                  t
        t        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         j                  j                   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                  | 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arange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptablemiopen_ctc_lossemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrand	rand_likerandn
randn_likerandintrandint_likerandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearestr   has_torch_function_unaryhas_torch_function_variadicr   
grouped_mmscaled_grouped_mm	scaled_mmsigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_dtensor__new___make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_count_philox_normal__philox_uniform_sysversion_infoadd__annotate__)r7   	functionss     r/   r   r   g   s    ( \\FGGG 	G 	%%	G
 	  G 	  G 	G 	G 	G 	G 	

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

G 	

G 	G  	

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

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

mGn 	

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

IGJ 	KGL 	

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

GGH 	

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

qGr 	sGt 	uGv 	

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

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

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

    Examples
    --------
    >>> import inspect
    >>> my_add = torch.overrides.get_testing_overrides()[torch.add]
    >>> inspect.signature(my_add)
    <Signature (input, other, out=None)>
    Nc                      yN inputouts     r/   <lambda>z'get_testing_overrides.<locals>.<lambda>      r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  output_sizes     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )inputsrQ  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  otherrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  batch1batch2alphabetarK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  tensor1tensor2valuerK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rd  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  mat1mat2rb  ra  rK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  matvecrb  ra  rK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  vec1vec2rb  ra  rK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  thetasizealign_cornerss      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  dims     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   Fc                      yrF  rH  rJ  r\  rtolatol	equal_nans        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  ptraininplaces       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rz  keepdimrK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   rG  )stablec                     yrF  rH  )rJ  rz  
descendingr  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  msgs     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  tensorss    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   r   c                      yrF  rH  )rJ  kernel_sizer  padding	ceil_modecount_include_pads         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r^  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   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  r2   c                      yrF  rH  )grad_outrJ  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rJ  r  r  r  input_gweight_gbias_gs           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r  r  r  counts           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  	generatorrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  input1input2r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   r  c                      yrF  rH  rJ  targetr  size_averagereduce	reduction
pos_weights          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  weights	minlengths      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  probr  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rl  	out_dtyperK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  selfrv  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  
boundaries	out_int32rightrK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                       yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  rz  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )x1x2r  compute_modes       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2         ?c                      yrF  rH  rJ  ra  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )rK  c                      yrF  rH  )rK  matricess     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  groupss     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  upperrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  check_errorsrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  numeln_binsratio	bit_widths        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  chunksrz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  minmaxrK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  rK  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  
correctionfweightsaweightss       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rwith_replacements      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )realimags     r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r2   c                      yrF  rH  )absangs     r/   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r2   c                      yrF  rH  )rJ  ords     r/   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>'  rM  r2   c                      yrF  rH  )rJ  padrg  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r2   c                      yrF  rH  rJ  r  r  r  r  dilationr  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r2   c                      yrF  rH  r,  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r2   c                      yrF  rH  r,  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r2   c	                      yrF  rH  )	rJ  r  r  r  r  r-  
transposedoutput_addingr  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r2   c                      yrF  rH  )rJ  r  r  r*  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r2   c                      yrF  rH  rJ  r  r  r  r  output_paddingr  r-  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r2   c                      yrF  rH  r5  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r2   c                      yrF  rH  r5  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r2   c                      yrF  rH  r  r  r  marginr  r  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r2   c                      yrF  rH  )r  r  rz  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r2   c                      yrF  rH  rJ  r\  rz  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r2   c                      yrF  rH  rB  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r2   c                      yrF  rH  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r/   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r2   c                      yrF  rH  rJ  rz  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r2   c                      yrF  rH  rM  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r2   c                      yrF  rH  rJ  rz  rK  rb   s       r/   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r2   c                      yrF  rH  rP  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r2   c                      yrF  rH  yxrz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r2   c                      yrF  rH  rM  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r2   c                      yrF  rH  rJ  diagonalrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r2   c                      yrF  rH  r]  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r2   c                      yrF  rH  )rJ  offsets     r/   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r2   c                      yrF  rH  )rJ  nrz  prependappendrK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r2   c                      yrF  rH  rJ  ra  dim1dim2s       r/   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r2   c                      yrF  rH  rg  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r2   c                      yrF  rH  )rJ  srcra  rh  ri  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r2   c                      yrF  rH  )r  rl  rv  r  storage_offsets        r/   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r2   c                      yrF  rH  )rJ  r\  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r2   c                      yrF  rH  rJ  r\  rounding_moderK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r2   c                      yrF  rH  rr  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r2   c                      yrF  rH  rJ  rl  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r2   c                      yrF  rH  )rk  rl  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r2   c                      yrF  rH  rJ  indices_or_sectionss     r/   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r2   c                      yrF  rH  rJ  UPLOrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r2   c                      yrF  rH  )equationoperandss     r/   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r2   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  r2   c
                      yrF  rH  )
rJ  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r/   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r2   c                      yrF  rH  rJ  rb   rp   ra   requires_grads        r/   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r2   c                      yrF  rH  rJ  r\  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r2   c                      yrF  rH  )rJ  scale
zero_pointaxis	quant_min	quant_maxs         r/   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r2   c                      yrF  rH  )rU  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  r2   c                      yrF  rH  rJ  packed_weightr  outputs       r/   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r2   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  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r2   c                      yrF  rH  )rJ  abs      r/   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r2   c                      yrF  rH  rJ  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r2   c                      yrF  rH  rJ  rc  rz  norms       r/   rL  z'get_testing_overrides.<locals>.<lambda>}  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  srz  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  	start_dimend_dims      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  dimss     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  exponentrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  
fill_valuerK  rb   rp   ra   r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  	dep_tokens      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  indexrK  sparse_grads        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rr  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  spacingrz  
edge_orders       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  gridinterpolation_modepadding_moderw  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  
num_groupsr  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c	                      yrF  rH  	rJ  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  w_ihw_hhb_ihb_hhs         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  lambds     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  r  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  valuesrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r=  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  binsr  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r'  r  r  r  densityrK  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r'  r   r  r)  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  taus     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rk  rl  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r{  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rz  r  sources       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r6  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  indicesr#  
accumulates       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  r  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  r  rg  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  r  r7  r  include_inputs         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r\   s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )eteassume_uniqueinverts       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rA  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rA  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c	                      yrF  rH  )	rJ  r  r  r  r  use_input_statsr  r  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r|  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   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  r2   c                      yrF  rH  rJ  r  r  r  r  
log_targets         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  krz  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  	hermitianr   rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rm  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )LDpivotsBrm  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  normalized_shaper  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  endr  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rk  rr  Xrc  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rz  rb   s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rU  rT  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  namesr  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c	                      yrF  rH  )	databatch_sizesr  r  r  r  r  r  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )Apivot	get_infosrK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r<  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  maskrg  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r7  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r2   c                      yrF  rH  rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r   rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )LUrq  rr  leftadjointrK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rc  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rc  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rm  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r-  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r-  return_indicesr  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  r  rb   rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r2   c                       yrF  rH  )r  r&   s     r/   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  exponential_average_factorepsilons           r/   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r2   c	                      yrF  rH  	rJ  r  r  r  r  r-  r  	benchmarkdeterministics	            r/   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r2   c	                      yrF  rH  )	rJ  r  zra  r  r  r  r-  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r2   c                      yrF  rH  r,  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r2   c
                      yrF  rH  )
rJ  r  r  r  r6  r  r-  r  r  r  s
             r/   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r2   c	                      yrF  rH  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r2   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  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r2   c                      yrF  rH  rJ  r7  destinations      r/   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>9  rM  r2   c                      yrF  rH  )rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r2   c                      yrF  rH  )rJ  num_samplesreplacementrK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r2   c                      yrF  rH  )rJ  ro  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r2   c                      yrF  rH  rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r2   c                      yrF  rH  )rJ  rz  startrd  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r2   c                      yrF  rH  )rJ  nanposinfneginfrK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  r  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r2   c                      yrF  rH  rJ  rt  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r2   c                      yrF  rH  rJ  rt  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r2   c                      yrF  rH  )rJ  r  r  NCHxWgroupr  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r2   c                      yrF  rH  )rJ  r  rz  r  rb   s        r/   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r2   c                      yrF  rH  rP  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r2   c                      yrF  rH  rP  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r2   c                      yrF  rH  rJ  rQ  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r2   c                      yrF  rH  rt  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r2   c                      yrF  rH  rJ  r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r2   c                      yrF  rH  rJ  r  r  r  r  r  divisor_overrides          r/   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  r  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r2   c                      yrF  rH  rJ  r  r  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r2   c                      yrF  rH  r<  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r2   c                      yrF  rH  )rJ  r  r  r  ignore_indexr  r  label_smoothings           r/   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r2   c	                      yrF  rH  )	rJ  linear_weightr  linear_biasr  r  r  r  optionss	            r/   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r2   c                      yrF  rH  rE  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r2   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  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rQ  r  r-  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  rQ  output_ratior  _random_sampless         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r.  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r.  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r.  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  varr   r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  approximates     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  rw  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )logitsr-  hardr  rz  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  min_valmax_valr  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r%  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  rL  r  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rv  scale_factorr  rw  recompute_scale_factor	antialiass          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rg  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  negative_sloper  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rv  ra  rb  rk  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  rz  _stacklevelrb   s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rU  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rU  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r<  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r-  r  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rZ  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r:  r  r  r  rQ  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ra  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ra  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   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  r2   c                      yrF  rH  )rJ  r  r  r=  r  r  r  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  r  r  s          r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rz  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r\   num_classess     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r*  r  rg  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  r  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  	log_inputr   r  r  r  r  s           r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  lowerr  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rf  rg  rg  rt  ro  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  r  rb  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  deltar  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rQ  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rQ  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rb  	thresholds      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  rg  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c
                      yrF  rH  
anchorpositivenegativer=  r  r  swapr  r  r  s
             r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )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  r2   c                      yrF  rH  )rJ  r  r-  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r\   r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r\   r  stdr  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r\   vals     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r\   r  r  nonlinearityr  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  as_tuples     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )r  c                     yrF  rH  )rJ  rv  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  rz  r  rK  rb   s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r$  rz  r  rK  rb   s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      	 yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  s     13 r2   c                      yrF  rH  )vpowrz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r,  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  input3r  	transposes        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  rz  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  qra  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r2   c                      yrF  rH  )rJ  rconds     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rm  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  upscale_factors     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  downscale_factors     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r   r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rb   s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r7  r;  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  somerK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  r  rz  r  interpolationrK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  )rJ  scaleszero_pointsr  rb   s        r/   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r2   c                      yrF  rH  )rJ  r  r  rb   s       r/   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r2   c                      yrF  rH  )rJ  rb   reduce_ranges      r/   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r2   c                      yrF  rH  )rJ  r  r  r  r5  r  output_scaleoutput_zero_points           r/   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r2   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  r2   c                      yrF  rH  r  s                 r/   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r2   r      c                      	 yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>,  s
     " r2   c                      	 yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>1  s     !# r2   c                      	 yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>7  s     !#	 r2   c                      yrF  rH  r  s                 r/   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r2   c                      yrF  rH  r  s                 r/   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r2   c                      yrF  rH  rB  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r2   c                      yrF  rH  )rJ  r  rz  maxnormrK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r2   c                      yrF  rH  )rJ  shapes     r/   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r2   c	                      yrF  rH  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r2   c	                      yrF  rH  r  s	            r/   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r2   c                      yrF  rH  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r2   c                      yrF  rH  )rJ  shiftsr  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r2   r   r  c                      yrF  rH  )rJ  rk  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r2   c                      yrF  rH  )r  r  compressed_indices_dtypes      r/   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r2   c                      yrF  rH  )rJ  r\  ra  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r2   c                      yrF  rH  rj  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r2   )rg  r  c                     yrF  rH  )rJ  rz  r  rl  rg  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>^  rM  r2   c                      yrF  rH  )rJ  rz  r  rl  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>_  rM  r2   c                      yrF  rH  )rJ  rz  r  rl  r  include_selfs         r/   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r2   c                      yrF  rH  )sorted_sequencerJ  r  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>a  rM  r2   c                      yrF  rH  )r  r  lengthsr:  r  r  unsafes          r/   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r2   c                      yrF  rH  )rJ  rz  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>c  rM  r2   c                      yrF  rH  )rJ  rl  rz  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r2   c                      yrF  rH  rJ  rl  rz  r  rz  steps         r/   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r2   c                      yrF  rH  r1  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>o  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>p  rM  r2   c                      yrF  rH  rx  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r2   c                      yrF  rH  rx  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r2   c                      yrF  rH  )r  rr  r  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r2   c                      yrF  rH  )r  rr  r  r   rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>u  rM  r2   )r  rK  c                     yrF  rH  )rJ  rz  r  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r2   c                      yrF  rH  r\   split_size_or_sectionsrz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r2   c                      yrF  rH  rE  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r2   c                      yrF  rH  rM  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r2   c                      yrF  rH  rj  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>}  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r]  r^  r_  r`  ra  pad_moderb  rc  re  align_to_windows              r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rX  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  cs      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r%   s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  
compute_uvrK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  Ms       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  full_matricesrK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rJ  dim0rh  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  axis0axis1s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r[  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r\  rK  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r:  rz  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  inds     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  rK  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r|  rz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rk  rz  r  rK  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rV  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rq  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rS  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rS  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  r  unitriangulars        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rr  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r]  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c
                      yrF  rH  r  s
             r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r]  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rI  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  rz  sizess      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  sortedreturn_inversereturn_countsrz  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rJ  r  r  rz  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r:  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r	  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rE  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rE  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  rM  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )rU  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  ry  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r{  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )	conditionrU  rT  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   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  r2   c                      yrF  rH  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  levels     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )primaltangentr  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rv  r  rn  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>	  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>
  rM  r2   c                      yrF  rH  )r  ra  rh  ri  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )implicitc                     yrF  rH  )r  rv  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rz  r  rd  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  rv  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rz  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rz  r  rz  r2  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  
split_sizerz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  split_sizesrz  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rr  rh  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>!  rM  r2   c                      yrF  rH  r  rb   s     r/   rL  z'get_testing_overrides.<locals>.<lambda>"  rM  r2   c                      yrF  rH  r  	dimensionrv  r2  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>#  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>$  rM  r2   c                      yrF  rH  r  r\  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>%  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>&  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>'  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>9  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>:  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>;  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda><  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>=  rM  r2   c                      yrF  rH  )r  arrays     r/   rL  z'get_testing_overrides.<locals>.<lambda>>  rM  r2   c                      yrF  rH  )r  idxs     r/   rL  z'get_testing_overrides.<locals>.<lambda>?  rM  r2   c                      yrF  rH  )r  memos     r/   rL  z'get_testing_overrides.<locals>.<lambda>@  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>A  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>B  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>C  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>D  rM  r2   c                      yrF  rH  )r  format_specs     r/   rL  z'get_testing_overrides.<locals>.<lambda>E  rM  r2   c                      yrF  rH  )r  protos     r/   rL  z'get_testing_overrides.<locals>.<lambda>F  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>G  rM  r2   )tensor_contentsc                     yrF  rH  )r  r:  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>H  rM  r2   c                      yrF  rH  )r  rk  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>I  rM  r2   c                      yrF  rH  )r  ds     r/   rL  z'get_testing_overrides.<locals>.<lambda>J  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>K  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>L  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>M  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>N  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>O  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>P  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>Q  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>R  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>S  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>T  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>U  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>V  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>W  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>X  rM  r2   c                      yrF  rH  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r/   rL  z'get_testing_overrides.<locals>.<lambda>Y  rM  r2   c                      yrF  rH  )r  rN  rO  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>Z  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>[  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>\  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>]  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>^  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>_  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>`  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>a  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>b  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>c  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>d  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>e  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>f  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>g  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>h  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>i  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>j  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>k  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>l  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>m  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>n  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>o  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>p  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>q  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>r  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>s  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>t  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>u  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>v  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>w  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>x  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>y  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>z  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>{  rM  r2   c                      yrF  rH  )r  rb   non_blockingr&   s       r/   rL  z'get_testing_overrides.<locals>.<lambda>|  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>}  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>~  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  callables     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  gradientretain_graphcreate_graphrS  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  ri   s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )r  c                     yrF  rH  )r  mediansigmar  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  	coalesceds     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rl  ru  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rl  ra  rh  ri  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  ambiguity_checks     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                     yrF  rH  )r  r   r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  rg  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                     yrF  rH  )r  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r\   s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                     yrF  rH  )r  r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r\   r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rU  rT  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rl  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r\  assigns      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  rd  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r:  r\   r;  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                     yrF  rH  )r  from_tor  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  streams     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  hooks     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r7  rn  rv  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rl  rz  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rl  rz  r  rz  r2  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  accumulate_matchess      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  size1size2	dense_dims       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rk  rl  rb  ra  rK  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  repss     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  rb   ru  copyri   s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   )masked_gradc                     yrF  rH  r  rb   r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s     r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  max_version	dl_devicer  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r  s      r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  r  r   drivers       r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   c                      yrF  rH  )r  ra   ru  r&   s       r/   rL  z'get_testing_overrides.<locals>.<lambda>   rM  r2   c                      yrF  rH  r  s    r/   rL  z'get_testing_overrides.<locals>.<lambda>  rM  r2   is______i__rbitwise_c                      yrF  rH  )r\   rl  r  async_op	group_srcs        r/   rL  z'get_testing_overrides.<locals>.<lambda>(  rM  r2   c                      yrF  rH  )r\   opr  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>)  rM  r2   c                      yrF  rH  )r\   dstr  r  r  	group_dsts         r/   rL  z'get_testing_overrides.<locals>.<lambda>*  rM  r2   c                      yrF  rH  )r  r  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>+  rM  r2   c                      yrF  rH  )tensor_listr\   r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>,  rM  r2   c                      yrF  rH  )output_tensorinput_tensorr  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>-  rM  r2   c                      yrF  rH  )output_tensor_listsinput_tensor_listr  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>.  rM  r2   c                      yrF  rH  )r\   gather_listr  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>/  rM  r2   c                      yrF  rH  )r\   scatter_listrl  r  r  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>0  rM  r2   c                      yrF  rH  )r  
input_listr  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>1  rM  r2   c                      yrF  rH  )r  rJ  r  r  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>2  rM  r2   c                      yrF  rH  )r  rJ  output_split_sizesinput_split_sizesr  r  s         r/   rL  z'get_testing_overrides.<locals>.<lambda>3  rM  r2   c                      yrF  rH  )output_tensor_listr&  r  r  s       r/   rL  z'get_testing_overrides.<locals>.<lambda>4  rM  r2   c                      yrF  rH  r\   r  r  tagr  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>5  rM  r2   c                      yrF  rH  r\   rl  r  r5  r  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>6  rM  r2   c                      yrF  rH  r4  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>7  rM  r2   c                      yrF  rH  r7  s        r/   rL  z'get_testing_overrides.<locals>.<lambda>8  rM  r2   N)r  r  N)r  N)h㈵>:0yE>F)F)NFN)rG  F)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)       @#use_mm_for_euclid_dist_if_necessary)r  F)FN)NNN)r  NN)   F)Nr  r   r  r  )Nr  r   r   r  r  )r   NNr  )r  r<  )rG  N)r   r  FrF  )r  rG  NNN)r   r   r  )r   rG  )r?  )LN)NNr=  FF)Nr?  Fr  FNN)NNNF)FF)NrG  N)Nr@  rG  N)r   rG  )TT)NF)NNr  )NNr;  T)      ?)NFr   N)r  NNr  )d   r   r   N)rD  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   r  F)Nr   r  FF)NFNN)rG  FN)        NNN)NNr;  )Nr;  )r?  NFN)rC  FF)Nr   FTN)NNF皙?r;  )NNNr  )NNNr  rE  )NNr  NrE  N)rC  TF)	NNr?  Fr  FNFN)r  r   r  )NNFN)Fư>r  )none)rG  )bilinearr   N)r  Fg|=rG  )g      r  F)NNNNTrF  r;  )NNnearestNNF)NNr  N)g{Gz?F)g-C6?g      ?r  )Nr4   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  )NNrG  Nr  )r?  r  g-q=N)r   r   )r=  rH  F)TFNr<  Nr  )NrH  )g      ?gUUUUUU?FF)NrE  )NNr  r  )r  r  N)r     )r  r?  rH  FNNr  )rE  r  N)r   fan_in
leaky_reluN)froNFNN)NNFNN)r?  NFNN)rO  rB  FNN)r?  r   )TF)NTr?  )V瞯<)rP  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  )	NNNTreflectFTNN)TTN)   r?  N)r?  N)TFF)TFFN)r  r?  )Nr   NN)NNNN)NNNFN)NNr   N(  r6   r7   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	bernoullirJ   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linear_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_lossrN  rR  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   r5  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  ra   rb   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizerp   namenbytesndim	output_nrr  r  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_valuesapply_rr   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipuconst_data_ptrdata_ptrr  rz  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalfr:  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rj   random_record_streamregister_hook"register_post_accumulate_grad_hook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_storagern  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_singleall_gather_coalescedreduce_scatterreduce_scatter_singleall_to_all_single
all_to_allisendirecvsendrecv)r7   retprivateuse1_backend_nameret2ignoredrk  r  r  subnamer$
  r    r  s               r/   r   r     s%Y   6 \\Fx%		-x%2x% 	!!#@x% 	!!#A	x%
 	

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

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

.+x%, 	

.-x%. 	J/x%0 	/1x%2 			F3x%4 	&5x%6 	&7x%8 	RR9x%: 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Emx%n 	<ox%p 	=qx%r 	Asx%t 			4ux%v 	9wx%x 	Uyx%z 	1{x%| 	+}x%~ 	:x%@ 	WAx%B 	!sCx%D 	&&(_Ex%F 	8Gx%H 	!fIx%J 	YKx%L 	!VMx%N 	UOx%P 	$$&>Qx%R 	3Sx%T 	:Ux%V 			-Wx%X 	2Yx%Z 	:[x%\ 	//1N]x%^ 	//1N_x%` 	//1dax%b 	<<>qcx%d 	//1dex%f 	<<>qgx%h 	//1dix%j 	<<>qkx%l 	'')Smx%n 	))+aox%p 	&& Bsx%v 	&& Byx%| 	&&xx%B 	$$&RCx%D 	00cGx%J 	<<tMx%P 	  "LQx%R 	11iUx%X 	)) L[x%^ 	00 [ax%d 	$$xgx%j 	##%Zkx%l 	%%'\mx%n 	%%'\ox%p 	%%'\qx%r 	!Ksx%t 	%%|wx%z 	)) J}x%@ 	113iAx%B 	  "mCx%D 	11zGx%J 	>>zMx%P 	11zSx%V 	>>zYx%\ 	--/u]x%^ 	  "F_x%` 	!9ax%b 	'')zcx%d 	&&(gex%f 	**,cgx%h 	&&(Cix%j 	$$&`kx%l 	00box%r 	)) Iux%x 	'' M{x%~ 	""  %Ax%@ 	##%|Ax%B 	&&(mCx%D 	&&(\Ex%F 	""$GGx%H 	//1gIx%J 	'')^Kx%L 	&&(8Mx%N 	%%'mOx%P 	%%'mQx%R 	%%'mSx%T 	//iWx%Z 	&&t]x%` 	33tcx%f 	&&tix%l 	33tox%r 	&&tux%x 	33t{x%~ 	((*zx%@ 	((*zAx%B 	((*zCx%D 	$$&}Ex%F 	88 _Ix%L 	--tOx%R 	22VUx%X 	77c[x%^ 	$$vax%d 	%%'Xex%f 	##%Fgx%h 	!Pix%j 	--/akx%l 	,,}ox%r 	!!#;sx%t 	  "Aux%v 	!!#Bwx%x 	$$&_yx%z 	!!#y{x%| 	  "A}x%~ 	  "Ax%@ 	  "AAx%B 	88:uCx%D 	**  -AEx%F 	&&(jGx%H 	,,.xIx%J 	##%ZKx%L 	##%ZMx%N 	$$&LOx%P 	&&(CQx%R 	$$&6Sx%T 	&&(8Ux%V 	%%'XWx%X 	// L[x%^ 	==DHQT[`lrvax%d 	""$bex%f 	 Ogx%h 	Six%j 	!7kx%l 	&&(xmx%n 	7ox%p 	Fqx%r 	(sx%t 	

\ux%v 	dwx%x 	  "hyx%z 	   +/#3{x%B 	9Cx%D 	dEx%F 	%Gx%H 	*Ix%J 	QKx%L 	!SMx%N 	+Ox%P 	IQx%R 	*Sx%T 	5Ux%V 	IWx%X 	=Yx%Z 	A[x%\ 	7]x%^ 	 Y_x%` 	6ax%b 	2cx%d 	-ex%f 	dgx%h 			7ix%j 	

0kx%l 			Dmx%n 	  "2ox%p 	""$4qx%r 	'')9sx%t 	'ux%v 	,wx%x 	7yx%z 	C{x%| 	f}x%~ 	ix%@ 	""$VAx%B 	!!#MCx%D 	))+PEx%F 	""$sGx%H 	   aKx%N 	!! aQx%T 	""J#Wx%^ 	""L #ax%j 	""O 	#mx%x 	%% a{x%~ 	%% aAx%D 	1Ex%F 	%Gx%H 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

u/D/DHix%j 	0E0EIkx%l 	0E0EImx%n 	

u/D/DHox%p 	

u/D/DHqx%r 	sx%t 	ux%v 	/wx%x 	!Oyx%z 	

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

u/D/DH]x%^ 	__x%` 	oax%b 	_cx%d 	/ex%f 	1gx%h 	/ix%j 	_kx%l 	MTMmx%n 	0ox%p 	0E0EIqx%r 	6sx%t 	5ux%v 			8wx%x 	@yx%z 	E{x%| 	?}x%~ 	x%@  	""OA x%B  	--C x%D  	%%E x%F  	G x%H  	oI x%J  	,K x%L  	?M x%N  	GO x%P  	Q x%R  	LDLS x%T  	5U x%V  	3W x%X  	113HY x%Z  	-[ x%\  	B] x%^  	1_ x%`  	-a x%b  	-c x%d  	0e x%f  	  "8g x%h  	Oi x%j  	[k x%l  	?m x%n  	oo x%p  	1F1FJq x%r  	_s x%t  	Wu x%v  	?w x%x  	1y x%z  	&&(W{ x%|  	G} x%~  	'')Q x%@! 	OA!x%B! 	C!x%D! 	E!x%F! 	G!x%H! 	_I!x%J! 	1K!x%L! 	+M!x%N! 			EUZUjUjnO!x%P! 	IIQ!x%R! 	GS!x%T! 	/U!x%V! 	W!x%X! 	/Y!x%Z! 	.[!x%\! 	=]!x%^! 	7_!x%`! 	a!x%b! 	+c!x%d! 	.e!x%f! 	og!x%h! 	di!x%j! 	  /+Fo!x%Cv! 	((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	
 ![ &&(j ))+x v x ##%g **,i &&  )N !m 

Z 

Z  		Y!" 		Y#	
, Jr2   
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>.wrappedZ  s^    &77M!-0,"b&)73]EIMS  (((r2   )	functoolsr   r   r%   r&   r   r   r   )r    r
  r
  s   `@r/   r#  z"wrap_torch_function.<locals>.innerY  sS    			)277 	)bii 	)B 	) 
	) HRV$g..r2   )r   r   r   )r
  r#  s   ` r/   r   r   ?  s/    4/HRV$ /"b&)9 / Lr2   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*
  r6   _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
  i  s    J  88..0	"%%!#O (s# ,,"67++8899:
   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%;(< r2   
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.r?  
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
  r6   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 )Nr6   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listr6   __all__r   r   dirr   r7   r  r   rq	  r
  endswithislowerr
  object
isinstancer
  
ModuleType
__future___Featurer  r
  r-  __set__r   r   AssertionErrorformatr
  re  )
overridable_funcsr  tested_namespacesnamespace_str	namespacens_funcsr
  r(   r    r  s
             r/   _get_overridable_functionsr
  B  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##r2   c                      t               d   S )a  List functions that are overridable via __torch_function__

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

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

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

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

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

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

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

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

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

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

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

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

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

    >>> class TensorLike:
    ...     @classmethod
    ...     def __torch_function__(cls, func, types, args, kwargs):
    ...         return -1
    >>> is_tensor_like(TensorLike())
    True
    r  )r*
  r6   r7   r
  )inps    r/   r   r     s%    D 9$J5I(JJr2   c                   H    e Zd ZU dZd ed<   d	dZd
dZd Zd Ze	d        Z
y)TorchFunctionModea  
    A ``TorchFunctionMode`` allows you to override the meaning of all
    ``__torch_function__`` 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    r2   c                     t         r:  )NotImplementedErrorr  r    r
  r%   r&   s        r/   r  z$TorchFunctionMode.__torch_function__%  s    !!r2   c                     t        |        | S r:  )
_push_moder  s    r/   	__enter__zTorchFunctionMode.__enter__(  s    4r2   c                     t                y r:  )	_pop_mode)r  exc_typeexc_valexc_tbs       r/   __exit__zTorchFunctionMode.__exit__,  s    r2   c                 D    t        j                  dd        | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`r?  r
  )r+   r
  )clsr%   r&   instances       r/   pushzTorchFunctionMode.push/  s*    ^	
 ''r2   )r#   NrH  N)r
  r
  __qualname____doc____annotations__r   r  r	  r  classmethodr  rH  r2   r/   r  r     s7    < "  r2   r  c                  @    t               } | dkD  rt        | dz
        S d S )Nr   r  )r   r   )	stack_lens    r/   r
  r
  9  s%    )+I4=M!)a-0KtKr2   c                  d    t               } t        |       D cg c]  }t        |       c}S c c}w r:  )r   r   r   )r  r
  s     r/    _get_current_function_mode_stackr  >  s)    )+I/4Y/?@!"1%@@@s   -c                     t        |        y r:  )r   )r  s    r/   r  r  C  s
    !$'r2   c                      t               } | S r:  )r   olds    r/   r  r  G  s    
#
%CJr2   c               #   `   K   t               } 	 |  t        |        y # t        |        w xY wwr:  )r  r  r  s    r/   r
  r
  L  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__V  s    >FT$V$$r2   r  )r
  r
  r  r  rH  r2   r/   r#  r#  U  s    %r2   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:  )r6   r
  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r/   _enable_torch_functionr+  \  sf     224I6**588+G+G+O+OP**95**95s   B+AB ' B+!B((B+c               #      K   t         j                  j                         5  	 d  	 d d d        y # w xY w# 1 sw Y   y xY wwr:  )r6   r
  _RestorePythonTLSSnapshotrH  r2   r/   r   r   f  sC      
	+	+	- 		  	 s#   A61	A36?Ac                 F    t         j                  j                  | |||      S )a
  Skip one level of ``__torch_function__`` dispatch and call the function.

    This is primarily useful for **Tensor subclasses** that want to call into
    a function's implementation while still intercepting PyTorch operations
    inside that function.

    Example with Tensor subclass. Only ops whose inputs include a
    ``LoggingTensor`` are intercepted; once ``redispatch_function`` returns
    a plain ``torch.Tensor``, subsequent ops (here ``+ 1``) are not logged.

        >>> from torch.overrides import has_torch_function, handle_torch_function
        >>> class LoggingTensor(torch.Tensor):
        ...     depth = 0
        ...
        ...     @classmethod
        ...     def __torch_function__(cls, func, types, args, kwargs=None):
        ...         print(f"{'  ' * cls.depth}Calling {func.__name__}")
        ...         cls.depth += 1
        ...         r = torch.overrides.redispatch_function(func, types, args, kwargs)
        ...         cls.depth -= 1
        ...         return r
        >>> def scaled_mul(a, b):
        ...     if has_torch_function((a, b)):
        ...         return handle_torch_function(scaled_mul, (a, b), a, b)
        ...     return a * b + 1
        >>> x = LoggingTensor(torch.tensor([3.0]))
        >>> y = LoggingTensor(torch.tensor([4.0]))
        >>> result = scaled_mul(x, y)
        Calling scaled_mul
          Calling mul
        >>> result
        tensor([13.])

    Note that only ``mul`` is logged, not ``add``: ``redispatch_function``
    returns a plain ``torch.Tensor``, so the ``+ 1`` inside ``scaled_mul``
    no longer sees a ``LoggingTensor`` input and ``__torch_function__`` is
    not triggered.

    With ``TorchFunctionMode`` the mode stays active across all inner ops,
    so the ``+ 1`` is now visible too.  Use ``with self:`` after
    ``redispatch_function`` to re-enable the mode for those inner calls.

        >>> from torch.overrides import TorchFunctionMode
        >>> class LoggingMode(TorchFunctionMode):
        ...     def __init__(self):
        ...         self.depth = 0
        ...
        ...     def __torch_function__(self, func, types, args, kwargs=None):
        ...         print(f"{'  ' * self.depth}Calling {func.__name__}")
        ...         self.depth += 1
        ...         with self:
        ...             r = torch.overrides.redispatch_function(
        ...                 func, types, args, kwargs
        ...             )
        ...         self.depth -= 1
        ...         return r
        >>> a = torch.tensor([3.0])
        >>> b = torch.tensor([4.0])
        >>> with LoggingMode():
        ...     result = scaled_mul(a, b)
        Calling scaled_mul
          Calling mul
          Calling add
        >>> result
        tensor([13.])
    )r6   r
  _skip_one_hop_torch_function)r    r
  r%   r&   s       r/   r   r   v  s     F 8800udFKKr2   )z.*is deprecated, please use.*r6   r:  )Er  r
  r
  
contextlibr
  r:  r
  r+   collections.abcr   r   r   typingr   r   r   typing_extensionsr	   r6   torch._Cr
   r   r   r   r   r   r   r   r   r
  r   r   r
  r1   cacher
  r   rC  dictr   r   r*
  r
  r
  r%   r&   r   r   r   r   r
  r
  r   r   r
  r:
  r   r   r  r
  r  r  r  contextmanagerr
  r#  r+  r   r   rH  r2   r/   <module>r8     s  ,     
   .  % % ' 
 
 
 t_T]
 1 
2r6
     b"f	 F _s8} _  _D	 c(m  2 UtHh$67 U  Up$'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  CLr2   