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

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

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

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

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

    Returns
    -------
    function
        The wrapped function.
    c                      t        j                         5  t        j                  dt                | i |cd d d        S # 1 sw Y   y xY w)Nignore)categorymessager"   )warningscatch_warningsfilterwarningsUserWarning)argskwargsr    r"   r!   s     N/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/overrides.pywrapperz'_disable_user_warnings.<locals>.wrapperK   sF    $$& 	)##H{EZ`a((	) 	) 	)s   %AAr   )r    r!   r"   r0   s   ``` r/   _disable_user_warningsr1   6   s"    * 4[) ) N    c                  p#   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                  j                  t         j                  t         j                  j                  t         j                  j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  t         j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                   t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j
                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                  t         j                  j                  j                   t         j                  j                  j"                  t         j                  j                  j$                  t         j                  j                  j&                  t         j                  j                  j(                  t         j                  j                  j*                  t         j,                  j.                  t        t
        t         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         j                  j                  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                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  j                  | j                  j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  | j.                  S )a%  
    Return public functions that cannot be overridden by ``__torch_function__``.

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

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

} 	

} 	

} 	} 	

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

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

o}p 	

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

K}L 	M}N 	

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

{}| 	

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

e}f 	g}h 	i}j 	

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

    Not ALL property accessors have this property; for example ``Tensor.T`` actually
    just creates a new transposed tensor on the fly, and so we SHOULD interpose on
    these calls (you need to check the implementation of the function to see if
    this is the case or not).  Additionally, if a property accessor doesn't return a Tensor,
    it doesn't have to be on this list (though it is harmless if it is).
    )r4   r5   _baser$  grad_gradr.  s    r/   get_default_nowrap_functionsr3  h  s>    $ \\F r2   c                  J   t         j                  } i t         j                  ddt         j                  ddt         j                  d t         j
                  d t         j                  ddt         j                  d t         j                  ddt         j                  dd	t         j                  dd
t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                   ddt         j"                  ddt         j$                  d i t         j&                  ddt         j(                  ddt         j*                  ddt         j,                  ddt         j.                  ddt         j0                  ddt         j2                  ddt         j4                  ddt         j6                  d t         j8                  d t         j:                  ddt         j<                  ddt         j>                  d  t         j@                  dd!t         jB                  dd"t         jD                  dd#t         jF                  dd$i t         jH                  dd%t         jJ                  dd&t         jL                  dd't         jN                  dd(t         jP                  dd)t         jR                  d* t         jT                  d+ t         jV                  d, t         jX                  dd-t         jZ                  dd.t         j\                  d/ t         j^                  d0 t         j`                  d1 t         jb                  d2 t         jd                  d3 t         jf                  d4 t         jh                  d5 i t         jj                  d6 t         jl                  dd7t         jn                  d8 t         jp                  	 dd:t         jr                  dd;t         jt                  dd<t         jv                  dd=t         jx                  dd>t         jz                  dd?t         j|                  dd@t         j~                  ddAt         j                  ddBt         j                  dC t         j                  ddDt         j                  dE t         j                  dF t         j                  ddGi t         j                  dH t         j                  ddIt         j                  ddJt         j                  ddKt         j                  ddLt         j                  ddMt         j                  ddOt         j                  ddPdQt         j                  dR t         j                  ddSt         j                  j                  ddTt         j                  j                  ddUt         j                  ddVt         j                  ddWt         j                  dX t         j                  d
dYt         j                  ddZi t         j                  dd[t         j                  dd\t         j                  dd]t         j                  dd^t         j                  dd_t         j                  d` t         j                  ddat         j                  db t         j                  ddct         j                  dd t         j                  j                  ddet         j                  ddft         j                  ddgt         j                  ddht         j                  ddit         j                  d
djt         j                  ddki t         j                  ddlt         j                  ddmt         j                  dn t         j                  d
dot         j                  ddpt         j                  ddqt         j                  ddrt         j                  ds t         j                  ddtt         j                  ddut         j                  ddvt         j                  ddwt         j                  dx t         j                  ddyt         j                  j                  dd{t         j                  	 dd|t         j                  dd}i t         j                  dd~t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  j                  d t         j                   d t         j                  ddt         j                  ddt         j                  d
dt         j                  ddt         j
                  ddt         j                  j
                  ddt         j                  ddi t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                   d t         j"                  ddt         j                  j$                  ddt         j                  j&                  ddt         j                  j(                  ddt         j                  j*                  ddt         j,                  d t         j.                  	 ddi t         j0                  	 ddt         j2                  ddt         j4                  ddt         j6                  d t         j8                  ddt         j:                  ddt         j<                  ddt         j>                  ddt         j@                  ddt         jB                  ddt         jD                  d t         jF                  d t         jH                  	 ddt         jJ                  d t         jL                  d t         jN                  d t         jP                  d i t         jR                  d t         jT                  d t         jV                  d t         jX                  d t         jZ                  d t         j\                  j^                  ddt         j\                  j`                  ddt         j\                  jb                  ddt         j\                  jd                  ddt         j\                  jf                  ddt         j\                  jh                  ddt         j\                  jj                  ddt         j\                  jl                  ddt         j\                  jn                  ddt         j\                  jp                  ddt         j\                  jr                  ddt         j\                  jt                  ddi t         j\                  jv                  ddt         j\                  jx                  ddÄt         j\                  jz                  ddĄt         j\                  j|                  ddńt         j\                  j~                  ddƄt         j\                  j                  ddǄt         j\                  j                  ddȄt         j\                  j\                  ddɄt         j                  ddʄt         j                  dd˄t         j                  d̄ t         j                  d̈́ t         j                  d΄ t         j                  ddτt         j                  ddЄt         j                  dф t         j                  dd҄i t         j                  ddӄt         j                  ddԄt         j                  ddՄt         j                  ddt         j                  ddfdքt         j                  dׄ t         j                  dd؄t         j                  ddلt         j                  ddڄt         j                  ddۄt         j                  dd܄t         j                  dd݄t         j                  ddބt         j                  dd߄t         j                  ddt         j                  ddt         j                  ddt         j                  d i t         j                  d t         j                  d t         j                  ddt         j                  d t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  j                  d t         j                  ddt         j                  d t         j                  ddi t         j                  ddt         j                  ddt         j                  ddt         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddt         j                  d t         j                  ddt         j                  d t         j                  dd t         j                  d t         j                  d t         j                  dѐdt         j                  dѐdt         j                  d i t         j                  d t         j                   dѐdt         j                  j                  dѐdt         j                  j                  dߐd	t         j                  d
 t         j                  d t         j
                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  dԐdt         j                  d t         j                  	 d dt         j                  ddi t         j                   d t         j"                  d֐dt         j                  j$                  dېdt         j                  j&                  dߐdt         j                  j(                  dߐdt         j*                  ddt         j,                  dѐdt         j.                  dѐdt         j0                  dѐdt         j2                  dѐd t         j4                  dѐd!t         j6                  dѐd"t         j8                  	 dd#t         j:                  dѐd$t         j<                  dѐd%t         j>                  dѐd&t         j@                  dѐd'i t         jB                  dѐd(t         jD                  dѐd)t         jF                  dѐd*t         jH                  d+ t         jJ                  dѐd,t         jL                  dѐd-t         jN                  dѐd.t         jP                  dѐd/t         jR                  dѐd0t         jT                  dѐd1t         jV                  dߐd2t         jX                  d3 t         jZ                  dؐd4t         j\                  dѐd5t         j^                  dѐd6t         j`                  dd7t         jb                  dѐd8i t         jd                  dd9t         jf                  d: t         jh                  d; t         jj                  dѐd<t         jl                  dѐd=t         j                  j`                  dd>t         j                  jn                  dd?t         j                  jp                  dd@t         j                  jb                  ddAt         j                  jl                  dѐdBt         jr                  dC t         j                  jr                  dѐdDt         j                  jt                  ddEt         j                  jv                  dѐdFt         jx                  dG t         j                  jx                  dH t         jz                  dѐdIi t         j|                  dѐdJt         j~                  dѐdKt         j                  ddLt         j                  ddMt         j                  ddNt         j                  	 ddOt         j                  dѐdPt         j                  ddQt         j                  dѐdRt         j                  dѐdSt         j                  dT t         j                  dѐdUt         j                  dѐdVt         j                  dѐdWt         j                  dX t         j                  dY t         j                  dZ i t         j                  d[ t         j                  d\ t         j                  d] t         j                  d^ t         j                  dѐd_t         j                  dd`t         j                  da t         j                  db t         j                  dߐdct         j                  dѐddt         j                  dѐdet         j                  dߐdft         j                  dѐdgt         j                  dh t         j                  di t         j                  d	djt         j                  dk i t         j                  dl t         j                  dm t         j                  d
dnt         j                  do t         j                  ddpt         j                  dq t         j                  dѐdrt         j                  dѐdst         j                  dѐdtt         j                  dѐdut         j                  dѐdvt         j                  j                  j                  dw t         j                  j                  j                  dx t         j                  j                  j
                  dՐdyt         j                  j                  j                  dՐdzt         j                  j                  j                  dՐd{t         j                  j                  j                  dՐd|i t         j                  j                  j                  dՐd}t         j                  j                  j                  dՐd~t         j                  j                  j                  dѐdt         j                  j                  j*                  ddt         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j\                  	 ddt         j                  j                  jn                  dѐdt         j                  j                  j                  	 ddt         j                  j                  jp                  	 dِdt         j                  j                  j                  dސdt         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddi t         j                  j                  j                  ddt         j                  j                  j                  dސdt         j                  j                  j.                  	 ddt         j                  j                  j0                  	 	 ddt         j                  j                  jX                  ddt         j                  j                  j                  ddt         j                  j                  j                  	 ddt         j                  j                  j                  	 	 ddt         j                  j                  j                   	 ddt         j                  j                  j                  	 	 ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j
                  ddt         j                  j                  j                  d
dt         j                  j                  j                  ddt         j                  j                  j                  ddi t         j                  j                  j                  ddt         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j*                  d
dt         j                  j                  j                  ddt         j                  j                  j                  dѐdt         j                  j                  j                  ddt         j                  j                  j<                  d dt         j                  j                  j                  d t         j                  j                  j                  ddt         j                  j                  j                  ddt         j                  j                  j                   ddt         j                  j                  jd                  	 ddt         j                  j                  j                  	 ddi t         j                  j                  j                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j"                  	 ddt         j                  j                  j                  	 ddt         j                  j                  j$                  	 ddt         j                  j                  j&                  d!dt         j                  j                  j(                  d!dt         j                  j                  j*                  d!dt         j                  j                  j,                  ddt         j                  j                  j.                  	 	 	 d"dt         j                  j                  j0                  	 d#dt         j                  j                  j2                  	 ddt         j                  j                  j4                  	 ddt         j                  j                  j6                  	 d$dt         j                  j                  j8                  d%dt         j                  j                  j:                  ddt         j                  j                  j<                  d&di t         j                  j                  j>                  d'dt         j                  j                  j@                  	 d(dt         j                  j                  jB                  dÄ t         j                  j                  jD                  dՐdĄt         j                  j                  jF                  dՐdńt         j                  j                  jH                  d)dƄt         j                  j                  jJ                  dՐdǄt         j                  j                  jL                  dՐdȄt         j                  j                  jN                  dՐdɄt         j                  j                  jP                  d*dʄt         j                  j                  jR                  d+d˄t         j                  j                  jT                  d,d̄t         j                  j                  jV                  dd̈́t         j                  j                  jX                  d d΄t         j                  j                  jZ                  d dτt         j                  j                  j\                  d-dЄt         j                  j                  j^                  ddфi t         j                  j                  j`                  d҄ t         j                  j                  jb                  dӄ t         j                  j                  jd                  dՐdԄt         j                  j                  jf                  	 d.dՄt         j                  j                  jh                  ddNdd9d֜dׄt         j                  j                  jj                  dd؄t         j                  jl                  jn                  d/dلt         j                  jl                  jp                  d/dڄt         j                  jl                  jr                  dۄ t         j                  jl                  jt                  d0d܄t         jv                  dՐd݄t         jx                  dzdޜd߄t         jz                  d t         j|                  d1dt         j                  j|                  d2dt         j                  j~                  d3dt         j                  j                  d4di t         j                  d5dt         j                  d1dt         j                  d t         j                  d t         j                  d6dt         j>                  d'dt         j                  d t         j                  d7dt         j                  ddt         j                  d8dt         j                  j                  d9dt         j                  d t         j                  d t         j                  dѐdt         j@                  d t         j                  dѐdt         j                  dѐdi t         jB                  d t         j                  ddt         j                  dѐdt         j                  dѐdt         j                  dՐdt         j                  d t         j                  d t         j                  d t         j                  d t         j                  d t         j                  dd t         j                  j                  d:dt         j                  d;dt         j                  d;dt         j                  d t         j                  d t         j                  d i t         j                  d t         j                  d t         j                  d	 t         j                  t               d
ddfdt         j                  t               dddfdt         j                  t               dddfdt         j                  d t         j                  d t         j                  dѐdt         j                  ddt         j                  dt         j                  ddfdt         j                  ddt         j                  d t         j                  dѐdt         j                  dѐdt         j                  j                  ddt         j                  d i t         j                  d t         j                  dѐdt         jD                  dՐd t         j                  dѐd!t         j                  dѐd"t         j                  dѐd#t         j                  d$ t         j                  d% t         j                  dؐd&t         j                  d' t         j                  dؐd(t         j                  dѐd)t         j                  d<d+t         j                  dѐd,t         j                  dѐd-t         j                  d. t         jH                  d)d/i t         j                   dѐd0t         j                  dd1t         j                  dҐd2t         j                  d3 t         j                  d4 t         j
                  dd5t         j                  dېd6t         j                  d=d7t         j                  d8 t         j                  d9 t         j                  d>d:t         j                  d>d;t         jJ                  dՐd<t         j                  dѐd=t         j                  dѐd>t         j                  dѐd?t         j                  dѐd@i t         j                   dѐdAt         j"                  dѐdBt         j$                  dѐdCt         j&                  dD t         j                  j&                  dE t         j(                  dF t         j*                  dG t         jX                  dѐdHt         j                  j,                  ddIt         j                  j.                  ddJt         j0                  d?dddKdLt         j2                  d
dMt         j4                  d
dNt         j6                  dѐdOt         j8                  dѐdPt         j:                  dؐdQt         j<                  dҐdRi t         j>                  dܐdSt         j@                  dѐdTt         jB                  dѐdUt         jD                  	 d@dVt         jF                  dѐdWt         jH                  dѐdXt         jJ                  dѐdYt         jL                  dZ t         jN                  d[ t         jP                  d\ t         jR                  d] t         jT                  d^ t         jV                  d_ t         jX                  d` t         jZ                  da t         j\                  db t         j^                  dc i t         j`                  dd t         jb                  de t         jd                  df t         jf                  dg t         jh                  dh t         jj                  di t         jl                  dѐdjt         jn                  dAdkt         jp                  dBdlt         j                  jn                  ddmt         j                  jr                  dѐdnt         jt                  do t         jv                  dp t         jx                  jz                  dq t         jx                  j|                  dr t         jx                  j~                  ds t         jx                  j                  dt i t         jx                  j                  du t         jx                  j                  dѐdvt         jx                  j                  dѐdwt         jx                  j                  dѐdxt         jx                  j                  dѐdyt         jx                  j                  dz t         jx                  j                  d{ t         jx                  j8                  d| t         jx                  j:                  d} t         jx                  j                  d~ t         jx                  j<                  d t         jx                  j@                  d t         jx                  j                  d t         jx                  jB                  d t         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j                  d i t         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j@                  d t         jx                  j                  d t         jx                  j<                  dѐdt         jx                  jT                  d t         jx                  jV                  dߐdt         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d i t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dѐdt         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  d t         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j                  dѐdt         jx                  j"                  d t         jx                  jX                  dѐdt         jx                  j                  d t         jx                  j                  dѐdt         jx                  jJ                  dѐdi t         jx                  j                  dѐdt         j                  d t         j                  d t         j                  dؐdt         j                  dѐdt         j                  dѐdt         j                  j                  ddt         j                  j                  dѐdt         j                  dCdt         j                  d
dt         jd                  dՐdt         j                  d t         j                  ddt         j                  d t         j                  d t         j                  ddt         j                  ddi t         j                  dDdt         j                  j                  d6dt         j                  dܐdt         jf                  	 d.dt         j                  dܐdt         j                  d t         j                  dѐdt         j                  d
dt         j                  d t         j                  dEdt         j                  dېdÄt         j                  dĄ t         j                  d
dńt         j                   d
dƄt         j                  d
dǄt         j                  dѐdȄt         j                  j                  dѐdɄi t         j                  dѐdʄt         j
                  dѐd˄t         j                  d̄ t         j                  dѐd̈́t         j                  dؐd΄t         j                  ddτt         j                  dЄ t         j                  dф t         j                  d҄ t         j                  dӄ t         j                  dԄ t         j                  dՄ t         j                   dѐdքt         j"                  dׄ t         j$                  dd؄t         j&                  ddٜdڄt         j(                  dۄ i t         j*                  d܄ t         j,                  d݄ t         j.                  dބ t         j0                  d߄ t         j2                  d>dt         j4                  d
dt         j6                  d
dt         j8                  d t         j:                  d t         j<                  d t         j>                  d t         j@                  d t         jB                  d t         jD                  d t         jF                  d t         jH                  d t         jJ                  d i t         jL                  d t         jN                  d t         jP                  d
dt         jR                  d t         jT                  d t         jV                  d | jX                  d | jZ                  d | j\                  d | j^                  d | j`                  d | jb                  d | jd                  d | jf                  d | jh                  d | jj                  d | jl                  d i | 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 | j                  d | j                  d i | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  ddd| j                  d | j                  d | j                  j                  d | j                  j                  d | j                  j                  d | j                  j                  d | j                  j                  d | j                  j                  d | j                  j                  d  i | j                  j                  d! | j                  j                  d" | j                  j                  d# | j                  j                  d$ | j                  j                  d% | j                  j                  d& | j                  d' | 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                  d0 | j                  j                  d1 i | 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 | j                  j                  dB i | j                  j                  dC | j                  j                  dD | j                  j                  dE | j                  j                  dF | j                  j                  dG | j                  j                  dH | j                   j                  dI | j                  ddJ| j                  dK | j                  dL | j                  dM | j
                  dN | j                  dO | j                  dP | j                  dQ | j                  dR | j                  dS i | j                  dT | j                  dU | j                  dV | j                  dW | j                  dX | j                  dY | j                   dZ | j"                  d[ | j$                  dd\| j&                  t         j(                  fd]| j*                  t         j(                  fd^| j,                  t         j(                  fd_| j.                  t         j(                  fd`| j0                  d*ddadb| j2                  dc | j4                  dd | j6                  t         j8                  fdei | j:                  dՐdf| j<                  t         j(                  fdg| j>                  t         j(                  fdh| j@                  t         j(                  fdi| jB                  t         j(                  fdj| jD                  dk | jF                  dl | j                  ddm| jH                  dn | jJ                  do | jL                  t         j(                  fdp| jN                  t         j(                  fdq| jP                  dr | jR                  ds | jT                  dt | jV                  dddadu| jX                  dv i | jZ                  dw | j\                  t         j(                  fdx| j^                  t         j(                  fdy| j`                  ddadz| jb                  d{ | jd                  t         j(                  fd|| jf                  t         j(                  fd}| jh                  d~ | jj                  d | jl                  t         j(                  fd| jn                  d | jp                  d | j                  d | jr                  d | jt                  d | jv                  d | jx                  d i | jz                  dFddad| j<                  d | j|                  t         j(                  fd| j~                  d | j                  d | j                  d | j                  dՐd| j(                  d | j                  d | j                  d | j                  d | j                  d | j                  d | jp                  d | j                  d | j                  d | j                  d i | j                  dՐd| j                  d | j                  dddad| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dd| j                  d | j                  d | j                  d | j                  d | j                  d | j                  d | j                  dGdi | 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 | j                  d | j                  d | j                  d i | j                  ddt         j(                  fd| j                  dddd| j                  dؐd| j                  d | j                  d | j                  d | j                  d | jj                  dÄ | jn                  d*dĄ| j                  dń | j                  dƄ | j                  dǄ | j                  dȄ | j                  dѐdɄ| j                  dʄ t         j                  j                  dؐd˄}i }t               }|j                         D ]  \  }}|j                  |j                  dz   d|j                  z   dz   d|j                  z   dz   d|j                  z   dz   g}|j                  j                  dЫ      rH|j                  t        dЫ      d }|j                  d|z   dz   d|z   dz   d|z   dz   g       |D ],  }t        | |d      }	t        |	      s|	|vs#|	|vs(|||	<   .  |j                  |       |S (H  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r6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  output_sizes     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )inputsrA  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  otherr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  batch1batch2alphabetar;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  tensor1tensor2valuer;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rT  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  mat1mat2rR  rQ  r;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  matvecrR  rQ  r;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  vec1vec2rR  rQ  r;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  thetasizealign_cornerss      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  dims     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   Fc                      yr6  r8  )r:  rL  trolatol	equal_nans        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  ptraininplaces       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rj  keepdimr;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rw  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  msgs     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  tensorss    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  kernel_sizer  padding	ceil_modecount_include_pads         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rN  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  )	r:  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )grad_outr:  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r:  r  r  r  input_gweight_gbias_gs           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  r  r  r  r  r  counts           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  	generatorr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  input1input2r  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   r  c                      	 yr6  r8  r:  targetr  size_averagereduce	reduction
pos_weights          r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    TV r2   c                      yr6  r8  )r:  weights	minlengths      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  probr  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r\  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  selfrf  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  
boundaries	out_int32rightr;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  rj  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )x1x2rq  compute_modes       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2         ?c                      yr6  r8  r:  rQ  rs  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )r;  c                      yr6  r8  )r;  matricess     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  groupss     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  upperr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  check_errorsr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  numeln_binsratio	bit_widths        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  chunksrj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  minmaxr;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  r;  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  
correctionfweightsaweightss       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rwith_replacements      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )realimags     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )absangs     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  ords     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  padrW  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  r  r  r  dilationr  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  )	r:  r  r  r  r  r  
transposedoutput_addingr  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>	  r=  r2   c                      yr6  r8  r:  r  r  r  r  output_paddingr  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>
  r=  r2   c                      yr6  r8  r!  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r!  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  r  r  marginr  r  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  rj  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rL  rj  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   r7  c                      yr6  r8  r.  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    .0 r2   c                      yr6  r8  r:  rj  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rj  r;  r`   s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r<  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  yxrj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>!  r=  r2   c                      yr6  r8  r:  diagonalr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>"  r=  r2   c                      yr6  r8  rI  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>#  r=  r2   c                      yr6  r8  )r:  offsets     r/   r<  z'get_testing_overrides.<locals>.<lambda>$  r=  r2   c                      yr6  r8  )r:  nrj  prependappendr;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>%  r=  r2   c                      yr6  r8  r:  rM  dim1dim2s       r/   r<  z'get_testing_overrides.<locals>.<lambda>&  r=  r2   c                      yr6  r8  rS  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>'  r=  r2   c                      yr6  r8  )r:  srcrM  rT  rU  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>(  r=  r2   c                      yr6  r8  )r  rX  rf  r  storage_offsets        r/   r<  z'get_testing_overrides.<locals>.<lambda>)  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>*  r=  r2   c                      yr6  r8  )r:  rL  rq  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>+  r=  r2   c                      yr6  r8  r:  rL  rounding_moder;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>,  r=  r2   c                      yr6  r8  r^  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>-  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>.  r=  r2   c                      yr6  r8  rp  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>/  r=  r2   c                      yr6  r8  r:  r\  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>0  r=  r2   c                      yr6  r8  )r[  r\  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>1  r=  r2   c                      yr6  r8  r:  indices_or_sectionss     r/   r<  z'get_testing_overrides.<locals>.<lambda>2  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>3  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>4  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>5  r=  r2   c                      yr6  r8  r:  UPLOr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>6  r=  r2   c                      yr6  r8  rm  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>7  r=  r2   c                      yr6  r8  )equationoperandss     r/   r<  z'get_testing_overrides.<locals>.<lambda>8  r=  r2   c                      	 yr6  r8  r:  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r/   r<  z'get_testing_overrides.<locals>.<lambda>9  s
    (* r2   c
                      	 yr6  r8  )
r:  r  offsetsrv  rw  rx  moder   per_sample_weightsru  s
             r/   r<  z'get_testing_overrides.<locals>.<lambda>;  s
    df r2   c                      yr6  r8  r:  r`   rn   r_   requires_grads        r/   r<  z'get_testing_overrides.<locals>.<lambda>=  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>>  r=  r2   c                      yr6  r8  r:  rL  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>?  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>@  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>A  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>B  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>C  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>D  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>E  r=  r2   c                      yr6  r8  )r:  scale
zero_pointaxis	quant_min	quant_maxs         r/   r<  z'get_testing_overrides.<locals>.<lambda>F  r=  r2   c                      yr6  r8  )r:  r  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>G  r=  r2   c                      	 yr6  r8  )rA  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r/   r<  z'get_testing_overrides.<locals>.<lambda>H  s
    ac r2   c                      yr6  r8  r:  packed_weightr  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>K  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>L  r=  r2   c                      yr6  r8  r:  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>M  r=  r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>N  
    SU r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>P  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>Q  r=  r2   c                      yr6  r8  )r:  abs      r/   r<  z'get_testing_overrides.<locals>.<lambda>R  r=  r2   c                      yr6  r8  r:  rq  rr  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>S  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>T  r=  r2   c                      yr6  r8  r:  rO  rj  norms       r/   r<  z'get_testing_overrides.<locals>.<lambda>U  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>V  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>W  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>X  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>Y  r=  r2   c                      yr6  r8  r:  srj  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>Z  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>[  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>\  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>]  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>^  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>_  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>`  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>a  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>b  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>c  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>d  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>e  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>f  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>g  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>h  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>i  r=  r2   c                      yr6  r8  )r:  	start_dimend_dims      r/   r<  z'get_testing_overrides.<locals>.<lambda>j  r=  r2   c                      yr6  r8  r:  dimss     r/   r<  z'get_testing_overrides.<locals>.<lambda>k  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>l  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>m  r=  r2   c                      yr6  r8  rw  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>n  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>o  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>p  r=  r2   c                      yr6  r8  r:  exponentr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>q  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>r  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>s  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>t  r=  r2   c                      yr6  r8  )r:  
fill_valuer;  r`   rn   r_   r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>u  r=  r2   c                      yr6  r8  )r:  r  	dep_tokens      r/   r<  z'get_testing_overrides.<locals>.<lambda>v  r=  r2   c                      yr6  r8  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r/   r<  z'get_testing_overrides.<locals>.<lambda>w  r=  r2   c                      yr6  r8  )r:  rj  indexr;  sparse_grads        r/   r<  z'get_testing_overrides.<locals>.<lambda>x  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>y  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>z  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>{  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>|  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>}  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>~  r=  r2   c                      yr6  r8  r:  rb  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  spacingrj  
edge_orders       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  gridinterpolation_modepadding_moderg  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  
num_groupsr  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  	r:  hxparams
has_biases
num_layersdropoutrr  bidirectionalbatch_firsts	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  w_ihw_hhb_ihb_hhs         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  lambds     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  valuesr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  r)  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  binsr  r  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r  densityr;  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r   r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  taus     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r[  r\  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rg  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rj  r  sources       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r   s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  indicesr  
accumulates       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  r  rW  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  r  r!  r  include_inputs         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rZ   s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )eteassume_uniqueinverts       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r+  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r+  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      	 yr6  r8  )	r:  r  r  r  r  use_input_statsr  r  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    -/ r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rL  rtolrm  rn  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c
                      	 yr6  r8  )
r:  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    Z\ r2   c                      yr6  r8  r:  r  r  r  r  
log_targets         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  krj  rx  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  	hermitianr  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rX  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )LDpivotsBrX  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  normalized_shaper  r  espr  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  endr  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  )r:  rV  r]  XrO  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    RT r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rj  r`   s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rA  r@  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  namesrx  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  )	databatch_sizesr  r  r  r  r  rr  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )Apivot	get_infosr;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r(  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  maskrW  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r!  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )LUr\  r]  leftadjointr;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rO  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rO  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rl  rX  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  r  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  r:  r  r  r  r  return_indicesr  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  
    OQ r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  rx  r`   r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                       yr6  r8  )r  r.   s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  )r:  r  r  r  r  r  exponential_average_factorepsilons           r/   r<  z'get_testing_overrides.<locals>.<lambda>  
    GI r2   c	                      yr6  r8  	r:  r  r  r  r  r  r  	benchmarkdeterministics	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  )	r:  r  zrQ  r  r  r  r  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c
                      	 yr6  r8  )
r:  r  r  r  r"  r  r  r  r  r  s
             r/   r<  z'get_testing_overrides.<locals>.<lambda>   r  r2   c	                      	 yr6  r8  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    <> r2   c                      	 yr6  r8  )r:  r  weight_stride0r  cxr{  hidden_sizer  r   r  rr  r  r  dropout_states                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    VX r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rw  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r!  destinations      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>	  r=  r2   c                      yr6  r8  )r:  
descendingr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>
  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  num_samplesreplacementr;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r_  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rq  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  startrO  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  nanposinfneginfr;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r  r  r  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r_  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  NCHxWgroupr  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rq  rj  rx  r`   s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r@  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r@  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rA  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>!  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>"  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>#  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>$  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>%  r=  r2   c                      yr6  r8  rd  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>&  r=  r2   c                      yr6  r8  r:  rq  r  rs  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>'  r=  r2   c                      	 yr6  r8  r:  r  r  r  r  r  divisor_overrides          r/   r<  z'get_testing_overrides.<locals>.<lambda>(  
    XZ r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>*  r   r2   c                      	 yr6  r8  )r:  r  r  r  r  r  r  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>,  s
    BD r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>.  r=  r2   c                      	 yr6  r8  r:  r  r  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>/  
    EG r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>1  s
    oq r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>3  r=  r2   c                      	 yr6  r8  r(  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>4  r  r2   c                      	 yr6  r8  )r:  r  r  r  ignore_indexr  r  label_smoothings           r/   r<  z'get_testing_overrides.<locals>.<lambda>6  
    `b r2   c                      	 yr6  r8  r1  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>8  s
    NP r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>:  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>;  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda><  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>=  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>>  r=  r2   c                      	 yr6  r8  rt  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>?  
    PR r2   c                      	 yr6  r8  )r:  r  rz  rv  rw  rx  r{  r   r|  include_last_offsetru  s              r/   r<  z'get_testing_overrides.<locals>.<lambda>A  s
    Y[ r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>D  r=  r2   c                      yr6  r8  )r:  rA  r  r  r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>E  r=  r2   c                      	 yr6  r8  r:  r  rA  output_ratior  _random_sampless         r/   r<  z'get_testing_overrides.<locals>.<lambda>F  r  r2   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>I  
    "$ r2   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>K  r  r2   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>N  r  r2   c                      yr6  r8  )r:  r  varr   r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>P  r=  r2   c                      yr6  r8  )r:  approximates     r/   r<  z'get_testing_overrides.<locals>.<lambda>Q  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>R  r=  r2   c                      yr6  r8  )r:  r  r{  r  rg  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>S  r=  r2   c                      yr6  r8  )r:  r  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>T  r=  r2   c                      yr6  r8  )logitsr  hardr  rj  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>U  r=  r2   c                      yr6  r8  r
  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>V  r=  r2   c                      yr6  r8  )r:  min_valmax_valrs  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>W  r=  r2   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>X  r  r2   c                      	 yr6  r8  )r:  r  r  r  r  r6  r  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>Z  s
    [] r2   c                      	 yr6  r8  )r:  rf  scale_factorr{  rg  recompute_scale_factor	antialiass          r/   r<  z'get_testing_overrides.<locals>.<lambda>\  r   r2   c                      yr6  r8  rR  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>^  r=  r2   c                      yr6  r8  r:  r  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>_  r=  r2   c                      yr6  r8  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>`  r=  r2   c                      yr6  r8  )r:  negative_slopers  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>a  r=  r2   c                      yr6  r8  )r:  r  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>b  r=  r2   c                      yr6  r8  )r:  rf  rQ  rR  rV  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>c  r=  r2   c                      yr6  r8  r:  rj  _stacklevelr`   s       r/   r<  z'get_testing_overrides.<locals>.<lambda>d  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>e  r=  r2   c                      yr6  r8  r:  rw  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>f  r=  r2   c                      yr6  r8  rC  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>g  r=  r2   c                      yr6  r8  rC  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>h  r=  r2   c                      	 yr6  r8  r(  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>i  s
    QS r2   c                      	 yr6  r8  r:  r  r  r  r  r  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>k  r  r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>m  
    ]_ r2   c                      	 yr6  r8  rH  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>o  r  r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>q  rJ  r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>s  r  r2   c                      	 yr6  r8  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>u  rJ  r2   c                      yr6  r8  r:  r$  r  r  r  rA  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>w  r=  r2   c                      yr6  r8  rP  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>x  r=  r2   c                      yr6  r8  rP  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>y  r=  r2   c                      yr6  r8  r8  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>z  r=  r2   c                      	 yr6  r8  )querykeyrW  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals                            r/   r<  z'get_testing_overrides.<locals>.<lambda>|  s     km r2   c                      	 yr6  r8  )r:  r  rq  r)  r  r  r  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r  r2   c                      	 yr6  r8  r8  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r  r2   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  
    Y[ r2   c                      	 yr6  r8  )r:  r  r  r  r  r  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    FH r2   c                      yr6  r8  )r:  rq  rj  r  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rZ   num_classess     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r{  rW  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  r  rq  r  rx  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  )r:  r  	log_inputr   r  r  r  r  s           r/   r<  z'get_testing_overrides.<locals>.<lambda>  ro  r2   c                      yr6  r8  r:  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rs  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  lowerr  r  rs  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rU  rV  rW  rc  r^  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r  rR  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  deltas       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r8  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r?  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r?  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rR  	thresholds      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r
  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  rW  rs  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c
                      	 yr6  r8  
anchorpositivenegativer)  rq  r  swapr  r  r  s
             r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    pr r2   )distance_functionr)  r  r  c                     	 yr6  r8  )r  r  r  r  r)  r  r  s          r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    ^` r2   c                      yr6  r8  )r:  r  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rZ   r  r  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rZ   r  stdr  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rZ   vals     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rZ   r  r{  nonlinearityr  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  as_tuples     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )r  c                     yr6  r8  )r:  rf  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rq  rj  rx  r;  r`   s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  rj  rx  r;  r`   s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )vpowrj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  input3r  	transposes        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rv  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  rj  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  qrL  rk  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rconds     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  rX  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  upscale_factors     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  downscale_factors     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  rx  r   r  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rz  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r~  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r`   s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r!  r%  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  somer;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r{  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  r  rj  rx  interpolationr;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  scaleszero_pointsr  r`   s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r`   s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r`   reduce_ranges      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  r#  r  output_scaleoutput_zero_points           r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      	 yr6  r8  r:  r  r  r  r  r  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    eg r2   c                      	 yr6  r8  r  s                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    fh r2   )r   )   c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r  r2   )r   r   )r  r  c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r  r2   )r   r   r   )r  r  r  c                      	 yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    JL r2   c                      	 yr6  r8  r  s                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  
    jl r2   c                      	 yr6  r8  r  s                 r/   r<  z'get_testing_overrides.<locals>.<lambda>  r  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r~  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  highr`   rn   r_   r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r~  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r.  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rq  rj  maxnormr;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  shapes     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c	                      yr6  r8  r  s	            r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  shiftsr  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   r   r  c                      yr6  r8  )r:  rV  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  compressed_indices_dtypes      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rL  rQ  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rZ  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rj  r  rX  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  r  rX  r  include_selfs         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )sorted_sequencer:  r  r  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  lengthsr$  rz  r  unsafes          r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rX  rj  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r:  rX  rj  r  rf  steps         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r"  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r|  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>	  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>
  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rd  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rd  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ru  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r]  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r]  r  r  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )stabler;  c                     yr6  r8  )r:  rj  r  r4  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rZ   split_size_or_sectionsrj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r7  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rZ  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>!  r=  r2   c
                      	 yr6  r8  )
r:  rH  rI  rJ  rK  rL  pad_moderM  rN  rP  s
             r/   r<  z'get_testing_overrides.<locals>.<lambda>"  s
    _a r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>$  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>%  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>&  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>'  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>(  r=  r2   c                      yr6  r8  r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>)  r=  r2   c                      yr6  r8  rI  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>*  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>+  r=  r2   c                      yr6  r8  )r  r  cs      r/   r<  z'get_testing_overrides.<locals>.<lambda>,  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>-  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>.  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>/  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>0  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>1  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>2  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>3  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>4  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>5  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>6  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>7  r=  r2   c                      yr6  r8  )r:  r  
compute_uvr;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>8  r=  r2   c                      yr6  r8  )r:  r  rk  Ms       r/   r<  z'get_testing_overrides.<locals>.<lambda>9  r=  r2   c                      yr6  r8  )r:  full_matricesr;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>:  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>;  r=  r2   c                      yr6  r8  r:  dim0rT  s      r/   r<  z'get_testing_overrides.<locals>.<lambda><  r=  r2   c                      yr6  r8  )r:  axis0axis1s      r/   r<  z'get_testing_overrides.<locals>.<lambda>=  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>?  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>@  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>A  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>B  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>C  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>D  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>E  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>F  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>G  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>H  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>I  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>J  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>K  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>L  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>M  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>N  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>O  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>P  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>Q  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>R  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>S  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>T  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>U  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>V  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>W  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>X  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>Y  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>Z  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>[  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>\  r=  r2   c                      yr6  r8  ru  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>]  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>^  r=  r2   c                      yr6  r8  rw  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>_  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>`  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>a  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>b  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>c  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>d  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>e  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>f  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>g  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>h  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>i  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>j  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>k  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>l  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>m  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>n  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>o  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>p  r=  r2   c                      yr6  r8  ru  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>q  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>r  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>s  r=  r2   c                      yr6  r8  rK  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>t  r=  r2   c                      yr6  r8  )r  rL  r;  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>u  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>v  r=  r2   c                      yr6  r8  )r:  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>w  r=  r2   c                      yr6  r8  )r:  r$  rj  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>x  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>y  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>z  r=  r2   c                      yr6  r8  )r  inds     r/   r<  z'get_testing_overrides.<locals>.<lambda>{  r=  r2   c                      yr6  r8  )r  r  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>|  r=  r2   c                      yr6  r8  )r  r  r  r;  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>}  r=  r2   c                      yr6  r8  )r:  rh  rj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>~  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rV  rj  r  r;  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rF  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ra  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r?  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r?  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  r  unitriangulars        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r]  r  r  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rI  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c
                      	 yr6  r8  r  s
             r/   r<  z'get_testing_overrides.<locals>.<lambda>  s
    VX r2   c                      yr6  r8  rI  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  rj  sizesr  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  sortedreturn_inversereturn_countsrj  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r:  r  r  rj  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r$  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r7  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r7  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r9  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )rA  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  ri  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  rg  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )	conditionrA  r@  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r~  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  levels     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )primaltangentr  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rf  r  rZ  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rM  rT  rU  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )implicitc                     yr6  r8  )r  rf  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rj  r  rO  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  rf  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rj  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rj  r  rf  r#  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  
split_sizerj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  split_sizesrj  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rb  rT  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  r`   s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  	dimensionrf  r#  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  rL  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  arrays     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  idxs     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  memos     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  format_specs     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  protos     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )tensor_contentsc                     yr6  r8  )r  r"  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  rV  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  ds     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r5  r6  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>	  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>
  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r`   non_blockingr.   s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>   r=  r2   c                      yr6  r8  )r  r  rs  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>!  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>"  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>#  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>$  r=  r2   c                      yr6  r8  )r  orderellipsis_idxs      r/   r<  z'get_testing_overrides.<locals>.<lambda>%  r=  r2   c                      yr6  r8  )r  callables     r/   r<  z'get_testing_overrides.<locals>.<lambda>&  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>'  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>(  r=  r2   c                      yr6  r8  )r  gradientretain_graphcreate_graphrC  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>)  r=  r2   c                      yr6  r8  r  rg   s     r/   r<  z'get_testing_overrides.<locals>.<lambda>*  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>+  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>,  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>-  r=  r2   )r  c                     yr6  r8  )r  mediansigmar  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>.  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>/  r=  r2   c                      yr6  r8  )r  	coalesceds     r/   r<  z'get_testing_overrides.<locals>.<lambda>0  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>1  r=  r2   c                      yr6  r8  )r  rX  r\  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>2  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>3  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>4  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>5  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>6  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>7  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>8  r=  r2   c                      yr6  r8  )r  rX  rM  rT  rU  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>9  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>:  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>;  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda><  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>=  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>?  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>@  r=  r2   c                     yr6  r8  )r  r  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>A  r=  r2   c                      yr6  r8  r  rW  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>B  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>C  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>D  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>E  r=  r2   c                     yr6  r8  )r  rq  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>F  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>G  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>H  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>I  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>J  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>K  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>L  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>M  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>N  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>O  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>P  r=  r2   c                      yr6  r8  )r  rZ   s     r/   r<  z'get_testing_overrides.<locals>.<lambda>Q  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>R  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>S  r=  r2   c                     yr6  r8  )r  r  r  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>T  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>U  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>V  r=  r2   c                      yr6  r8  )r  rZ   rn  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>W  r=  r2   c                      yr6  r8  )r  rA  r@  rn  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>X  r=  r2   c                      yr6  r8  )r  r\  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>Y  r=  r2   c                      yr6  r8  )r  rL  assigns      r/   r<  z'get_testing_overrides.<locals>.<lambda>Z  r=  r2   c                      yr6  r8  )r  r  r  rO  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>[  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>\  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>]  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>^  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>_  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>`  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>a  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>b  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>c  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>d  r=  r2   c                      yr6  r8  )r  r$  rZ   r%  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>e  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>f  r=  r2   c                     yr6  r8  )r  from_tor  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>g  r=  r2   c                      yr6  r8  r  streams     r/   r<  z'get_testing_overrides.<locals>.<lambda>h  r=  r2   c                      yr6  r8  )r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>i  r=  r2   c                      yr6  r8  r  hooks     r/   r<  z'get_testing_overrides.<locals>.<lambda>j  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>k  r=  r2   c                      yr6  r8  )r  names     r/   r<  z'get_testing_overrides.<locals>.<lambda>l  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>m  r=  r2   c                      yr6  r8  )r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>n  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>o  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>p  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>q  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>r  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>s  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>t  r=  r2   c                      yr6  r8  )r  r!  rZ  rf  r  s        r/   r<  z'get_testing_overrides.<locals>.<lambda>u  r=  r2   c                      yr6  r8  )r  rX  rj  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>v  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>w  r=  r2   c                      yr6  r8  rv  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>x  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>y  r=  r2   c                      yr6  r8  )r  rX  rj  r  rf  r#  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>z  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>{  r=  r2   c                      yr6  r8  )r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>|  r=  r2   c                      yr6  r8  )r  r  accumulate_matchess      r/   r<  z'get_testing_overrides.<locals>.<lambda>}  r=  r2   c                      yr6  r8  r  size1size2	dense_dims       r/   r<  z'get_testing_overrides.<locals>.<lambda>~  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r[  r\  rR  rQ  r;  s         r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  repss     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r`   r\  copyrg   s        r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   )masked_gradc                     yr6  r8  r  r`   r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  r  s      r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s     r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  r  s    r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   c                      yr6  r8  )r  r  r   drivers       r/   r<  z'get_testing_overrides.<locals>.<lambda>  r=  r2   _____i__rbitwise_N)r  r  N)r  N)h㈵>:0yE>F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)       @#use_mm_for_euclid_dist_if_necessary)r  F)FN)NNN)r  NN)   F)Nr  r   r  r  )Nr  r   r   r  r  )r   NNr  )r  r   )r7  N)r   r  Fr6  )r  r7  NNN)r   r   r  )r   r7  )r  )LN)NNr  FF)Nr  Fr  FNN)NNNF)FF)Nr7  N)Nr  r7  N)r   r7  )TT)NF)NNr  )NNr  T)      ?)r  NNr  )d   r   r   N)r  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   r  F)Nr   r  FF)NFNN)r7  FN)        NNN)NNr  )r  NFN)r  FF)Nr   FTN)NNF皙?r  )NNNr  )NNNr  r	  )r  TF)	NNr  Fr  FNFN)r  r   r  )NNFN)Fư>r  )none)r7  )bilinearr   N)r  Fg|=r7  )g      r  F)NNNNTr
  r  )NNnearestNNF)NNr  )g{Gz?F)g-C6?g      ?r  )N   N)Nr   N)TNTNFNNNNNNF)r  r  NNNr  )NNr  Nr  )r  r  g-q=N)r   r   )r  r  F)TFNr   Nr  )g      ?gUUUUUU?FF)Nr	  )NNr  r  )r  r  )r     )r  r  r  FNNr  )r	  r  N)r   fan_in
leaky_reluN)froNFNN)NNFNN)r  NFNN)r  r  FNN)r  r   )TF)NTr  )V瞯<)r  F)reducedN)NFlinearN)r  r  )r  NNNr   F)r   NNr  )r7  F)NNNTreflectFTN)TTN)   r  N)r  N)TFF)TFFN)r  r  )Nr   NN(~  r4   r5   r  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshaddaddbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	bernoullir   binary_cross_entropy_with_logitsbincountbinomialbitwise_andbitwise_not
bitwise_orbitwise_xorbitwise_left_shiftbitwise_right_shift
block_diagbmmbroadcast_tensorsbroadcast_to	bucketizecartesian_prodcatconcatconcatenatecdistceilceluchain_matmulchannel_shufflecholeskylinalgcholesky_excholesky_inversecholesky_solvechoose_qparams_optimizedchunkclampclip	clamp_min	clamp_maxcolumn_stackcovclonecombinationscomplexcopysignpolarr   conjconj_physicalresolve_conjresolve_negconstant_pad_ndconv1dconv2dconv3dconvolutionconv_tbcconv_transpose1dconv_transpose2dconv_transpose3dcorrcoefcoscosine_embedding_losscoshcosine_similaritycount_nonzerocrossctc_losscummaxcummincumprodcumsumcumulative_trapezoidlogcumsumexpdeg2rad
dequantizedetdetachdiag
diag_embeddiagflatdiffrJ  diagonal_scatteras_strided_scatterdigammadistdivdividedotr  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdgegreater_equalgeqrfi0inneroutergerrr  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrink	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacr
  	index_add
index_copy	index_putindex_select
index_fillindex_reduceisfiniteisinisinfisrealisposinfisneginfinstance_normint_reprinverseinvinv_ex
is_complexis_conjis_negis_distributedis_inferenceis_floating_point
is_nonzerois_same_size	is_signediscloseisnanistftkl_divkronkthvalueldl_factor_ex
ldl_factor	ldl_solve
layer_normlcmldexple
less_equallerplgammalobpcgloglog_softmaxlog10log1plog2	logaddexp
logaddexp2logdetxlogylogical_andlogical_not
logical_orlogical_xorlogit	logsumexplstm	lstm_cellltlesslulu_solvemargin_ranking_lossmasked_fillmasked_scattermasked_selectmatmul	lu_factorlu_factor_exmatrix_powermatrix_rank	multi_dot
matrix_expr  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr{  	nanmedianmeshgridr  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr{  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_lossr  r  local_response_norm
logsigmoid	lp_pool1d	lp_pool2d	lp_pool3dmax_pool2d_with_indicesmax_pool3d_with_indicesmax_unpool1dmax_unpool2dmax_unpool3dmse_lossmulti_head_attention_forwardmulti_margin_lossmultilabel_margin_lossmultilabel_soft_margin_lossnll_loss	normalizeone_hotr  pairwise_distancepoisson_nll_lossprelurelurelu6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tuplequantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2deg	rand_likerandint_like
randn_likeravelr	  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_ite	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r#  var_meanvsplitvstackwhere
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_hooksr0  _cdatar1  r2  _grad_fngrad_fn_version_autocast_to_reduced_precision_autocast_to_full_precisionr  r_   r`   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_ort	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizern   r  r  nbytesndim	output_nrr  r  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_update_names_valuesalign_asalign_toapply_rq   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudaxpuipudata_ptrr  rj  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_rY   halfchalf	has_namesr$  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rh   random_record_streamrefine_namesregister_hook"register_post_accumulate_grad_hookrenamerepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_shortrf  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storagerZ  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   items__name__
startswithlenextendgetattrrn  update)
r5   retret2ignoredrV  r  r  subnamer  r    s
             r/   r   r     sW   6 \\Fy%		-y%2y% 	!!#@y% 	!!#A	y%
 	

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

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

.+y%, 	

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

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

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

.Wy%X 	

=Yy%Z 	$:[y%\ 	8]y%^ 	?_y%` 	9ay%b 	  "Pcy%d 	 Gey%f 	Ngy%h 	&&(Yiy%j 	4ky%l 	Cmy%n 	

Boy%p 	8qy%r 	8sy%t 	8uy%v 			Owy%x 	%yy%z 	I{y%| 	,}y%~ 	9y%@ 	(Ay%B 	5Cy%D 	

.Ey%F 	7Gy%H 	6Iy%J 	5Ky%L 	=My%N 	dOy%P 	dQy%R 	dSy%T 	wUy%V 	=Wy%X 	  !AYy%Z 	  !A[y%\ 	  !A]y%^ 	(_y%` 			-ay%b 	##  &Ccy%d 	

.ey%f 	!Cgy%h 	-iy%j 	@ky%l 	Emy%n 	',1oy%r 	5sy%t 	5uy%v 	Bwy%x 	Ayy%z 	""$@{y%| 	;}y%~ 	1y%@ 	*Ay%B 			#Cy%D 	*Ey%F 	&Gy%H 	

:Iy%J 	@Ky%L 	2My%N 	

VOy%P 	BQy%R 	KSy%T 	 OUy%V 	  "YWy%X 	1Yy%Z 	

0[y%\ 			H]y%^ 	K_y%` 			4ay%b 	@cy%d 	

*ey%f 	

)gy%h 	;iy%j 	2ky%l 	4my%n 	8oy%p 	?qy%r 	Csy%t 	4uy%v 	!&+wy%z 	^bg{y%~ 	ey%@ 	3Ay%B 	,Cy%D 			-Ey%F 	

.Gy%H 	0Iy%J 			-Ky%L 	

.My%N 	/Oy%P 	..0oQy%R 	--/hSy%T 	--Z_1dUy%Z 	'')N[y%\ 	779^]y%^ 	'')}_y%` 	77 ;Vay%d 	++-=ey%f 	**,<gy%h 	**,Biy%j 	##%?ky%l 	9my%n 			Coy%p 			Cqy%r 			Dsy%t 			Cuy%v 			Dwy%x 			Jyy%z 			K{y%| 			D}y%~ 			Ey%@ 			EAy%B 			FCy%D 			FEy%F 			GGy%H 			IIy%J 			JKy%L 			JMy%N 			KOy%P 			6Qy%R 			7Sy%T 			BUy%V 			-Wy%X 	@Yy%Z 	

*[y%\ 	&]y%^ 	&_y%` 	Qay%b 	/cy%d 	3ey%f 	?gy%h 	

5iy%j 	

.ky%l 	/my%n 	t4PUP]P]fjz  Doy%p 	&&(Hqy%r 	\sy%t 	Ouy%v 			4wy%x 	3yy%z 	>{y%| 	/}y%~ 	,y%@ 	6Ay%B 	5Cy%D 			3Ey%F 	NGy%H 	cIy%J 	fKy%L 	fMy%N 	mOy%P 			sQy%R 	NSy%T 	3Uy%V 	8Wy%X 	5Yy%Z 	;[y%\ 	""$z]y%^ 	G_y%` 	may%b 	Ycy%d 	((*?ey%f 	4gy%h 	;iy%j 	2ky%l 	6my%n 	7oy%p 	8qy%r 	

.sy%t 	=uy%v 	>wy%x 	Lyy%z 	B{y%| 	=}y%~ 	\y%@ 	)Ay%B 	

GCy%D 	&Ey%F 	'Gy%H 	2Iy%J 	2Ky%L 	 0My%P 	(Qy%R 	1Sy%T 	4Uy%V 	KWy%X 	*Yy%Z 	'[y%\ 	&]y%^ 	._y%` 	,ay%b 	!1cy%d 	*ey%f 	3gy%h 	)iy%j 	Wky%l 	%my%n 	SX]oy%r 	rsy%t 	

+uy%v 	Nwy%x 	""$cyy%z 	!L{y%| 	 S}y%~ 	sy%@	 			4A	y%B	 	6C	y%D	 	3E	y%F	 	;G	y%H	 	

;I	y%J	 	0K	y%L	 	LPUM	y%P	 			-Q	y%R	 	<S	y%T	 	/U	y%V	 	/W	y%X	 	

.Y	y%Z	 	:[	y%\	 	;]	y%^	 	&_	y%`	 	.a	y%b	 	<c	y%d	 	5e	y%f	 	;g	y%h	 	<i	y%j	 	/k	y%l	 	Im	y%n	 	

so	y%p	 	Oq	y%r	 	3s	y%t	 	

5u	y%v	 	Ew	y%x	 	By	y%z	 	!!  $A{	y%|	 	8}	y%~	 	<	y%@
 	=A
y%B
 	7C
y%D
 	?E
y%F
 	 FG
y%H
 	!!#]I
y%J
 	[K
y%L
 	>M
y%N
 	/O
y%P
 	!!#@Q
y%R
 	  "MS
y%T
 	 <U
y%V
 	*W
y%X
 	!1Y
y%Z
 			-[
y%\
 	8]
y%^
 	

5_
y%`
 	la
y%b
 	lc
y%d
 	le
y%f
 	%%HM)Rg
y%j
 	

.k
y%l
 	Vm
y%n
 	0o
y%p
 	3q
y%r
 	5s
y%t
 			-u
y%v
 	8w
y%x
 	

5y
y%z
 	 #J{
y%~
 	  "}
y%@ 	))+vAy%B 	%%'hCy%D 	** .REy%H 	** .?Iy%L 	 YMy%P 	2Qy%R 	

ESy%T 	<Uy%V 	=Wy%X 	AYy%Z 			4[y%\ 	9]y%^ 	U_y%` 	1ay%b 	+cy%d 	:ey%f 	Wgy%h 	!siy%j 	&&(_ky%l 	8my%n 	!foy%p 	!Vqy%r 	Usy%t 	$$&?uy%v 	3wy%x 	:yy%z 			-{y%| 	2}y%~ 	:y%@ 	//1NAy%B 	//1NCy%D 	//1dEy%F 	<<>qGy%H 	//1dIy%J 	<<>qKy%L 	//1dMy%N 	<<>qOy%P 	'')SQy%R 	))+aSy%T 	&&RV*[Uy%X 	&&RV*[Yy%\ 	&&;@*E]y%` 	$$&Ray%b 	00=C4Hcy%f 	<<im@rgy%j 	  "Lky%l 	11KQ5Vmy%p 	))[^-cqy%t 	$$GL(Quy%x 	##%Zyy%z 	%%'\{y%| 	%%'\}y%~ 	%%'\y%@ 	!KAy%B 	%%IN)SCy%F 	))txSW-\Gy%L 	113iMy%N 	  "mOy%P 	11Z^5cQy%T 	>>[` %Wy%Z 	11Z^5c[y%^ 	>>[` %ay%d 	--/uey%f 	  "Fgy%h 	!9iy%j 	'')zky%l 	&&(gmy%n 	**,coy%p 	&&(Cqy%r 	$$&^sy%t 	00=C4Huy%x 	))TY-^yy%| 	''QV+[}y%@ 	""  %AAy%B 	##%oCy%D 	&&(mEy%F 	&&(\Gy%H 	""$GIy%J 	//1gKy%L 	'')^My%N 	&&(8Oy%P 	%%'mQy%R 	%%'mSy%T 	%%'mUy%V 	//IO3TWy%Z 	&&IN*S[y%^ 	33V[7`_y%b 	&&IN*Scy%f 	33V[7`gy%j 	&&IN*Sky%n 	33V[7`oy%r 	((*zsy%t 	((*zuy%v 	((*zwy%x 	$$&pyy%z 	88fjquchm}y%D 	--GM1REy%H 	22?E6JIy%L 	77QW;\My%P 	$$>D(IQy%T 	%%'XUy%V 	##%FWy%X 	!PYy%Z 	--/a[y%\ 	,,QW0\]y%` 	!!#;ay%b 	  "Acy%d 	!!#Bey%f 	!!#ygy%h 	  "Aiy%j 	  "Aky%l 	  "Amy%n 	88:uoy%p 	**,qy%r 	&&(\sy%t 	,,.xuy%v 	##%Zwy%x 	##%Zyy%z 	$$&L{y%| 	&&(C}y%~ 	$$&6y%@ 	&&(8Ay%B 	%%'XCy%D 	//hn3sEy%H 	==RV_bEJV\AaIy%N 	""$bOy%P 	 MQy%R 	QSy%T 	!7Uy%V 	&&(xWy%X 	7Yy%Z 	F[y%\ 	(]y%^ 	

\_y%` 	day%b 	  "hcy%d 	  "pey%f 	9gy%h 	diy%j 	%ky%l 	*my%n 	Qoy%p 	!Sqy%r 	+sy%t 	Iuy%v 	*wy%x 	5yy%z 	I{y%| 	=}y%~ 	Ay%@ 	7Ay%B 	 YCy%D 	6Ey%F 	2Gy%H 	-Iy%J 	dKy%L 			7My%N 	

0Oy%P 			DQy%R 	  "2Sy%T 	""$4Uy%V 	'')9Wy%X 	'Yy%Z 	,[y%\ 	7]y%^ 	C_y%` 	fay%b 	icy%d 	""$Vey%f 	!!#Mgy%h 	))+Piy%j 	""$sky%l 	   $hmy%r 	!! %isy%v 	""uwX\.2e&Hwy%z 	""uwX^.4&J{y%~ 	""uwXa.75&My%B 	%% )mCy%F 	%% )mGy%J 	1Ky%L 	dMy%N 	d5==Y]mrvOy%P 	eQy%R 	%Sy%T 	

.Uy%V 	

5Wy%X 	FYy%Z 	,[y%\ 	/]y%^ 	4_y%` 	

3ay%b 	:cy%d 	Aey%f 	!;gy%h 	.iy%j 	xky%l 	Smy%n 	xoy%p 	Sqy%r 	

7sy%t 	7uy%v 	/wy%x 	5yy%z 	P{y%| 	`}y%~ 	/y%@ 	

4Ay%B 	MCy%D 	8Ey%F 	<Gy%H 	ZIy%J 	eKy%L 	|My%N 	2Oy%P 	?Qy%R 	WSy%T 	WUy%V 	

3Wy%X 	1Yy%Z 	

.[y%\ 	1]y%^ 			-_y%` 			-ay%b 	

.cy%d 	

.ey%f 	'gy%h 	.iy%j 			)ky%l 	

*my%n 	8oy%p 	@qy%r 	Wsy%t 	

YeQUYuy%v 	Ewy%x 	 Pyy%z 	

.{y%| 	0}y%~ 	;y%@ 	OAy%B 	8Cy%D 			-Ey%F 	2Gy%H 	

Y]bIy%L 			4My%N 	9Oy%P 			-Qy%R 	)Sy%T 	'Uy%V 	Wy%X 	Yy%Z 	'[y%\ 	)]y%^ 	)_y%` 	(ay%b 	)cy%d 	(ey%f 	)gy%h 	(iy%j 	)ky%l 	)my%n 	)oy%p 	)qy%r 	0sy%t 			Iuy%v 	Awy%x 	Hyy%z 	8{y%| 	4}y%~ 	6y%@ 	/Ay%B 	!1Cy%D 	!1Ey%F 	!1Gy%H 	!1Iy%J 	,,.KKy%L 	,,.KMy%N 	,,.KOy%P 	,,.KQy%R 	/Sy%T 	,Uy%V 	+Wy%X 	,Yy%Z 	-[y%\ 	.]y%^ 	,_y%` 	-ay%b 	-cy%d 	 Aey%f 	!Bgy%h 	/iy%j 	**,Iky%l 	++-Jmy%n 	*oy%p 	+qy%r 	*sy%t 	+uy%v 	++-Jwy%x 	++-Jyy%z 	-{y%| 	 0}y%~ 	!!#Dy%@ 	-Ay%B 	!OCy%D 	((*:Ey%F 	((*:Gy%H 	((*:Iy%J 	((*:Ky%L 	""$7My%N 	,Oy%P 	-Qy%R 	!>Sy%T 	+Uy%V 	-Wy%X 	//1AYy%Z 	//1A[y%\ 	446S]y%^ 	446S_y%` 	446Say%b 	446Scy%d 	,ey%f 	@gy%h 	))+;iy%j 	@ky%l 	>my%n 	<oy%p 	!qy%r 	

+sy%t 	Kuy%v 			-wy%x 	

.yy%z 	 3{y%| 	  "<}y%~ 	:y%@ 	HAy%B 	JCy%D 	

*Ey%F 	

KGy%H 	%Iy%J 	5Ky%L 	1My%N 	5Oy%P 	 eQy%R 	%%'aSy%T 	

:Uy%V 	!!NT%YWy%\ 	

:]y%^ 	2_y%` 	/ay%b 	-cy%d 	<ey%f 	hgy%h 	  "giy%j 	6ky%l 	;my%n 	Loy%p 	%%'Wqy%r 	8sy%t 	1uy%v 			-wy%x 	2yy%z 	;{y%| 	2}y%~ 	9y%@ 	eAy%B 	5Cy%D 	@Ey%F 	Gy%H 	""OIy%J 	/Ky%L 	oMy%N 	QOy%P 	'')>Qy%R 	FSy%T 	%CUy%V 	>Wy%X 	1Yy%Z 	!!#@[y%\ 	6]y%^ 	?_y%` 	Nay%b 	<cy%d 	##%Hey%f 	0gy%h 	oiy%j 	9ky%l 	2my%n 	_oy%p 	Oqy%r 	Osy%t 	?uy%v 	wy%x 	yy%z 	{y%| 	}y%~ 	1y%@ 	/Ay%B 	ACy%D 	/Ey%F 	3Gy%H 	4Iy%J 	4Ky%L 	2My%N 	3Oy%P 	3Qy%R 	1Sy%T 	2Uy%V 	2Wy%X 	1Yy%Z 	2[y%\ 	2]y%^ 	._y%` 	-ay%b 	.cy%d 	/ey%f 	Ogy%h 	0iy%j 	ky%l 	3my%n 	oy%p 	?qy%r 	.sy%t 	/uy%v 	/wy%x 	5yy%z 	0{y%| 	2}y%~ 	y%@ 	Ay%B 	/Cy%D 	Ey%F 	7Gy%H 	4Iy%J 	_Ky%L 	AMy%N 	1Oy%P 	/Qy%R 	/Sy%T 	/Uy%V 			?Wy%X 			?Yy%Z 	&&[y%\ 	**22O]y%^ 	o_y%` 	ay%b 	_cy%d 	oey%f 	gy%h 	iy%j 	ky%l 	--/pmy%n 	**,Voy%p 	_qy%r 	sy%t 	ouy%v 	wy%x 	yy%z 	{y%| 	}y%~ 	y%@ 	Ay%B 	##_Cy%D 	Ey%F 	Gy%H 	Iy%J 	  /Ky%L 	My%N 	  /Oy%P 	##_Qy%R 	  /Sy%T 	$$oUy%V 	  /Wy%X 	Yy%Z 	[y%\ 	_]y%^ 	o_y%` 	ay%b 	_cy%d 	  /ey%f 	$$ogy%h 	oiy%j 	ky%l 	_my%n 	_oy%p 	''//qy%r 	Nsy%t 	ouy%v 	owy%x 	yy%z 	{y%| 	_}y%~ 	_y%@ 	OAy%B 	_Cy%D 	OEy%F 	=Gy%H 	Iy%J 	Ky%L 	/My%N 	=Oy%P 	0Qy%R 	8Sy%T 	9Uy%V 	kWy%X 	E4I4IMYy%Z 	0E0EI[y%\ 	0E0EI]y%^ 	0E0EI_y%` 	MTMay%b 	cy%d 	6ey%f 	e6M6MQgy%h 	>iy%j 	

u/D/DHky%l 	0E0EImy%n 	

u/D/DHoy%p 	

u/D/DHqy%r 	sy%t 	/uy%v 	!Owy%x 	

Oyy%z 	/{y%| 	%2G2GK}y%~ 	53H3HLy%@ 	_Ay%B 	,Cy%D 	0Ey%F 	HHGy%H 	,Iy%J 	5Ky%L 	1F1FJMy%N 	%2G2GKOy%P 	@Qy%R 	?Sy%T 	0E0EIUy%V 	1F1FJWy%X 	/Yy%Z 	[y%\ 	

u/D/DH]y%^ 	__y%` 	oay%b 	_cy%d 	/ey%f 	1gy%h 	/iy%j 	_ky%l 	MTMmy%n 	0oy%p 	0E0EIqy%r 	6sy%t 	5uy%v 			(wy%x 	@yy%z 	E{y%| 	?}y%~ 	y%@ 	""OAy%B 	--Cy%D 	%%Ey%F 	Gy%H 	oIy%J 	,Ky%L 	?My%N 	GOy%P 	Qy%R 	LDLSy%T 	5Uy%V 	3Wy%X 	3Yy%Z 	113H[y%\ 	,]y%^ 	-_y%` 	Bay%b 	1cy%d 	-ey%f 	-gy%h 	0iy%j 	  "8ky%l 	Omy%n 	[oy%p 	?qy%r 	osy%t 	1F1FJuy%v 	_wy%x 	Wyy%z 	?{y%| 	1}y%~ 	&&(Wy%@ 	GAy%B 	'')QCy%D 	OEy%F 	Gy%H 	Iy%J 	Ky%L 	_My%N 	1Oy%P 	+Qy%R 			EUZUjUjnSy%T 	IIUy%V 	GWy%X 	/Yy%Z 	[y%\ 	/]y%^ 	._y%` 	=ay%b 	7cy%d 	ey%f 	+gy%h 	.iy%j 	oky%l 	7my%n 	  /oy%p 	Fqy%Cv D#%G		 1 JJJJ1::$AJJ%AJJ%
 ::  , jjZ!12GLLw%$&$&   	D64.D~$c/d'6IT
	)2 JJtJr2   
dispatcherc                       fd}|S )a  Wraps a given function with ``__torch_function__`` -related functionality.

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

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

    Examples
    --------
    >>> def dispatcher(a): # Must have the same signature as func
    ...     return (a,)
    >>> @torch.overrides.wrap_torch_function(dispatcher)
    >>> def func(a): # This will make func dispatchable by __torch_function__
    ...     return a + 0
    c                 J     t        j                          fd       S )Nc                  Z     | i |}t        |      rt        |g| i |S  | i |S r  )r   r   )r-   r.   relevant_argsr]
  r    wrappeds      r/   rb
  z3wrap_torch_function.<locals>.inner.<locals>.wrapped  sA    &77M!-0,WmUdUfUU(((r2   )	functoolsr   )r    rb
  r]
  s   `@r/   r  z"wrap_torch_function.<locals>.inner  s%    			) 
	) r2   r8  )r]
  r  s   ` r/   r   r     s    .	 Lr2   ra
  get_type_fnc                    |t         }t        j                  j                         sg S t	               }g }| D ]  } ||      }||vst        |d      s|j                  t        j                  j                  k7  sE|rX|j                  |       t        |      }t        |      D ]  \  }}t        | ||            s|} n |j                  ||       |h}|g} |S )a
  Returns a list of arguments on which to call __torch_function__.

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

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

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

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

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

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

    .. _NEP-0018:
       https://numpy.org/neps/nep-0018-array-function-protocol.html
    r   )r	  r4   _C_is_torch_function_enabledsethasattrr   _disabled_torch_function_implr!  rU
  	enumerate
issubclassinsert)	ra
  rd
  overloaded_typesoverloaded_argsargarg_typer  iold_args	            r/   _get_overloaded_argsrt
    s    D  88..0	"%%!#O (s# ,,CW1X++uxx/U/UU   $$X. O,"+O"< JAw!(K,@A !  &&uc2$,: #&%3(4 r2   
public_apic           	         t        |      }t        t        t        |            }t	               r1t               5 }|j                  | |||      }ddd       t        ur|S |D ]u  }|j                  }	t        |	d      rD|	j                  |u r6|	t        j                  j                  urt        j                  dt                |	| |||      }|t        uss|c S  | j                    d| j"                   }
d|
 d|D cg c]  }t        |       c} }t	               r|dt%                z  }t'        |      # 1 sw Y   xY wc c}w )a=  Implement a function with checks for ``__torch_function__`` overrides.

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

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

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

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

    Example
    -------
    >>> def func(a):
    ...     if has_torch_function_unary(a):
    ...         return handle_torch_function(func, (a,), a)
    ...     return a + 0
    N__self__zDefining your `__torch_function__ as a plain method is deprecated and will be an error in future, please define it as a classmethod..zno implementation found for 'z.' on types that implement __torch_function__: z nor in mode )rt
  r  mapr	  r   _pop_mode_temporarilyr   NotImplementedri
  rw
  r4   rf
  rj
  r)   warnDeprecationWarning
__module__rS
  _get_current_function_mode	TypeError)ru
  ra
  r-   r.   ro
  typesr{  resultoverloaded_argtorch_func_method	func_namerp
  r  s                r/   r   r   $  sf   L +=9O#dO,-E '( #$ 	N,,ZfMF	N'M *  +==$j16G6P6PTb6b!)O)OOMM [,. #:udFC'M" (():+>+>*?@I
'	{ 35DEcS	EF	H  '(9;<==
C.=	N 	N4  Fs   D9=E
9Ea  Check for __torch_function__ implementations in the elements of an iterable
    or if a __torch_function__ mode is enabled.  Considers exact ``Tensor`` s
    and ``Parameter`` s non-dispatchable.  Use this to guard a call to
    :func:`handle_torch_function`; don't use it to test if something
    is Tensor-like, use :func:`is_tensor_like` instead.
    Arguments
    ---------
    relevant_args : iterable
        Iterable or arguments to check for __torch_function__ methods.
    Returns
    -------
    bool
        True if any of the elements of relevant_args have __torch_function__
        implementations, False otherwise.
    See Also
    ________
    torch.is_tensor_like
        Checks if something is a Tensor-like, including an exact ``Tensor``.
    zSpecial case of `has_torch_function` for single inputs.
    Instead of:
      `has_torch_function((t,))`
    call:
      `has_torch_function_unary(t)`
    which skips unnecessary packing and unpacking work.
    a'  Special case of `has_torch_function` that skips tuple creation.

    This uses the METH_FASTCALL protocol introduced in Python 3.7

    Instead of:
      `has_torch_function((a, b))`
    call:
      `has_torch_function_variadic(a, b)`
    which skips unnecessary packing and unpacking work.
    c                  d   t        j                  t              } i }dt        t        j                  fdt        j
                  t        j
                  j                  fdt        j                  j
                  t        t        j                  j
                        fdt        j                  j                  t        t        j                  j                        fdt        j                  t        t        j                        fdt        j                  t        t        j                        fdt        j                  t        t        j                        fdt        j                  t        t        j                        fg}|D ],  \  }}}|D ]  }d	}|t        j                  urV|j                  d
      r*|j                  d      rd}nW|j                  d      rd}nC|d   j                         sd}n-|dk(  r(nt!        ||      }t!        t"        |d       |k(  r|dk(  rt!        ||      }|t        j                  u rt!        t"        |d       |k(  rt%        |t&        j(                        rt%        |t*        j,                        rt/        |      st1        |d      r| d| d||j2                  <   | d| d||j4                  <   |rC|j2                  t7               v r;d}	|j2                  t9               vs!J |	j;                  ||j<                               | |   j?                  |j2                         t/        |      s| d| ||<   |r|t7               v r1d}	|t9               vs!J |	j;                  ||j<                               | |   j?                  |       " / | |fS )Nr4   ztorch.functionalztorch.nn.functionalztorch.nn.initztorch.Tensorztorch.linalgz	torch.fftztorch.specialFr  r  Tr   
unique_dim__weakref__r$  rx
  z.__get__z.__set__zk{}.{} is in the tuple returned by torch._overrides.get_ignored_functions but still has an explicit override) collectionsdefaultdictlistr4   __all__r   r   dirr   r5   re  r   r5	  rT
  endswithislowerrW
  object
isinstancer
  
ModuleType
__future___Featurern  ri
  r$  __set__r   r   formatrS
  rQ  )
overridable_funcsr  tested_namespacesnamespace_str	namespacens_funcsr
  r&   r    r  s
             r/   _get_overridable_functionsr
    s*   #//5E	%'	U--u/?/?/G/GH	 3 3S9L9L5MN	%((--UXX]]);<	s5<<'89	s5<<'89	eiiUYY0	%--U]]);<	 /@ ;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#::@C<</D/FFl

S\^b^k^kHllF%d+224<<@D>*O1YK8E$K ,..<#8#::`CJJyRVR_R_<``:i(//5u:	6;6x 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
  r8  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
  r4   _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   rh
  r4   r5   )r
  methodss     r/   _get_tensor_methodsr
    s&     23#ELL12GNr2   c                 <    | t               v xs | j                  dk(  S )aw  
    Returns True if the function passed in is a handler for a
    method or property belonging to ``torch.Tensor``, as passed
    into ``__torch_function__``.

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    Independent subclasses of :class:`TorchFunctionMode` are compositional:
    modes can be pushed onto a stack using ``with MyMode():``.
    When you call functions in the PyTorch API inside your
    ``__torch_function__`` implementation, by default, they will forward on to
    the next mode on the mode stack.  If you want recursively call back into
    your current ``__torch_function__`` implementation, either explicitly
    invoke ``self.__torch_function__(...)``, or use the context manager
    ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
    API self-referential (beware of infinite loops, in this case!)
    r  c                      y r  r8  r  s    r/   r   zTorchFunctionMode.__init__s  s    r2   Nc                     t               r  )NotImplementedErrorr  r    r
  r-   r.   s        r/   r   z$TorchFunctionMode.__torch_function__v  s    !##r2   c                     t        |        | S r  )
_push_moder  s    r/   	__enter__zTorchFunctionMode.__enter__y  s    4r2   c                     t                y r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r/   __exit__zTorchFunctionMode.__exit__}  s    r2   c                 @    t        j                  d        | |i |}|S )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`)r)   r|
  )clsr-   r.   instances       r/   pushzTorchFunctionMode.push  s#    hi''r2   r8  N)rS
  r~
  __qualname____doc____annotations__r   r   r
  r
  classmethodr
  r8  r2   r/   r
  r
  R  s7    : $  r2   r
  c                  @    t               } | dkD  rt        | dz
        S d S )Nr   r  )r   r   )	stack_lens    r/   r
  r
    s%    )+I4=M!)a-0KtKr2   c                  d    t               } t        |       D cg c]  }t        |       c}S c c}w r  )r   r   r   )r
  rr
  s     r/    _get_current_function_mode_stackr
    s)    )+I/4Y/?@!"1%@@@s   -c                     t        |        y r  )r   )r{  s    r/   r
  r
    s
    !$'r2   c                      t               } | S r  )r   olds    r/   r
  r
    s    
#
%CJr2   c               #   `   K   t               } 	 |  t        |        y # t        |        w xY wwr  )r
  r
  r
  s    r/   rz
  rz
    s%     
+C	3
3s   . .+.c                       e Zd ZddZy)BaseTorchFunctionModeNc                     |i } ||i |S r  r8  r
  s        r/   r   z(BaseTorchFunctionMode.__torch_function__  s    >FT$V$$r2   r
  )rS
  r~
  r
  r   r8  r2   r/   r
  r
    s    %r2   r
  c               #      K   t         j                  j                         5  	 d  	 d d d        y # w xY w# 1 sw Y   y xY wwr  )r4   rf
  _RestorePythonTLSSnapshotr8  r2   r/   r   r     sC      
	+	+	- 		  	 s#   A61	A36?A)z.*is deprecated, please use.*r4   r  )9r
  r
  r
  rc
  r
  r)   typingr   r   r   r   r   r   r	   r
   r   
contextlibr4   torch._Cr   r   r   r   r   r   r   r   r   r
  r
  r1   	lru_cacher   r3  r   r   rt
  r   r   r   r   r
  r   r   r
  r	  r   r   r
  r
  r
  r
  r
  contextmanagerrz
  r
  r   r8  r2   r/   <module>r
     s  ,      H H H   % % % U\"NQ`h: TPs8} P  Pf Tc(m  2 TrtHh$67 r  rh!"H "HE EHcUTX[DY Eeijmen EPLL-5c]LPSL\ ! . '	  * 	  TI$E$sDN/B*CT(TW-EX*X$Y I$ I$V 	+4T(^(;#< 	+ 	+ 2 2& TS]   Gx GD G G0"KH2 2jL
A(
  %- %  r2   