+
    Pj                    $   R t ^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIH	t	H
t
 ^ RIHt ^ RIHtHtHt ^ RIHt ^ RIt^ RIHtHtHtHtHtHtHtHtHt . R0Ot]! R4      t]! R4      tR1R	 R
 llt ]PB                  ] R R l4       4       t"]PB                  R R l4       t#]PB                  ] R R l4       4       t$R R lt%R2R R llt&R R lt']! ]R4      t(]! ]R4      t)]! ]R4      t*]PB                  R R l4       t+] R R l4       t,] R 4       t-]PB                  R R  l4       t.] R! R" l4       t/R# t0 ! R$ R%4      t1R& t2R' t3R( t4R) t5]Pl                  R* 4       t7 ! R+ R,]14      t8]Pl                  R- 4       t9]Pl                  R. 4       t:R/ t;R# )3aE  
Python implementation of ``__torch_function__``

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

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

If changing this file in a way that can affect ``__torch_function__`` overhead,
please report the benchmarks in ``benchmarks/overrides_benchmark``. See the
instructions in the ``README.md`` in that directory.
N)CallableIterablewraps)AnycastTypeVar)	ParamSpec)	_add_docstr_get_function_stack_at_has_torch_function_has_torch_function_unary_has_torch_function_variadic_is_torch_function_mode_enabled_len_torch_function_stack_pop_torch_function_stack_push_on_torch_function_stack_P_Rc          
          V ^8  d   QhR\         \        \        3,          R\        R\        R\         \        \        3,          /# )   funcregexmodulereturn)r   r   r   str)formats   "h/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/torch/overrides.py__annotate__r   D   sB        
2r6
     b"f	     c                >   a aa \        S 4      R V VV3R ll4       pV# )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                d    V ^8  d   QhR\         P                  R\         P                  R\        /# r   argskwargsr   r   r#   r$   r   )r   s   "r   r   ,_disable_user_warnings.<locals>.__annotate__]   s)     ) )rww )")) ) )r   c            	         < \         P                  ! 4       ;_uu_ 4        \         P                  ! R \        SSR7       S! V / VB uuRRR4       #   + '       g   i     R# ; i)ignore)categorymessager   N)warningscatch_warningsfilterwarningsUserWarning)r#   r$   r   r   r   s   *,r   wrapper'_disable_user_warnings.<locals>.wrapper\   sG    $$&&##;f ((	 '&&&s   &AA!	r   )r   r   r   r/   s   fff r   _disable_user_warningsr1   D   s'    0 4[) ) ) Nr   c                :    V ^8  d   QhR\         \        ,          /# r   r   setr   )r   s   "r   r   r   i   s     _ _s8} _r   c                 %   \         P                  p 0 \         P                  k\         P                  k\         P                  k\         P
                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                   k\         P"                  k\         P$                  k\         P&                  k\         P(                  k\         P*                  k\         P,                  k\         P.                  k\         P0                  k\         P2                  k\         P4                  k\         P6                  k\         P8                  k\         P:                  k\         P<                  k\         P>                  k\         P@                  k\         PB                  k\         PD                  k\         PF                  k\         PH                  k\         PJ                  k\         PL                  k\         PN                  k\         PP                  k\         PR                  k\         PT                  k\         PV                  k\         PX                  k\         PZ                  k\         P\                  k\         P^                  k\         P`                  k\         Pb                  k\         Pd                  k\         Pf                  k\         Ph                  k\         Pj                  k\         Pl                  k\         Pn                  k\         Pp                  k\         Pr                  k\         Pt                  k\         Pv                  k\         Px                  k\         Pz                  k\         P|                  k\         P~                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  P                  k\         P                  P                  k\         P                  P                  k\         P                  P                  k\         P                  k\         P                  P                  k\         P                  P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  k\         P                  P                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP
                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                  k\         P                  EP                   EP                   k\         P                  EP"                  EP$                  k\         P                  EP"                  EP&                  k\         P                  EP"                  P                  k\         P                  EP"                  EP(                  k\         P                  EP"                  P                  k\         P                  EP"                  EP*                  k\         P                  EP"                  EP,                  k\         P                  EP"                  EP.                  k\         P                  EP"                  EP0                  k\         P                  EP"                  EP2                  k\         P                  EP"                  EP4                  k\         P                  EP"                  EP6                  k\         EP8                  EP:                  kE\
        kE\        k\         EP<                  k\         EP>                  k\         EP@                  k\         EPB                  k\         EPD                  k\         EPF                  k\         EPH                  k\         EPJ                  k\         EPL                  k\         EPN                  k\         EPP                  k\         EPR                  k\         EPT                  k\         EPV                  k\         EPX                  k\         EPZ                  k\         EP\                  k\         EP^                  k\         EP`                  k\         EPb                  k\         EPd                  k\         EPf                  k\         EPh                  k\         P                  EP                   EPj                  k\         EPl                  k\         EPn                  k\         EPp                  k\         EPr                  k\         EPt                  k\         EPv                  k\         EPx                  k\         EPz                  k\         EP|                  k\         EP~                  k\         EP                  k\         EP                  k\         EP                  k\         EP                  k\         EP                  k\         EP                  k\         EP                  k\         EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  EP                  kV EP                  EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                  kV EP                   kV EP                  kV EP                  kV EP:                  kV EP                  kV EP                  kV EP
                  kpE\        EP                  R8  d   VEP                  V EP                  4       V# )a  
Return public functions that cannot be overridden by ``__torch_function__``.

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

Examples
--------
>>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions()
True
>>> torch.add in torch.overrides.get_ignored_functions()
False
)      (
  torchTensortypename	is_tensor
is_storageset_default_tensor_typeset_default_deviceget_default_deviceset_rng_stateget_rng_statemanual_seedinitial_seedseedthread_safe_generatorsaveloadset_printoptionsforkget_default_dtypeget_num_interop_threadsget_num_threadsinit_num_threadsimport_ir_moduleimport_ir_module_from_bufferis_anomaly_enabledis_anomaly_check_nan_enabledis_grad_enabledmerge_type_from_type_commentparse_irparse_schemaparse_type_commentset_anomaly_enabledset_flush_denormalset_num_interop_threadsset_num_threadswait	as_tensor
from_numpytensordefault_generatorhas_cuda	has_cudnn
has_lapackdevicedtypefinfohas_mklhas_mps
has_mkldnn
has_openmpiinfomemory_formatqschemeset_grad_enabledno_gradenable_gradinference_modeis_inference_mode_enabledlayoutarange
as_stridedbartlett_windowblackman_windowbroadcast_shapescan_castcompilecudnn_affine_grid_generatorcudnn_batch_normcudnn_convolutioncudnn_convolution_transposecudnn_convolution_relucudnn_convolution_add_relucudnn_grid_samplercudnn_is_acceptablemiopen_ctc_lossemptyempty_permutedempty_stridedempty_quantizedexportregister_dataclasseyefftfftfreqrfftfreq	from_filefullfillhamming_windowhann_windowkaiser_windowlinspacelogspacemkldnn_adaptive_avg_pool2dmkldnn_convolutionmkldnn_max_pool2dmkldnn_max_pool3dmkldnn_linear_backward_weightsmkldnn_rnn_layernormalonespromote_typesrand	rand_likerandn
randn_likerandintrandint_likerandpermrangeresult_typescalar_tensorsparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensorsym_constrain_rangesym_constrain_range_for_sizesym_fresh_sizetril_indicestriu_indicesvanderzeros_jit_internalboolean_dispatchnn
functionalassert_int_or_pairupsampleupsample_bilinearupsample_nearesthas_torch_functionhas_torch_function_unaryhas_torch_function_variadichandle_torch_function
grouped_mmscaled_grouped_mm	scaled_mmsigmoidhardsigmoidtanh_canonical_mask_none_or_dtypeinitcalculate_gainuniformconstantdiracxavier_uniformxavier_normalkaiming_uniformkaiming_normal
orthogonalsparsenestedto_padded_tensorset_autocast_enabledis_autocast_enabledset_autocast_dtypeget_autocast_dtypeclear_autocast_cacheset_autocast_cpu_enabledis_autocast_cpu_enabledset_autocast_xla_enabledis_autocast_xla_enabledset_autocast_ipu_enabledis_autocast_ipu_enabledset_autocast_cpu_dtypeget_autocast_cpu_dtypeset_autocast_ipu_dtypeget_autocast_ipu_dtypeget_autocast_gpu_dtypeset_autocast_gpu_dtypeget_autocast_xla_dtypeset_autocast_xla_dtypeautocast_increment_nestingautocast_decrement_nestingis_autocast_cache_enabledset_autocast_cache_enabled	hardswishis_vulkan_available$are_deterministic_algorithms_enableduse_deterministic_algorithms-is_deterministic_algorithms_warn_only_enabledset_deterministic_debug_modeget_device_moduleget_deterministic_debug_modeset_float32_matmul_precisionget_float32_matmul_precisionunify_type_listis_warn_always_enabledset_warn_alwaysvmapcond
frombufferasarray_functional_sym_constrain_range_make_dep_token__delitem____dir____getattribute____init____iter____init_subclass____delattr____setattr____torch_function____torch_dispatch____new__	__class____subclasshook____hash__as_subclasseiglstsq	reinforcenew
new_tensor	new_emptynew_empty_strided	new_zerosnew_onesnew_full_make_subclasssolvesymeigstride	unflattento_sparse_cooto_sparse_csrto_sparse_cscto_sparse_bsrto_sparse_bsc
_to_sparse_to_sparse_csr_to_sparse_csc_to_sparse_bsr_to_sparse_bsc_typed_storage_reduce_ex_internal_fix_weakref
_view_func_view_func_unsafe_rev_view_func_unsafe_dtensor__new___make_wrapper_subclass_python_dispatch__get___has_symbolic_sizes_strides_conj_conj_physical_lazy_clone	_neg_view_is_zerotensor_is_all_true_is_any_true_addmm_activation
_use_count_philox_normal__philox_uniform_sysversion_infoaddr   )r:   	functionss     r   get_ignored_functionsrC  g   s    ( \\FGGG 	G 	%%	G
 	  G 	  G 	G 	G 	G 	G 	

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

G 	

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

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

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

mGn 	

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

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

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

GGH 	

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

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

wGx 	yGz 	{G| 	  }G~ 	G@ 	AGB 	CGD 	EGF 	GGH 	IGJ 	KGL 	MGN 	OGP 	QGR 	SGT 	UGV 	WGX 	YGZ 	[G\ 	]G^ 	_G` 	aGb 	cGd 	""eGf 	gGh 	iGj 	  kGl 	$$mGn 	oGp 	%%qGr 	''sGt 	**22uGv 	wGx 	yGz 	{G| 	}G~ 	G@ 	AGB 	CGD 	  EGF 	GGH 	IGJ 	KGL 	MGIR 7"f))*r   c                :    V ^8  d   QhR\         \        ,          /# r3   r4   )r   s   "r   r   r     s      c(m r   c                     \         P                  p V P                  P                  V P                  P                  V P
                  P                  0# )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).
)r9   r:   _baser2  grad_grad)r:   s    r   get_default_nowrap_functionsrI    s>    $ \\F r   c                F    V ^8  d   QhR\         \        \        3,          /# r3   )dictr   )r   s   "r   r   r     s      U UtHh$67 Ur   c                    \         P                  p / \         P                  ERR lb\         P                  ERR lb\         P                  R b\         P
                  R b\         P                  ERR lb\         P                  R b\         P                  ERR lb\         P                  ERR	 lb\         P                  ERR
 lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                   ERR lb\         P"                  ERR lb\         P$                  R b/ \         P&                  ERR lb\         P(                  ERR lb\         P*                  ERR lb\         P,                  ERR lb\         P.                  ERR lb\         P0                  ERR lb\         P2                  ERR lb\         P4                  ERR lb\         P6                  R b\         P8                  R b\         P:                  ERRR/R llb\         P<                  ERR  lb\         P>                  R! b\         P@                  ERR" lb\         PB                  ERR# lb\         PD                  ERR$ lb\         PF                  ERR% lbC/ \         PH                  ERR& lb\         PJ                  ERR' lb\         PL                  ERR( lb\         PN                  ERR) lb\         PP                  ERR* lb\         PR                  R+ b\         PT                  R, b\         PV                  R- b\         PX                  ERR. lb\         PZ                  ERR/ lb\         P\                  R0 b\         P^                  R1 b\         P`                  R2 b\         Pb                  R3 b\         Pd                  R4 b\         Pf                  R5 b\         Ph                  R6 bC/ \         Pj                  R7 b\         Pl                  ERR8 lb\         Pn                  R9 b\         Pp                  ERR; lb\         Pr                  ERR< lb\         Pt                  ERR= lb\         Pv                  ERR> lb\         Px                  ERR? lb\         Pz                  ERR@ lb\         P|                  ERRA lb\         P~                  ERRB lb\         P                  ERRC lb\         P                  RD b\         P                  ERRE lb\         P                  RF b\         P                  RG b\         P                  ERRH lbC/ \         P                  RI b\         P                  ERRJ lb\         P                  ERRK lb\         P                  ERRL lb\         P                  ERRM lb\         P                  ERRN lb\         P                  ERRP lb\         P                  RQR/RR lb\         P                  RS b\         P                  ERRT lb\         P                  P                  ERRU lb\         P                  P                  ERRV lb\         P                  ERRW lb\         P                  ERRX lb\         P                  RY b\         P                  ERRZ lb\         P                  ERR[ lbC/ \         P                  ERR\ lb\         P                  ERR] lb\         P                  ERR^ lb\         P                  ERR_ lb\         P                  ERR` lb\         P                  Ra b\         P                  ERRb lb\         P                  Rc b\         P                  ERRd lb\         P                  Re b\         P                  P                  ERRf lb\         P                  ERRg lb\         P                  ERRh lb\         P                  ERRi lb\         P                  ERRj lb\         P                  ERRk lb\         P                  ERRl lbC/ \         P                  ERRm lb\         P                  ERRn lb\         P                  Ro b\         P                  ERRp lb\         P                  ERRq lb\         P                  ERRr lb\         P                  ERRs lb\         P                  Rt b\         P                  ERRu lb\         P                  ERRv lb\         P                  ERRw lb\         P                  ERRx lb\         P                  Ry b\         P                  ERRz lb\         P                  P                  ERR{ lb\         P                  ER R| lb\         P                  ERR} lbC/ \         P                  ERR~ lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                  ERR lb\         P                  R b\         P                  R b\         P                  P                  R b\         EP                   R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP
                  ERR lb\         P                  EP
                  ERR lb\         EP                  ERR lbC/ \         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                   R b\         EP"                  ERR lb\         P                  EP$                  ERR lb\         P                  EP&                  ERR lb\         P                  EP(                  ERR lb\         P                  EP*                  ERR lb\         EP,                  R b\         EP.                  ERR lbC/ \         EP0                  ERR lb\         EP2                  ER	R lb\         EP4                  ERR lb\         EP6                  R b\         EP8                  ERR lb\         EP:                  ERR lb\         EP<                  ERR lb\         EP>                  ERR lb\         EP@                  ERR lb\         EPB                  ERR lb\         EPD                  R b\         EPF                  R b\         EPH                  ER
R lb\         EPJ                  R b\         EPL                  R b\         EPN                  R b\         EPP                  R bC/ \         EPR                  R b\         EPT                  R b\         EPV                  R b\         EPX                  R b\         EPZ                  R b\         EP\                  EP^                  ERR lb\         EP\                  EP`                  ERR lb\         EP\                  EPb                  ERR lb\         EP\                  EPd                  ERR lb\         EP\                  EPf                  ERR lb\         EP\                  EPh                  ERR lb\         EP\                  EPj                  ERR lb\         EP\                  EPl                  ERR lb\         EP\                  EPn                  ERR lb\         EP\                  EPp                  ERR lb\         EP\                  EPr                  ERR lb\         EP\                  EPt                  ERR lbC/ \         EP\                  EPv                  ERR lb\         EP\                  EPx                  ERR lb\         EP\                  EPz                  ERR lb\         EP\                  EP|                  ERR lb\         EP\                  EP~                  ERR lb\         EP\                  EP                  ERR lb\         EP\                  EP                  ERR lb\         EP\                  EP\                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                  R b\         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                  ERR lbC/ \         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  RR\         EP                  RR3R lb\         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lbC/ \         EP                  R b\         EP                  R b\         EP                  R b\         EP                  ERR lb\         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         P                  EP                  R b\         EP                  ERR lbC/ \         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                  R b\         EP                  ERR lb\         EP                  ERR lb\         EP                  R b\         EP                  ERER  lb\         EP                  ER b\         EP                  ER
ER lb\         EP                  ER b\         EP                  ER b\         EP                  ERER lbC/ \         EP                  ERER lb\         EP                   ER b\         EP                  ER b\         EP                  ERER	 lb\         P                  EP                  ERER
 lb\         P                  EP                  ERER lb\         EP
                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         EP                  ER bC/ \         EP                   ERER lb\         EP"                  ERER lb\         EP$                  ER b\         EP&                  ERER lb\         P                  EP(                  ERER lb\         P                  EP*                  ERER lb\         P                  EP,                  ERER lb\         EP.                  ERER lb\         EP0                  ERER lb\         EP2                  ERER  lb\         EP4                  ERER! lb\         EP6                  ERER" lb\         EP8                  ERER# lb\         EP:                  ERER$ lb\         EP<                  ERER% lb\         EP>                  ERER& lb\         EP@                  ERER' lbC/ \         EPB                  ERER( lb\         EPD                  ERER) lb\         EPF                  ERER* lb\         EPH                  ERER+ lb\         EPJ                  ERER, lb\         EPL                  ER- b\         EPN                  ERER. lb\         EPP                  ERER/ lb\         EPR                  ERER0 lb\         EPT                  ERER1 lb\         EPV                  ERER2 lb\         EPX                  ERER3 lb\         EPZ                  ERER4 lb\         EP\                  ER5 b\         EP^                  ERER6 lb\         EP`                  ERER7 lb\         EPb                  ERER8 lbC/ \         EPd                  ERER9 lb\         EPf                  ERER: lb\         EPh                  ERER; lb\         EPj                  ER< b\         EPl                  ER= b\         EPn                  ERER> lb\         EPp                  ERER? lb\         P                  EPd                  ERER@ lb\         P                  EPr                  ERERA lb\         P                  EPt                  ERERB lb\         P                  EPf                  ERERC lb\         P                  EPp                  ERERD lb\         EPv                  ERE b\         P                  EPv                  ERERF lb\         P                  EPx                  ERERG lb\         P                  EPz                  ERERH lb\         EP|                  ERI bC/ \         P                  EP|                  ERJ b\         EP~                  ERERK lb\         EP                  ERERL lb\         EP                  ERERM lb\         EP                  ERERN lb\         EP                  ERERO lb\         EP                  ERERP lb\         EP                  ERERQ lb\         EP                  ERERR lb\         EP                  ER ERS lb\         EP                  ERERT lb\         EP                  ERERU lb\         EP                  ERV b\         EP                  ERERW lb\         EP                  ERERX lb\         EP                  ERERY lb\         EP                  ERZ bC/ \         EP                  ER[ b\         EP                  ER\ b\         EP                  ER] b\         EP                  ER^ b\         EP                  ER_ b\         EP                  ER` b\         EP                  ERERa lb\         EP                  ER!ERb lb\         EP                  ERc b\         EP                  ERd b\         EP                  ERERe lb\         EP                  ERERf lb\         EP                  ERERg lb\         EP                  ERERh lb\         EP                  ERERi lb\         EP                  ERj b\         EP                  ERk bC/ \         EP                  ER"ERl lb\         EP                  ERm b\         EP                  ERn b\         EP                  ERo b\         EP                  ER#ERp lb\         EP                  ER$ERq lb\         EP                  ERr b\         EP                  ER%ERs lb\         EP                  ERt b\         EP                  ERERu lb\         EP                  ERERv lb\         EP                  ERERw lb\         EP                  ERERx lb\         EP                  ERERy lb\         EP                  EP                  EP                  ERz b\         EP                  EP                  EP                  ER{ b\         EP                  EP                  P
                  ERER| lbC/ \         EP                  EP                  EP                  ERER} lb\         EP                  EP                  EP                  ERER~ lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  P*                  ER&ER lb\         EP                  EP                  EP                  ER'ER lb\         EP                  EP                  EP                  ER'ER lb\         EP                  EP                  P\                  ER(ER lb\         EP                  EP                  Pn                  ERER lb\         EP                  EP                  EP                  ER)ER lb\         EP                  EP                  Pp                  ERER lb\         EP                  EP                  P                  ERER lb\         EP                  EP                  P                  ERER lb\         EP                  EP                  EP                  ER*ER lb\         EP                  EP                  EP                  ER+ER lbC/ \         EP                  EP                  P                  ER ER lb\         EP                  EP                  EP                  ER,ER lb\         EP                  EP                  EP                  ER,ER lb\         EP                  EP                  EP                  ER,ER lb\         EP                  EP                  EP                  ER,ER lb\         EP                  EP                  EP                   ERER lb\         EP                  EP                  EP.                  ERER lb\         EP                  EP                  EP0                  ER-ER lb\         EP                  EP                  EPX                  ER&ER lb\         EP                  EP                  EP                  ER.ER lb\         EP                  EP                  EP                  ER/ER lb\         EP                  EP                  EP                  ER/ER lb\         EP                  EP                  EP                  ER/ER lb\         EP                  EP                  EP
                  ER/ER lb\         EP                  EP                  EP                  ER0ER lb\         EP                  EP                  EP                  ER1ER lb\         EP                  EP                  EP                  ER2ER lbC/ \         EP                  EP                  EP                  ER3ER lb\         EP                  EP                  EP                  ER#ER lb\         EP                  EP                  EP                  ER4ER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ER5ER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                   ER6ER lb\         EP                  EP                  EP                  ER7ER lb\         EP                  EP                  EP"                  ERER lb\         EP                  EP                  EP                  ER8ER lb\         EP                  EP                  EP.                  ER#ER lb\         EP                  EP                  EP                  ER9ER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                   ER:ER lb\         EP                  EP                  EP@                  ER;ER lb\         EP                  EP                  EP"                  ER b\         EP                  EP                  EP$                  ERER lbC/ \         EP                  EP                  EP&                  ERER lb\         EP                  EP                  EP(                  ERER lb\         EP                  EP                  EPh                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP*                  ERER lb\         EP                  EP                  EP                  ERER lb\         EP                  EP                  EP,                  ERER lb\         EP                  EP                  EP.                  ER<ER lb\         EP                  EP                  EP0                  ER<ER lb\         EP                  EP                  EP2                  ER<ER lb\         EP                  EP                  EP4                  ER8ER lb\         EP                  EP                  EP6                  ER=ER lb\         EP                  EP                  EP8                  ER>ER lb\         EP                  EP                  EP:                  ER?ER lb\         EP                  EP                  EP<                  ER)ER lbC/ \         EP                  EP                  EP>                  ER@ER lb\         EP                  EP                  EP@                  ERAER lb\         EP                  EP                  EPB                  ER2ER lb\         EP                  EP                  EPD                  ERBER lb\         EP                  EP                  EPF                  ERCER lb\         EP                  EP                  EPH                  ERDER lb\         EP                  EP                  EPJ                  ER b\         EP                  EP                  EPL                  ERER lb\         EP                  EP                  EPN                  ERER lb\         EP                  EP                  EPP                  EREER lb\         EP                  EP                  EPR                  ERFER lb\         EP                  EP                  EPT                  ERER lb\         EP                  EP                  EPV                  ERER lb\         EP                  EP                  EPX                  ERER lb\         EP                  EP                  EPZ                  ERGER lb\         EP                  EP                  EP\                  ERHER lb\         EP                  EP                  EP^                  ERIER lbC/ \         EP                  EP                  EP`                  ER?ER lb\         EP                  EP                  EPb                  ER;ER lb\         EP                  EP                  EPd                  ER;ER lb\         EP                  EP                  EPf                  ERJER lb\         EP                  EP                  EPh                  ERER lb\         EP                  EP                  EPj                  ER b\         EP                  EP                  EPl                  ER b\         EP                  EP                  EPn                  ERER lb\         EP                  EP                  EPp                  ERKER lb\         EP                  EP                  EPr                  ERRERROERRERR:/ER lb\         EP                  EP                  EPt                  ER.ER lb\         EP                  EPv                  EPx                  ERLER lb\         EP                  EPv                  EPz                  ERLER lb\         EP                  EPv                  EP|                  ER b\         EP                  EPv                  EP~                  ERMER lb\         EP                  ERER lb\         EP                  ERER/ER lbC/ \         EP                  ER b\         EP                  ERNER lb\         P                  EP                  EROER lb\         P                  EP                  ERPER lb\         P                  EP                  ERQER lb\         EP                  ERRER lb\         EP                  ERNER lb\         EP                  ER b\         EP                  ER b\         EP                  ERSER lb\         EPF                  ERCER lb\         EP                  ER b\         EP                  ERTER lb\         EP                  ERER lb\         EP                  ERUER lb\         P                  EP                  ERVER lb\         EP                  ER bC/ \         EP                  ER b\         EP                  ERER lb\         EPH                  ER b\         EP                  ERER lb\         EP                  ERER lb\         EPJ                  ER b\         EP                  ER	ER lb\         EP                  ERER  lb\         EP                  ERER lb\         EP                  ERER lb\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         P                  EP                  ERWER	 lbC/ \         EP                  ERXER
 lb\         EP                  ERXER lb\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ER b\         EP                  ERZER lb\         EP                  ER[ER lb\         EP                  ER\ER lb\         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         EP                  ER b\         EP                  ERER lb\         EP                  ERER lbC/ \         P                  EP                  ERER lb\         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         EPL                  ERER lb\         EP                  ERER  lb\         EP                  ERER! lb\         EP                  ERER" lb\         EP                  ER# b\         EPP                  EREER$ lb\         EP                  ER% b\         EP                  ERER& lb\         EP                  ER' b\         EP                  ERER( lb\         EP                  ERER) lb\         EP                  ER^ER* lb\         EP                  ERER+ lbC/ \         EP                  ERER, lb\         EP                   ER- b\         EPR                  ERFER. lb\         EP                  ERER/ lb\         EP                  ERYER0 lb\         EP                  ERER1 lb\         EP                  ERER2RER3R/ER4 llb\         EP
                  ER5 b\         EP                  ERER6 lb\         EP                  ERER7 lb\         EP                  ER_ER8 lb\         EP                  ER9 b\         EP                  ER: b\         EP                  ER`ER; lb\         EP                  ER`ER< lb\         EPT                  ERER= lb\         EP                  ERER> lbC/ \         EP                  ERER? lb\         EP                  ERER@ lb\         EP                   ERERA lb\         EP"                  ERERB lb\         EP$                  ERERC lb\         EP&                  ERERD lb\         EP(                  ERE b\         P                  EP(                  ERF b\         EP*                  ERERG lb\         EP,                  ERERH lb\         EPb                  ERERI lb\         P                  EP.                  ERERJ lb\         P                  EP0                  ERERK lb\         EP2                  ERRRRQR/ERL llb\         EP4                  ERERM lb\         EP6                  ERERN lb\         EP8                  ERERO lbC/ \         EP:                  ERERP lb\         EP<                  ERERQ lb\         EP>                  ERERR lb\         EP@                  ERERS lb\         EPB                  ERERT lb\         EPD                  ERERU lb\         EPF                  ERaERV lb\         EPH                  ERERW lb\         EPJ                  ERERX lb\         EPL                  ERERY lb\         EPN                  ERZ b\         EPP                  ER[ b\         EPR                  ER\ b\         EPT                  ER] b\         EPV                  ER^ b\         EPX                  ER_ b\         EPZ                  ER` bC/ \         EP\                  ERa b\         EP^                  ERb b\         EP`                  ERc b\         EPb                  ERd b\         EPd                  ERe b\         EPf                  ERf b\         EPh                  ERg b\         EPj                  ERh b\         EPl                  ERi b\         EPn                  ERj b\         EPp                  ERERk lb\         EPr                  ERbERl lb\         EPt                  ERcERm lb\         P                  EPr                  ERERn lb\         P                  EPv                  ERERo lb\         EPx                  ERp b\         EPz                  ERq bC/ \         EP|                  EP~                  ERr b\         EP|                  EP                  ERs b\         EP|                  EP                  ERt b\         EP|                  EP                  ERu b\         EP|                  EP                  ERv b\         EP|                  EP                  ERERw lb\         EP|                  EP                  ERERx lb\         EP|                  EP                  ERERy lb\         EP|                  EP                  ERERz lb\         EP|                  EP                  ER{ b\         EP|                  EP                  ER| b\         EP|                  EP8                  ER} b\         EP|                  EP:                  ER~ b\         EP|                  EP                  ER b\         EP|                  EP<                  ER b\         EP|                  EP@                  ER b\         EP|                  EP                  ER bC/ \         EP|                  EPB                  ER b\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EP                  ER b\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EPD                  ER b\         EP|                  EP                  ER b\         EP|                  EP@                  ERER lb\         EP|                  EPX                  ER b\         EP|                  EPZ                  ERER lbC/ \         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ERER lb\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ER b\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EP                  ERER lb\         EP|                  EP$                  ER bC/ \         EP|                  EPb                  ERER lb\         EP|                  EP                  ER b\         EP|                  EP                  ERER lb\         EP|                  EPN                  ERER lb\         EP|                  EP                  ERER lb\         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         EP                  ERER lb\         EP                  ERER lb\         P                  EP                  ERER lb\         P                  EP                  ERER lb\         EP                  ERdER lb\         EP                  ERER lb\         EPn                  ERER lb\         EP                  ER b\         EP                  ER!ER lbC/ \         EP                  ER b\         EP                  ER b\         EP                  ERER lb\         EP                  ERER lb\         EP                  EReER lb\         P                  EP                  ERSER lb\         EP                  ERER lb\         EPp                  ERKER lb\         EP                  ERER lb\         EP                  ER b\         EP                  ERER lb\         EP                  ERER lb\         EP                  ER b\         EP                  ERfER lb\         EP                  ERER lb\         EP                   ER b\         EP                  ERER lbC/ \         EP                  ERER lb\         EP                  ERER lb\         EP                  ERER lb\         P                  EP
                  ERER lb\         EP                  ERER lb\         EP                  ERER lb\         EP                  ER b\         EP                  ERER lb\         EP                  ERER lb\         EP                  ER b\         EP                  ER b\         EP                  ER	ER lb\         EP                  ER b\         EP                  ER b\         EP                   ER b\         EP"                  ER b\         EP$                  ER bC/ \         EP&                  ER b\         EP(                  ERER lb\         EP*                  ER b\         EP,                  ERER lb\         EP.                  ERR/ER lb\         EP0                  ER b\         EP2                  ER b\         EP4                  ER b\         EP6                  ER b\         EP8                  ER b\         EP:                  ER`ER lb\         EP<                  ERER lb\         EP>                  ERER lb\         EP@                  ER b\         EPB                  ER b\         EPD                  ER b\         EPF                  ER bC/ \         EPH                  ER b\         EPJ                  ER b\         EPL                  ER b\         EPN                  ER b\         EPP                  ER b\         EPR                  ER b\         EPT                  ER b\         EPV                  ER b\         EPX                  ERER lb\         EPZ                  ER b\         EP\                  ER b\         EP^                  ER bV EP`                  ER bV EPb                  ER bV EPd                  ER bV EPf                  ER bV EPh                  ER bC/ V EPj                  ER bV EPl                  ER bV EPn                  ER bV EPp                  ER bV EPr                  ER bV EPt                  ER  bV EPv                  ER bV EPx                  ER bV EPz                  ER bV EP|                  ER bV EP~                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER	 bV EP                  ER
 bV EP                  ER bC/ V EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ERR/ER lbV EP                  ER bV EP                  ER bV EP                  EP                  ER bC/ V EP                  EP                  ER bV EP                  EP                  ER bV EP                  EP                  ER  bV EP                  EP                  ER! bV EP                  EP                  ER" bV EP                  EP                  ER# bV EP                  EP                  ER$ bV EP                  EP                  ER% bV EP                  EP                  ER& bV EP                  EP                  ER' bV EP                  EP                  ER( bV EP                  EP                  ER) bV EP                  EP                  ER* bV EP                  ER+ bV EP                  ER, bV EP                  ER- bV EP                  EP                  ER. bC/ V EP                  EP                  ER/ bV EP                  EP                  ER0 bV EP                  EP                  ER1 bV EP                  EP                  ER2 bV EP                  EP                  ER3 bV EP                  EP                  ER4 bV EP                  EP                  ER5 bV EP                  EP                  ER6 bV EP                  EP                  ER7 bV EP                  EP                  ER8 bV EP                  EP                  ER9 bV EP                  EP                  ER: bV EP                  EP                  ER; bV EP                  EP                  ER< bV EP                  EP                  ER= bV EP                  EP                  ER> bV EP                  EP                  ER? bC/ V EP                  EP                  ER@ bV EP                  EP                  ERA bV EP                  EP                  ERB bV EP                  EP                  ERC bV EP                  EP                  ERD bV EP                  EP                  ERE bV EP                   EP                  ERF bV EP                  EP                  ERG bV EP                  EP                  ERH bV EP                  EP                  ERI bV EP                  EP                  ERJ bV EP                  EP                  ERK bV EP                  EP                  ERL bV EP
                  EP                  ERM bV EP                  ERERN lbV EP                  ERO bV EP                  ERP bC/ V EP                  ERQ bV EP                  ERR bV EP                  ERS bV EP                  ERT bV EP                  ERU bV EP                  ERV bV EP                  ERW bV EP                   ERX bV P                  ERY bV EP"                  ERZ bV EP$                  ER[ bV EP&                  ER\ bV EP(                  ER/ER] lbV EP*                  \         EP,                  3ER^ lbV EP.                  \         EP,                  3ER_ lbV EP0                  \         EP,                  3ER` lbV EP2                  \         EP,                  3ERa lbC/ V EP4                  ER]ERbR/ERc llbV EP6                  ERd bV EP8                  ERe bV EP:                  \         EP<                  3ERf lbV EP>                  ERERg lbV EP@                  \         EP,                  3ERh lbV EPB                  \         EP,                  3ERi lbV EPD                  \         EP,                  3ERj lbV EPF                  \         EP,                  3ERk lbV EPH                  \         EP,                  3ERl lbV EPJ                  ERm bV EPL                  ERn bV EPN                  ERo bV EP                  ERERp lbV EPP                  ERq bV EPR                  ERERr lbV EPT                  \         EP,                  3ERs lbC/ V EPV                  \         EP,                  3ERt lbV EPX                  ERu bV EPZ                  ERv bV EP\                  ERw bV EP^                  ERYERbR/ERx llbV EP`                  ERy bV EPb                  ERz bV EPd                  \         EP,                  3ER{ lbV EPf                  \         EP,                  3ER| lbV EPh                  ERbR/ER} lbV EP                  ER~ bV EPj                  \         EP,                  3ER lbV EPl                  \         EP,                  3ER lbV EPn                  ER bV EPp                  \         EP,                  3ER lbV EPr                  ER bV EPt                  ER bC/ V EP                  ER bV EPv                  ER bV EPx                  ER bV EPz                  ER bV EP|                  ER bV EP~                  ERgERbR/ER llbV EP@                  ER bV EP                  \         EP,                  3ER lbV EP                  ER bV EP                  ER bV EP                  ERER lbV EP                  ERER lbV EP0                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bC/ V EP                  ER bV EPz                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ERER lbV EP                  ER bV EP                  ERERbR/ER llbV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ERER lbV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bC/ V EP                  ER bV EP                  ER bV EP                  ERhER lbV EP                  ER bV EP                  ER bV EP                  \         EP,                  3ER lbV EP                  ER bV EP                  ER`ER lbV EP                  ER bV EP                  ER bV EP                  ERER lbV EP                  ER bV EP                  ER bV EP>                  ERER lbV EP                  ER bV EP                  ER bV EP                  ER bC/ V EP                  ER bV EP                  ER bV EP                  ER bV EP                  RR\         EP,                  3ER lbV EP                  ERERR/ER llbV EP                  ERER lbV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EPt                  ER bV EPx                  ER]ER lbV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ER bV EP                  ERiER lbCV EP                  ER V EP                  ER \         P                  EP                  ERER l/Cp\         EP                  EP                  EP                  pE\        W4      '       d6   ERER lVE\        W4      &   ER VE\        V ERV 24      EP                  &   / pE\        4       pVEP                  4        EF>  w  rVVEP                  VEP                  ER,           ERVEP                  ,           ER,           ERVEP                  ,           ER,           ERVEP                  ,           ER,           .pVEP                  EP                  ER4      '       da   VEP                  E\        ER4      R pVEP                  ERV,           ER,           ERV,           ER,           ERV,           ER,           .4       V F8  p	E\        W	R4      p
E\        V
4      '       g   K$  W9  g   K,  W9  g   K4  WcV
&   K:  	  EKA  	  VEP                  V4       \         EP                  EP	                  4       '       Ed^   ^ REIEHp VEP                  / VEP                  ER/ER lbVEP                  ERER lbVEP                  ERjER lbVEP                  ERER lbVEP                  ERER lbVEP                  ERER lbVEP                  ERER lbVEP                  ERjER lbVEP                  ERjER lbVEP                  ERER lbVEP                  ERER lbVEP                  ER	ER lbVEP                   ERER lbVEP"                  ERkER lbVEP$                  ERkER lbVEP&                  ERkER lbVEP(                  ERkER lb4       V# (l  a:  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                     R#     inputouts   &&r   <lambda>'get_testing_overrides.<locals>.<lambda>      2r   c                     R# rN  rQ  rR  s   &&r   rU  rV        r   c                     R# rN  rQ  rS  output_sizes   &&r   rU  rV        br   c                     R# rN  rQ  )inputsr\  s   &&r   rU  rV        rr   c                     R# rN  rQ  rR  s   &&r   rU  rV        Br   c                     R# rN  rQ  rS  s   &r   rU  rV        Rr   c                     R# rN  rQ  rR  s   &&r   rU  rV        br   c                     R# rN  rQ  rR  s   &&r   rU  rV        Rr   c                     R# rN  rQ  rR  s   &&r   rU  rV        rr   c                     R# rN  rQ  rS  otherrT  s   &&&r   rU  rV        "r   c                     R# rN  rQ  rS  batch1batch2alphabetarT  s   &&&&&&r   rU  rV        rr   c                     R# rN  rQ  rS  tensor1tensor2valuerT  s   &&&&&r   rU  rV        "r   c                     R# rN  rQ  rx  s   &&&&&r   rU  rV    r|  r   c                     R# rN  rQ  rS  mat1mat2ru  rt  rT  s   &&&&&&r   rU  rV    r|  r   c                     R# rN  rQ  )rS  matvecru  rt  rT  s   &&&&&&r   rU  rV        r   c                     R# rN  rQ  )rS  vec1vec2ru  rt  rT  s   &&&&&&r   rU  rV        r   c                     R# rN  rQ  thetasizealign_cornerss   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  dims   &&r   rU  rV    rW  r   Fc                     R# rN  rQ  rS  rn  rtolatol	equal_nans   &&&&&r   rU  rV        VXr   c                     R# rN  rQ  rS  ptraininplaces   &&&&r   rU  rV        Br   c                     R# rN  rQ  r  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rS  r  keepdimrT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV        Br   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   stablec                    R# rN  rQ  )rS  r  
descendingr  s   &&&$r   rU  rV        PRr   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  )rS  msgs   &&r   rU  rV    rY  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV        Br   c                     R# rN  rQ  rm  s   &&&r   rU  rV        br   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rk  r   c                      R# rN  rQ  tensorss   *r   rU  rV    rW  r   c                      R# rN  rQ  r  s   *r   rU  rV    rW  r   c                      R# rN  rQ  r  s   *r   rU  rV    rW  r   c                     R# rN  rQ  )rS  kernel_sizer  padding	ceil_modecount_include_pads   &&&&&&r   rU  rV        vxr   c                     R# rN  rQ  rq  s   &&&&&&r   rU  rV    r  r   c	                     R# rN  rQ  )	rS  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	   &&&&&&&&&r   rU  rV        y{r   c                     R# rN  rQ  )grad_outrS  meaninvstdr  sum_dy
sum_dy_xmucount_tensors   &&&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  rS  r  r  r  input_gweight_gbias_gs   &&&&&&&&r   rU  rV        sur   c                     R# rN  rQ  )rS  r  r  r  r  r  s   &&&&&&r   rU  rV    rv  r   c                     R# rN  rQ  rS  r  r  r  r  r  r  counts   &&&&&&&&r   rU  rV        tvr   c                     R# rN  rQ  r  s   &&&&&&&&r   rU  rV    	      ACr   c                     R# rN  rQ  rS  r  s   &&r   rU  rV        2r   c                     R# rN  rQ  )rS  r  r  r  s   &&&&r   rU  rV        Z\r   c                     R# rN  rQ  )rS  	generatorrT  s   &&&r   rU  rV        r   c                     R# rN  rQ  input1input2r  r  s   &&&&r   rU  rV        Rr   r  c                     R# rN  rQ  rS  targetr  size_averagereduce	reduction
pos_weights   &&&&&&&r   rU  rV        rtr   c                     R# rN  rQ  )rS  weights	minlengths   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  probr  s   &&&r   rU  rV        Br   c                     R# rN  rQ  rm  s   &&&r   rU  rV        "r   c                     R# rN  rQ  rR  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV        r   c                     R# rN  rQ  rm  s   &&&r   rU  rV     r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV        "r   c                      R# rN  rQ  r  s   *r   rU  rV    rW  r   c                     R# rN  rQ  rS  r  	out_dtyperT  s   &&&&r   rU  rV    r  r   c                      R# rN  rQ  r  s   *r   rU  rV    ro  r   c                     R# rN  rQ  selfr  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  )rS  
boundaries	out_int32rightrT  s   &&&&&r   rU  rV        []r   c                      R# rN  rQ  r  s   *r   rU  rV    rk  r   c                     R# rN  rQ  r  r  rT  s   &&&r   rU  rV  	  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  
      rr   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )x1x2r  compute_modes   &&&&r   rU  rV        _ar   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r         ?c                     R# rN  rQ  rS  rt  r  s   &&&r   rU  rV    r  r   rT  c                     R# rN  rQ  )rT  matricess   $*r   rU  rV        r   c                     R# rN  rQ  rS  groupss   &&r   rU  rV        Rr   c                     R# rN  rQ  rS  upperrT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rS  check_errorsrT  s   &&&r   rU  rV        br   c                     R# rN  rQ  r+  s   &&&r   rU  rV        Rr   c                     R# rN  rQ  )r  r  r,  rT  s   &&&&r   rU  rV        Br   c                     R# rN  rQ  )rS  numeln_binsratio	bit_widths   &&&&&r   rU  rV        WYr   c                     R# rN  rQ  rS  chunksr  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  rS  minmaxrT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r@  s   &&&&r   rU  rV        r   c                     R# rN  rQ  )rS  rA  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  rB  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  rT  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  
correctionfweightsaweightss   &&&&r   rU  rV        Rr   c                     R# rN  rQ  rd  s   &r   rU  rV        2r   c                     R# rN  rQ  )rS  rwith_replacements   &&&r   rU  rV        rr   c                     R# rN  rQ  )realimags   &&r   rU  rV         "r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  !  r  r   c                     R# rN  rQ  )absangs   &&r   rU  rV  "      br   c                     R# rN  rQ  )rS  ords   &&r   rU  rV  #  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  $  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  %  r)  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  &  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  '  r  r   c                     R# rN  rQ  )rS  padr{  s   &&&r   rU  rV  (      2r   c                     R# rN  rQ  rS  r  r  r  r  dilationr(  s   &&&&&&&r   rU  rV  )      bdr   c                     R# rN  rQ  rg  s   &&&&&&&r   rU  rV  *  ri  r   c                     R# rN  rQ  rg  s   &&&&&&&r   rU  rV  +  ri  r   c	                     R# rN  rQ  )	rS  r  r  r  r  rh  
transposedoutput_addingr(  s	   &&&&&&&&&r   rU  rV  ,      uwr   c                     R# rN  rQ  )rS  r  r  rd  s   &&&&r   rU  rV  -  re  r   c                     R# rN  rQ  rS  r  r  r  r  output_paddingr(  rh  s   &&&&&&&&r   rU  rV  .  	      Ar   c                     R# rN  rQ  rr  s   &&&&&&&&r   rU  rV  /  rt  r   c                     R# rN  rQ  rr  s   &&&&&&&&r   rU  rV  0  rt  r   c                     R# rN  rQ  rd  s   &r   rU  rV  1  r\  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  2  rW  r   c                     R# rN  rQ  r  r  r  marginr  r  r  s   &&&&&&&r   rU  rV  3  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  4  rb  r   c                     R# rN  rQ  )r  r  r  r  s   &&&&r   rU  rV  5  r  r   c                     R# rN  rQ  rd  s   &r   rU  rV  6  rW  r   c                     R# rN  rQ  rS  rn  r  rT  s   &&&&r   rU  rV  7  r]  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  8      2r   c                     R# rN  rQ  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys   &&&&&&&r   rU  rV  :  r  r   c                     R# rN  rQ  rS  r  rT  s   &&&r   rU  rV  <  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  =  r  r   c                     R# rN  rQ  rS  r  rT  re   s   &&&&r   rU  rV  >  rD  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  ?  r`  r   c                     R# rN  rQ  yxr  s   &&&r   rU  rV  @  r]  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  A  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  B  rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV  C      r   c                     R# rN  rQ  rd  s   &r   rU  rV  D      r   c                     R# rN  rQ  rd  s   &r   rU  rV  E  r  r   c                     R# rN  rQ  rd  s   &r   rU  rV  F  r  r   c                     R# rN  rQ  rS  diagonalrT  s   &&&r   rU  rV  G  r%  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  H  r]  r   c                     R# rN  rQ  )rS  offsets   &&r   rU  rV  I  rY  r   c                     R# rN  rQ  )rS  nr  prependappendrT  s   &&&&&&r   rU  rV  J      TVr   c                     R# rN  rQ  rS  r  dim1dim2s   &&&&r   rU  rV  K  rD  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  L  r  r   c                     R# rN  rQ  )rS  srcr  r  r  s   &&&&&r   rU  rV  M  rM  r   c                     R# rN  rQ  )r  r  r  r  storage_offsets   &&&&&r   rU  rV  N  r;  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  O  rk  r   c                     R# rN  rQ  )rS  rn  r  s   &&&r   rU  rV  P  rg  r   c                     R# rN  rQ  rS  rn  rounding_moderT  s   &&&&r   rU  rV  Q      br   c                     R# rN  rQ  r  s   &&&&r   rU  rV  R  r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  S  ro  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  T  r]  r   c                     R# rN  rQ  rS  r  r
  s   &&&r   rU  rV  U  r%  r   c                     R# rN  rQ  )r  r  s   &&r   rU  rV  V      rr   c                     R# rN  rQ  rS  indices_or_sectionss   &&r   rU  rV  W  r  r   c                     R# rN  rQ  rH  s   &&r   rU  rV  X  rY  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  Y  ro  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  Z  r  r   c                     R# rN  rQ  rS  UPLOrT  s   &&&r   rU  rV  [  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  \  r  r   c                     R# rN  rQ  )equationoperandss   &*r   rU  rV  ]  ro  r   c                     R# rN  rQ  rS  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s   &&&&&&&r   rU  rV  _      z|r   c
                     R# rN  rQ  )
rS  r  offsetsr  r  r  moder   per_sample_weightsr  s
   &&&&&&&&&&r   rU  rV  b  s	      hjr   c                     R# rN  rQ  rS  re   rs   rd   requires_grads   &&&&&r   rU  rV  d      cer   c                     R# rN  rQ  rm  s   &&&r   rU  rV  e      r   c                     R# rN  rQ  rS  rn  s   &&r   rU  rV  f  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  g  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  h  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  i  rg  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  j  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  k  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  l  ri  r   c                     R# rN  rQ  )rS  scale
zero_pointaxis	quant_min	quant_maxs   &&&&&&r   rU  rV  m      mor   c                     R# rN  rQ  )rS  r  r  r  r  s   &&&&&r   rU  rV  n      fhr   c                     R# rN  rQ  )r  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants   &&&&&&&&&&&&&r   rU  rV  p  s	      ACr   c                     R# rN  rQ  rS  packed_weightr  outputs   &&&&r   rU  rV  r  r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  s      dfr   c                     R# rN  rQ  rS  r  packedcol_offsetsweight_scaleweight_zero_pointr  s   &&&&&&&r   rU  rV  t      {}r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV  v      ^`r   c                     R# rN  rQ  rd  s   &r   rU  rV  x  re  r   c                     R# rN  rQ  rd  s   &r   rU  rV  y  r  r   c                     R# rN  rQ  )rS  abs   &&&r   rU  rV  z  rD  r   c                     R# rN  rQ  rS  r  r  s   &&&r   rU  rV  {  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  |  r  r   c                     R# rN  rQ  rS  r  r  norms   &&&&r   rU  rV  }  r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  ~  r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  sr  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r3  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r)  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    rD  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  )rS  	start_dimend_dims   &&&r   rU  rV    r]  r   c                     R# rN  rQ  rS  dimss   &&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    rv  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rS  exponentrT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  )rS  
fill_valuerT  re   rs   rd   r  s   &&&&&&&r   rU  rV    s	      BDr   c                     R# rN  rQ  )rS  r  	dep_tokens   &&&r   rU  rV    r  r   c                     R# rN  rQ  )LU_data	LU_pivotsunpack_dataunpack_pivotss   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  indexrT  sparse_grads   &&&&&r   rU  rV    rM  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rK  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  spacingr  
edge_orders   &&&&r   rU  rV    r5  r   c                     R# rN  rQ  rS  gridinterpolation_modepadding_moder  s   &&&&&r   rU  rV        acr   c                     R# rN  rQ  rQ  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  rQ  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  
num_groupsr  r  r  r  s   &&&&&&r   rU  rV        kmr   c	                     R# rN  rQ  	rS  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	   &&&&&&&&&r   rU  rV        qsr   c                     R# rN  rQ  rS  r]  w_ihw_hhb_ihb_hhs   &&&&&&r   rU  rV    r5  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  lambds   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  r  rT  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  valuesrT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r{  r  r  r  s   &&&&&&r   rU  rV        xzr   c                     R# rN  rQ  )rS  binsrA  rB  rT  s   &&&&&r   rU  rV    r3  r   c                     R# rN  rQ  )rS  rw  rA  rB  r  densityrT  s   &&&&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  )rS  rw  r   r  ry  s   &&&&&r   rU  rV    r;  r   c                     R# rN  rQ  rS  taus   &&r   rU  rV    r  r   c                     R# rN  rQ  )r  r  rT  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r)  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rS  r  rA  sources   &&&&r   rU  rV    re  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  indicesrr  
accumulates   &&&&r   rU  rV    r|  r   c                     R# rN  rQ  )rS  r  rA  rT  s   &&&&r   rU  rV    rD  r   c                     R# rN  rQ  )rS  r  rA  r{  s   &&&&r   rU  rV    re  r   c                     R# rN  rQ  )rS  r  rA  r  r  include_inputs   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r_   s   &r   rU  rV    r  r   c                     R# rN  rQ  )eteassume_uniqueinverts   &&&&r   rU  rV    r3  r   c                     R# rN  rQ  r  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &r   rU  rV    re  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rY  r   c	                     R# rN  rQ  )	rS  r  r  r  r  use_input_statsr  r  r  s	   &&&&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ro  r   c                     R# rN  rQ  r/  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    re  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rb  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV        UWr   c                     R# rN  rQ  rd  s   &r   rU  rV    rO  r   c
                     R# rN  rQ  )
rS  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
   &&&&&&&&&&r   rU  rV    	      bdr   c                     R# rN  rQ  rS  r  r  r  r  
log_targets   &&&&&&r   rU  rV    s    prr   c                     R# rN  rQ  r  s   &&r   rU  rV        r   c                     R# rN  rQ  )rS  kr  r  rT  s   &&&&&r   rU  rV    r5  r   c                     R# rN  rQ  )rS  	hermitianr0  rT  s   &&&&r   rU  rV    rU  r   c                     R# rN  rQ  )rS  r  rT  s   &&&r   rU  rV    r|  r   c                     R# rN  rQ  )LDpivotsBr  rT  s   &&&&&r   rU  rV        QSr   c                     R# rN  rQ  )rS  normalized_shaper  r  r  r  s   &&&&&&r   rU  rV    rd  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  endr  rT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  )rS  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss   &&&&&&&&&&&&&&r   rU  rV    s	      IKr   c                     R# rN  rQ  rR  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rS  r  re   s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r%  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  rT  s   &&&r   rU  rV    rb  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  )rS  namesr  rT  s   &&&&r   rU  rV    rS  r   c	                     R# rN  rQ  )	databatch_sizesr]  r^  r_  r`  ra  r  rb  s	   &&&&&&&&&r   rU  rV     rd  r   c                     R# rN  rQ  rf  s   &&&&&&r   rU  rV    rM  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )Apivot	get_infosrT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )r	  r<  r=  rT  s   &&&&r   rU  rV    rD  r   c                     R# rN  rQ  rz  s   &&&&&&&r   rU  rV    rt  r   c                     R# rN  rQ  )rS  maskr{  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  rT  s   &&&r   rU  rV  	  re  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  
  r)  r   c                     R# rN  rQ  rS  r  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r0  rT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )LUr  r  leftadjointrT  s   &&&&&&r   rU  rV        Y[r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  rS  r  rT  s   &&&r   rU  rV    r]  r   c                     R# rN  rQ  )rS  r  r  s   &&&r   rU  rV        2r   c                     R# rN  rQ  rH  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r  r  rh  r  s   &&&&&&r   rU  rV        jlr   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r  r  rh  return_indicesr  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  )rS  r  r  re   rT  s   &&&&&r   rU  rV     r  r   c                     R# rN  rQ  r  s   &&r   rU  rV  !  rg  r   c                     R# rN  rQ  r  s   &&r   rU  rV  "  r  r   c                      R# rN  rQ  )r  r$   s   *,r   rU  rV  #  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  $  rW  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  %  r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  &  r  r   c                     R# rN  rQ  )rS  r  r  r  r  r  exponential_average_factorepsilons   &&&&&&&&r   rU  rV  (  r  r   c	                     R# rN  rQ  	rS  r  r  r  r  rh  r(  	benchmarkdeterministics	   &&&&&&&&&r   rU  rV  *  r  r   c	                     R# rN  rQ  )	rS  r  zrt  r  r  r  rh  r(  s	   &&&&&&&&&r   rU  rV  +  r  r   c                     R# rN  rQ  rg  s   &&&&&&&r   rU  rV  ,  r  r   c
                     R# rN  rQ  )
rS  r  r  r  rs  r  rh  r(  r'  r(  s
   &&&&&&&&&&r   rU  rV  .  ro  r   c	                     R# rN  rQ  r&  s	   &&&&&&&&&r   rU  rV  1      egr   c                     R# rN  rQ  )rS  r  weight_stride0r]  cxr  hidden_sizer`  rc  ra  r  rb  r  dropout_states   &&&&&&&&&&&&&&r   rU  rV  4  r  r   c                     R# rN  rQ  r	  s   &&&&r   rU  rV  6  rD  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  7  r  r   c                     R# rN  rQ  rS  r  destinations   &&&r   rU  rV  8  r  r   c                     R# rN  rQ  r7  s   &&&r   rU  rV  9  re  r   c                     R# rN  rQ  )rS  r  rT  s   &&&r   rU  rV  :  r`  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  ;  ro  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  <  r  r   c                     R# rN  rQ  )rS  num_samplesreplacementrT  s   &&&&r   rU  rV  =      SUr   c                     R# rN  rQ  )rS  r  rT  s   &&&r   rU  rV  >  rk  r   c                     R# rN  rQ  rS  r  s   &&r   rU  rV  ?  r  r   c                     R# rN  rQ  )rS  r  startr  s   &&&&r   rU  rV  @  r%  r   c                     R# rN  rQ  )rS  nanposinfneginfrT  s   &&&&&r   rU  rV  A  r  r   c                     R# rN  rQ  )rS  r  r  r  r  r  r  r  s   &&&&&&&&r   rU  rV  B  rd  r   c                     R# rN  rQ  )rS  r  r  r  r  r  s   &&&&&&r   rU  rV  C      ]_r   c                     R# rN  rQ  r  s   &&&r   rU  rV  D  r  r   c                     R# rN  rQ  rS  r  r  r  r  s   &&&&&r   rU  rV  E  r  r   c                     R# rN  rQ  rS  r  r  r  s   &&&&r   rU  rV  F  r;  r   c                     R# rN  rQ  )rS  r  r  NCHxWgroupr  s   &&&&&&&&r   rU  rV  G  r  r   c                     R# rN  rQ  )rS  r  r  r  re   s   &&&&&r   rU  rV  H  r@  r   c                     R# rN  rQ  r'  s   &&r   rU  rV  I  r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  J  r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  K  r%  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  L  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  M  rY  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  N  r%  r   c                     R# rN  rQ  r[  s   &&r   rU  rV  O  r5  r   c                     R# rN  rQ  r[  s   &&r   rU  rV  P  r5  r   c                     R# rN  rQ  rS  r\  r  s   &&&r   rU  rV  Q  ri  r   c                     R# rN  rQ  ra  s   &&&r   rU  rV  R      oqr   c                     R# rN  rQ  ra  s   &&&r   rU  rV  S  ri  r   c                     R# rN  rQ  ra  s   &&&r   rU  rV  T  rc  r   c                     R# rN  rQ  ra  s   &&&r   rU  rV  U  ri  r   c                     R# rN  rQ  ra  s   &&&r   rU  rV  V  rc  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  W  r  r   c                     R# rN  rQ  rS  r  r  r  s   &&&&r   rU  rV  X  r  r   c                     R# rN  rQ  rS  r  r  r  r  r  divisor_overrides   &&&&&&&r   rU  rV  Z  	      @Br   c                     R# rN  rQ  rl  s   &&&&&&&r   rU  rV  ]  rn  r   c                     R# rN  rQ  )rS  r  r  r  r  r  r  r  s   &&&&&&&&r   rU  rV  `  r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  b  r  r   c                     R# rN  rQ  rS  r  r  r  r  r  s   &&&&&&r   rU  rV  d  rU  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV  g  r  r   c                     R# rN  rQ  r"  s   &&&r   rU  rV  i  r|  r   c                     R# rN  rQ  rz  s   &&&&&&&r   rU  rV  k      gir   c                     R# rN  rQ  )rS  r  r  r  ignore_indexr  r  label_smoothings   &&&&&&&&r   rU  rV  n  	      JLr   c	                     R# rN  rQ  )	rS  linear_weightr  linear_biasr  r  ry  rz  optionss	   &&&&&&&&&r   rU  rV  q  s	      Y[r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV  t  r  r   c                     R# rN  rQ  rj  s   &&&&r   rU  rV  v      XZr   c                     R# rN  rQ  rj  s   &&&&r   rU  rV  w  r  r   c                     R# rN  rQ  rj  s   &&&&r   rU  rV  x  r  r   c                     R# rN  rQ  rj  s   &&&&r   rU  rV  y  r  r   c                     R# rN  rQ  r"  s   &&&r   rU  rV  z  r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV  |  r  r   c                     R# rN  rQ  )rS  r  r  r  r  r  r  r   r  include_last_offsetr  s   &&&&&&&&&&&r   rU  rV    s	      HJr   c                     R# rN  rQ  rj  s   &&&&r   rU  rV    rw  r   c                     R# rN  rQ  )rS  r\  r  rh  r  r  s   &&&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  rS  r  r\  output_ratior  _random_sampless   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  )rS  r  varr   r  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  approximates   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  rR  r  rT  r  s   &&&&&r   rU  rV    ru  r   c                     R# rN  rQ  )rS  rY  r  r  r  s   &&&&&r   rU  rV    r.  r   c                     R# rN  rQ  )logitsr}  hardr  r  s   &&&&&r   rU  rV    rU  r   c                     R# rN  rQ  rn  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  min_valmax_valr  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rt  s   &&&&&&r   rU  rV        `br   c                     R# rN  rQ  )rS  r  r  r  r  r  r  r  s   &&&&&&&&r   rU  rV    s	      GIr   c                     R# rN  rQ  )rS  r  scale_factorr  r  recompute_scale_factor	antialiass   &&&&&&&r   rU  rV    s	      KMr   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    rt  r   c                     R# rN  rQ  rS  r  r  r  r  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rO  s   &&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  )rS  negative_sloper  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  s   &&&r   rU  rV    r3  r   c                     R# rN  rQ  )rS  r  rt  ru  r  s   &&&&&r   rU  rV    r.  r   c                     R# rN  rQ  rS  r  _stacklevelre   s   &&&&r   rU  rV    s    \^r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r  r  r  s   &&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    rZ  r   c                     R# rN  rQ  rz  s   &&&&&&&r   rU  rV    rw  r   c                     R# rN  rQ  rS  r  r  r  rh  r  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r  r  r  r\  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ru  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  )querykeyr{  embed_dim_to_check	num_headsin_proj_weightin_proj_biasbias_kbias_vadd_zero_attn	dropout_pout_proj_weightout_proj_biasr  key_padding_maskneed_weights	attn_maskuse_separate_proj_weightq_proj_weightk_proj_weightv_proj_weightstatic_kstatic_vaverage_attn_weights	is_causals   &&&&&&&&&&&&&&&&&&&&&&&&&r   rU  rV    s	      ]_r   c                     R# rN  rQ  )rS  r  r  r{  r  r  r  r  s   &&&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  rs  s   &&&&&&r   rU  rV    rU  r   c                     R# rN  rQ  )rS  r  r  r  ry  r  r  s   &&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  r  rT  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )r_   num_classess   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  rd  r  r{  s   &&&&r   rU  rV    r1  r   c                     R# rN  rQ  r  r  r  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  	log_inputr   r  r  r  r  s   &&&&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rS  r  s   &&r   rU  rV    r`  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rD  r   c                     R# rN  rQ  rQ  s   &&&&r   rU  rV    rL  r   c                     R# rN  rQ  rS  lowerr,  r  r  s   &&&&&r   rU  rV    s    wyr   c                     R# rN  rQ  r  s   &&r   rU  rV    r`  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r`  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r`  r   c                     R# rN  rQ  )r  r  r{  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  r  r  ru  s   &&&&&&r   rU  rV    rt  r   c                     R# rN  rQ  )rS  r  r  deltar  s   &&&&&r   rU  rV        hjr   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  ru  	thresholds   &&&r   rU  rV    r|  r   c                     R# rN  rQ  rn  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rS  r  r{  r  s   &&&&r   rU  rV    r  r   c
                     R# rN  rQ  
anchorpositivenegativer{  r  r  swapr  r  r  s
   &&&&&&&&&&r   rU  rV    r{  r   distance_functionr{  r
  r  c                    R# rN  rQ  )r  r  r	  r  r{  r
  r  s   &&&$$$$r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  rh  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )r_   r  r	  r  s   &&&&r   rU  rV    rM  r   c                     R# rN  rQ  )r_   r  stdr  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )r_   vals   &&r   rU  rV    r)  r   c                     R# rN  rQ  )r_   r  r  nonlinearityr  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  as_tuples   &&r   rU  rV    r)  r   r8  c                    R# rN  rQ  )rS  r  r8  s   &$$r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rS  r  r  r  rT  re   s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r^  r  r  rT  re   s   &&&&&&r   rU  rV    ri  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    s     13r   c                     R# rN  rQ  )vpowr  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV    ri  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rO  r   c                     R# rN  rQ  r|  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  input3r  	transposes   &&&&&r   rU  rV    rv  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  qr  r  s   &&&&r   rU  rV  	  rS  r   c                     R# rN  rQ  rC  s   &&r   rU  rV  
  r  r   c                     R# rN  rQ  )rS  rconds   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r/  r  s   &&&r   rU  rV    rS  r   c                     R# rN  rQ  )rS  upscale_factors   &&r   rU  rV    re  r   c                     R# rN  rQ  )rS  downscale_factors   &&r   rU  rV    r`  r   c                     R# rN  rQ  )rS  r  s   &&r   rU  rV    r)  r   c                     R# rN  rQ  )rS  r  r  r   r  r  s   &&&&&&r   rU  rV    r;  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    ri  r   c                     R# rN  rQ  r2  s   &&&r   rU  rV    r)  r   c                     R# rN  rQ  )rS  re   s   &&r   rU  rV    rg  r   c                     R# rN  rQ  )rS  rA  r  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rY  r   c                     R# rN  rQ  rd  s   &r   rU  rV    ro  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    re  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  )rS  somerT  s   &&&r   rU  rV    r)  r   c                     R# rN  rQ  )rS  r  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rS  r,  r  r  interpolationrT  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rG  s   &&&&&&r   rU  rV     rw  r   c                     R# rN  rQ  )rS  scaleszero_pointsr  re   s   &&&&&r   rU  rV  !  r  r   c                     R# rN  rQ  )rS  r  r  re   s   &&&&r   rU  rV  "  r  r   c                     R# rN  rQ  )rS  re   reduce_ranges   &&&r   rU  rV  #  r1  r   c                     R# rN  rQ  )rS  r  r  r  r  r  output_scaleoutput_zero_points   &&&&&&&&r   rU  rV  $  rd  r   c                     R# rN  rQ  rS  r]  rg  rh  ri  rj  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs   &&&&&&&&&&&&&&r   rU  rV  &  	      _ar   c                     R# rN  rQ  rT  s   &&&&&&&&&&&&&&r   rU  rV  )  r]  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  ,  s     "r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  1  s     !#r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  7  s     !#r   c                     R# rN  rQ  rT  s   &&&&&&&&&&&&&&r   rU  rV  >  r]  r   c                     R# rN  rQ  rT  s   &&&&&&&&&&&&&&r   rU  rV  A  r]  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  C  rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV  D  rO  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  E  rb  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  F  r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV  G  r  r   c                     R# rN  rQ  rd  s   &r   rU  rV  H  rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV  I  ri  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  J  ro  r   c                     R# rN  rQ  r  s   &&r   rU  rV  K  r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV  L  r%  r   c                     R# rN  rQ  )rS  r  r  maxnormrT  s   &&&&&r   rU  rV  M  r`  r   c                     R# rN  rQ  r  s   &&r   rU  rV  N  r  r   c                     R# rN  rQ  )rS  shapes   &&r   rU  rV  O  rb  r   c                     R# rN  rQ  rQ  s   &&&&r   rU  rV  P  rv  r   c	                     R# rN  rQ  r\  s	   &&&&&&&&&r   rU  rV  Q  r  r   c                     R# rN  rQ  rf  s   &&&&&&r   rU  rV  R  r  r   c	                     R# rN  rQ  r\  s	   &&&&&&&&&r   rU  rV  S  r  r   c                     R# rN  rQ  rf  s   &&&&&&r   rU  rV  T  r  r   c                     R# rN  rQ  )rS  shiftsr+  s   &&&r   rU  rV  U  r)  r   c                     R# rN  rQ  )rS  r  r+  s   &&&r   rU  rV  V  r)  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  W  ri  r   c                     R# rN  rQ  rH  s   &&r   rU  rV  X  r  r   c                     R# rN  rQ  )r  r  compressed_indices_dtypes   &&&r   rU  rV  Y  r1  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV  Z  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  [  ri  r   c                     R# rN  rQ  )rS  rn  rt  s   &&&r   rU  rV  \  ro  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  ]  r  r   r{  r  c                    R# rN  rQ  )rS  r  rA  r  r{  r  s   &&&&$$r   rU  rV  ^  r;  r   c                     R# rN  rQ  )rS  r  rA  r  s   &&&&r   rU  rV  _  r  r   c                     R# rN  rQ  )rS  r  rA  r  r  include_selfs   &&&&&&r   rU  rV  `  r  r   c                     R# rN  rQ  )sorted_sequencerS  r  r  rT  s   &&&&&r   rU  rV  a  r  r   c                     R# rN  rQ  )r  r  lengthsr  r  r  unsafes   &&&&&&&r   rU  rV  b  r  r   c                     R# rN  rQ  )rS  r  rA  s   &&&r   rU  rV  c  rY  r   c                     R# rN  rQ  )rS  r  r  rA  s   &&&&r   rU  rV  d  r  r   c                     R# rN  rQ  rS  r  r  rE  r  steps   &&&&&&r   rU  rV  e  r  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  f  r  r   c                     R# rN  rQ  r  s   &&r   rU  rV  g  r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  h  rk  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  i  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  j  rk  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  k  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  l  rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  m  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  n  rb  r   c                     R# rN  rQ  rd  s   &r   rU  rV  o  re  r   c                     R# rN  rQ  rd  s   &r   rU  rV  p  rb  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  q  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  r  r%  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  s  r  r   c                     R# rN  rQ  )r  r  r  rT  s   &&&&r   rU  rV  t  r]  r   c                     R# rN  rQ  )r  r  r  r0  rT  s   &&&&&r   rU  rV  u  r  r   c                    R# rN  rQ  )rS  r  r  r  rT  s   &&&$$r   rU  rV  v  r;  r   c                     R# rN  rQ  r_   split_size_or_sectionsr  s   &&&r   rU  rV  w  r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  x  r1  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  y  rb  r   c                     R# rN  rQ  rR  s   &&r   rU  rV  z  rg  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  {  r  r   c                     R# rN  rQ  r  s   &&&&&&r   rU  rV  |  rM  r   c                     R# rN  rQ  r  s   &&&r   rU  rV  }  r  r   c                     R# rN  rQ  r  s   &&r   rU  rV  ~  rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  )rS  r  r  r  r  r  pad_moder  r  r  align_to_windows   &&&&&&&&&&&r   rU  rV    s	      ~@r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    re  r   c                     R# rN  rQ  r  r	  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    re  r   c                     R# rN  rQ  )r  r	  cs   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r#   s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  )rS  rD  
compute_uvrT  s   &&&&r   rU  rV    rS  r   c                     R# rN  rQ  )rS  r,  r  Ms   &&&&r   rU  rV    r`  r   c                     R# rN  rQ  )rS  full_matricesrT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rS  dim0r  s   &&&r   rU  rV    ro  r   c                     R# rN  rQ  )rS  axis0axis1s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    ri  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rk  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    ri  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rb  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r`  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    rD  r   c                     R# rN  rQ  rd  s   &r   rU  rV    ri  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    rS  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rg  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    rM  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r%  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r%  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r%  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r%  r   c                     R# rN  rQ  rC  s   &&r   rU  rV    r)  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r`  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r`  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r]  r   c                     R# rN  rQ  rd  s   &r   rU  rV    r  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r]  r   c                     R# rN  rQ  rm  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  rn  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    s    rr   c                     R# rN  rQ  )rS  rA  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  rT  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  )r  inds   &&r   rU  rV    r  r   c                     R# rN  rQ  )r  r	  r+  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r	  r+  rT  s   &&&&r   rU  rV    r%  r   c                     R# rN  rQ  )rS  r  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  r*  s   &&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r  r  rT  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  rd  s   &r   rU  rV    rO  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    rk  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r,  r'  unitriangulars   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r  r,  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r%  r   c
                     R# rN  rQ  r  s
   &&&&&&&&&&r   rU  rV    r{  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r%  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  rR  s   &&r   rU  rV    ri  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  )rS  r  sizess   &&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  sortedreturn_inversereturn_countsr  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  r*  r+  r  s   &&&&r   rU  rV    r.  r   c                     R# rN  rQ  )r  rr  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r=  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r|  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  rS  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  )	conditionr  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r   r  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )rS  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels   &&&&&&&r   rU  rV     r  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  levels   &&r   rU  rV    r  r   c                     R# rN  rQ  )primaltangentrC  s   &&&r   rU  rV    r]  r   c                     R# rN  rQ  r  s   &r   rU  rV    rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  )r  r  r  r  s   &&&&r   rU  rV  	  rv  r   c                     R# rN  rQ  r  s   &&r   rU  rV  
  r  r   c                     R# rN  rQ  )r  r  r  r  s   &&&&r   rU  rV    r  r   implicitc                    R# rN  rQ  )r  r  rO  s   &&$r   rU  rV    r  r   c                     R# rN  rQ  )r  r  rE  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r+  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  r  r  r  s   &&&r   rU  rV    r]  r   c                     R# rN  rQ  )r  r  rA  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  rE  r  r  s   &&&&&r   rU  rV    r5  r   c                     R# rN  rQ  )r  
split_sizer  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  split_sizesr  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r*  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rO  r   c                     R# rN  rQ  )r  r  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r*  s   &&r   rU  rV    rY  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV    ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV     ri  r   c                     R# rN  rQ  r*  s   &&r   rU  rV  !  rk  r   c                     R# rN  rQ  r  re   s   &&r   rU  rV  "  ri  r   c                     R# rN  rQ  r  	dimensionr  r  s   &&&&r   rU  rV  #  r`  r   c                     R# rN  rQ  rH  s   &r   rU  rV  $  r  r   c                     R# rN  rQ  r  rn  s   &&r   rU  rV  %  r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  &  ro  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  '  ro  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  (  rY  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  )  r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  *  r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  +  rk  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  ,  rY  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  -  rY  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  .  rk  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  /  rY  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  0  rY  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  1  rb  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  2  rW  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  3  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  4  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  5  r  r   c                     R# rN  rQ  rj  s   &&r   rU  rV  6  rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV  7  r\  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  8  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  9  re  r   c                     R# rN  rQ  rH  s   &r   rU  rV  :  r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  ;  rb  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  <  ri  r   c                     R# rN  rQ  rp  s   &&r   rU  rV  =  ri  r   c                     R# rN  rQ  )r  arrays   &&r   rU  rV  >  r  r   c                     R# rN  rQ  )r  idxs   &&r   rU  rV  ?  rg  r   c                     R# rN  rQ  )r  memos   &&r   rU  rV  @  rY  r   c                     R# rN  rQ  rH  s   &r   rU  rV  A  re  r   c                     R# rN  rQ  rH  s   &r   rU  rV  B  r\  r   c                     R# rN  rQ  rH  s   &r   rU  rV  C  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  D  re  r   c                     R# rN  rQ  )r  format_specs   &&r   rU  rV  E  r)  r   c                     R# rN  rQ  )r  protos   &&r   rU  rV  F  ro  r   c                     R# rN  rQ  rH  s   &r   rU  rV  G  rW  r   tensor_contentsc                    R# rN  rQ  )r  r  s   &$r   rU  rV  H  r`  r   c                     R# rN  rQ  )r  r  r   s   &&&r   rU  rV  I  rk  r   c                     R# rN  rQ  )r  ds   &&r   rU  rV  J  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  K  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  L  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  M  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  N  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  O  r)  r   c                     R# rN  rQ  rH  s   &r   rU  rV  P  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  Q  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  R  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  S  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  T  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  U  rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV  V  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  W  rY  r   c                     R# rN  rQ  rH  s   &r   rU  rV  X  rg  r   c                     R# rN  rQ  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes   &&&&&r   rU  rV  Y  s    npr   c                     R# rN  rQ  )r  r  r  s   &&&r   rU  rV  Z  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  [  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  \  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  ]  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  ^  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  _  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  `  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  a  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  b  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  c  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  d  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  e  ro  r   c                     R# rN  rQ  rH  s   &r   rU  rV  f  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  g  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  h  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  i  rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV  j  ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV  k  rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV  l  ro  r   c                     R# rN  rQ  rH  s   &r   rU  rV  m  rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV  n  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  o  rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV  p  rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV  q  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  r  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  s  rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV  t  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  u  rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV  v  r  r   c                     R# rN  rQ  rH  s   &r   rU  rV  w  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  x  rg  r   c                     R# rN  rQ  rH  s   &r   rU  rV  y  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  z  rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV  {  r]  r   c                     R# rN  rQ  )r  re   non_blockingr$   s   &&&,r   rU  rV  |  r5  r   c                     R# rN  rQ  rH  s   &r   rU  rV  }  rO  r   c                     R# rN  rQ  rH  s   &r   rU  rV  ~  rO  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rH  s   &r   rU  rV        "r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                     R# rN  rQ  )r  callables   &&r   rU  rV    rg  r   c                     R# rN  rQ  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  rT  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  gradientretain_graphcreate_graphr_  s   &&&&&r   rU  rV    s    ikr   c                     R# rN  rQ  r  rl   s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   r  c                    R# rN  rQ  )r  mediansigmar  s   &&&$r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r\  r   c                     R# rN  rQ  )r  	coalesceds   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rv  r   c                     R# rN  rQ  )r  r  r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r  r  r  s   &&&&&r   rU  rV    rM  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  ambiguity_checks   &&r   rU  rV    r]  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r|  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    rg  r   c                    R# rN  rQ  )r  ro  r  s   &&$r   rU  rV    r  r   c                     R# rN  rQ  r  r{  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                    R# rN  rQ  )r  r  r  s   &&$r   rU  rV    r]  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r_   s   &&r   rU  rV    rk  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                    R# rN  rQ  )r  r  r  r  s   &&&$r   rU  rV    r  r   c                     R# rN  rQ  r*  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rS  r   c                     R# rN  rQ  )r  r_   r  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r
  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  rn  assigns   &&&r   rU  rV    r]  r   c                     R# rN  rQ  )r  rm  rE  r  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r\  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rO  r   c                     R# rN  rQ  r*  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r_   r  s   &&&&r   rU  rV    r3  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                    R# rN  rQ  )r  from_tor  s   &&&$r   rU  rV    r|  r   c                     R# rN  rQ  )r  streams   &&r   rU  rV    r  r   c                     R# rN  rQ  r  hooks   &&r   rU  rV    r  r   c                     R# rN  rQ  r3  s   &&r   rU  rV    r  r   c                     R# rN  rQ  r  s   &*r   rU  rV    rW  r   c                     R# rN  rQ  )r  r  s   &&r   rU  rV    rD  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  r  s   &*r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rW  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    rg  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r  r  r  s   &&&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r  rA  s   &&&&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  r  rE  r  r  s   &&&&&&r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  r  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  )r  r  accumulate_matchess   &&&r   rU  rV    r  r   c                     R# rN  rQ  r  size1size2	dense_dims   &&&&r   rU  rV    r3  r   c                     R# rN  rQ  rI  s   &&&&r   rU  rV    rv  r   c                     R# rN  rQ  )r  r  r  ru  rt  rT  s   &&&&&&r   rU  rV    rM  r   c                     R# rN  rQ  rH  s   &r   rU  rV    re  r   c                     R# rN  rQ  rH  s   &r   rU  rV    ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rW  r   c                     R# rN  rQ  r  s   &&r   rU  rV    rk  r   c                     R# rN  rQ  )r  repss   &*r   rU  rV    r  r   c                     R# rN  rQ  )r  re   r  copyrl   s   &&&&&r   rU  rV    s    lnr   masked_gradc                    R# rN  rQ  r  re   rX  s   &&$r   rU  rV    rS  r   c                     R# rN  rQ  rZ  s   &&&r   rU  rV    r3  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rl  s   &&&&r   rU  rV    re  r   c                     R# rN  rQ  )r  r.  r/  s   &&&r   rU  rV    r)  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   c                     R# rN  rQ  )r  rr  s   &&r   rU  rV    r  r   c                     R# rN  rQ  rp  s   &&r   rU  rV    rb  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rO  r   c                     R# rN  rQ  )r  r1  max_version	dl_devicerW  s   &&&&&r   rU  rV    ri  r   c                     R# rN  rQ  rH  s   &r   rU  rV    rk  r   c                     R# rN  rQ  )r  r  r	  s   &&&r   rU  rV    r  r   c                     R# rN  rQ  )r  r	  r   drivers   &&&&r   rU  rV    r  r   c                     R# rN  rQ  )r  rd   r  r$   s   &&&,r   rU  rV     r  r   c                     R# rN  rQ  rH  s   &r   rU  rV    r  r   is______i__rbitwise_c                     R# rN  rQ  )r_   r  rV  async_op	group_srcs   &&&&&r   rU  rV  (  r.  r   c                     R# rN  rQ  )r_   oprV  rv  s   &&&&r   rU  rV  )  r  r   c                     R# rN  rQ  )r_   dstry  rV  rv  	group_dsts   &&&&&&r   rU  rV  *  rZ  r   c                     R# rN  rQ  )r  ry  rV  rv  s   &&&&r   rU  rV  +  r  r   c                     R# rN  rQ  )tensor_listr_   rV  rv  s   &&&&r   rU  rV  ,  r  r   c                     R# rN  rQ  )output_tensorinput_tensorrV  rv  s   &&&&r   rU  rV  -  r  r   c                     R# rN  rQ  )output_tensor_listsinput_tensor_listrV  rv  s   &&&&r   rU  rV  .  r  r   c                     R# rN  rQ  )r_   gather_listr{  rV  rv  r|  s   &&&&&&r   rU  rV  /  r  r   c                     R# rN  rQ  )r_   scatter_listr  rV  rv  rw  s   &&&&&&r   rU  rV  0  r  r   c                     R# rN  rQ  )r  
input_listry  rV  rv  s   &&&&&r   rU  rV  1  r.  r   c                     R# rN  rQ  )r  rS  ry  rV  rv  s   &&&&&r   rU  rV  2  rw  r   c                     R# rN  rQ  )r  rS  output_split_sizesinput_split_sizesrV  rv  s   &&&&&&r   rU  rV  3  s	      LNr   c                     R# rN  rQ  )output_tensor_listr  rV  rv  s   &&&&r   rU  rV  4  rZ  r   c                     R# rN  rQ  r_   r{  rV  tagr|  s   &&&&&r   rU  rV  5  r  r   c                     R# rN  rQ  r_   r  rV  r  rw  s   &&&&&r   rU  rV  6  r  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV  7  r;  r   c                     R# rN  rQ  r  s   &&&&&r   rU  rV  8  r;  r   N)rO  rO  N)rO  N)h㈵>:0yE>F)F)NFNrP  )rP  F)N    FT)NN)NNNr  N)Nr  )FFNr  N)       @#use_mm_for_euclid_dist_if_necessary)r   F)FNr  )NNN)rO  NN)r   F)NrO  r  rO  rO  )NrO  r  r  rO  rO  )r  NNr  )rO  r  )rP  N)r  r  F)NrP  )rO  rP  NNN)r  r  rO  )r  rP  )r   )LN)NNr  FF)Nr   Fr  FNN)NNNF)FF)NrP  N)Nr  rP  N)r  rP  )TT)NF)NNrO  )NNr  T)      ?)NFr  N)r   NNr  )d   r  r  N)r  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr  rO  F)Nr  rO  FF)NFNN)rP  FN)        NNN)NNr  )Nr  )r   NFN)r  FF)Nr  FTN)NNF皙?r  )NNNr  )NNNr  r  )NNr  Nr  N)r  TF)	NNr   Fr  FNFN)rO  r  rO  )NNFN)Fư>r  )none)rP  )bilinearr   N)rO  Fg|=rP  )g      r   F)NNNNTr  r  )NNnearestNNF)NNr  N)g{Gz?F)g-C6?g      ?r   )Nr7   N)Nr  N)TNTNFNNNNNNF)rO  r   NNNr  )NNr  )NNr  Nr  )r   rO  g-q=N)r   r  )r  r  F)TFNr  Nr  )Nr  )g      ?gUUUUUU?FF)Nr  )NNr  r   )r  r   N)rO     )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rO  )rQ  r  r  F)rQ  )r  r  )rO  rO  F)rQ  )r  r  r  )rO  rO  rO  Fr  rO  )rO  r  )rB  NNNr  F)r  NNrO  )	NNNTreflectFTNN)TTN)   r   N)r   N)TFF)TFFN)rO  r   )Nr  NN)NNNN)NNNFN)NNr  N(  r9   r:   rZ  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshrA  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diffr  diagonal_scatteras_strided_scatterdigammadistdivdividedotra  dsmmhsmmdsplitdstackr  eigvalseigheigvalsheinsum	embeddingembedding_bag
empty_likeeqequalerferfcerfinvexpexp2expm1 fake_quantize_per_channel_affinefake_quantize_per_tensor_affinefused_moving_avg_obs_fake_quantfbgemm_linear_fp16_weight)fbgemm_linear_fp16_weight_fp32_activationfbgemm_linear_int8_weight)fbgemm_linear_int8_weight_fp32_activationfbgemm_linear_quantize_weightfbgemm_pack_gemm_matrix_fp16fbgemm_pack_quantized_matrixfeature_alpha_dropoutfeature_dropoutr   ifftrfftirffthfftihffthfft2ihfft2hfftnihfftnfftnifftnrfftnirfftnfft2ifft2rfft2irfft2fftshift	ifftshiftfixflattenflipfliplrflipudfrobenius_normfloorfloor_dividefloat_powerfmodfracfrexp	full_likestrided_functional_assert_async	lu_unpackgathergcdge
get_devicegreater_equalgeqrfi0inneroutergerr  grid_samplergrid_sampler_2dgrid_sampler_3d
group_normgrugru_cellgtgreater
hardshrinkhash_tensor	heavisidehinge_embedding_losshistc	histogramhistogramddhouseholder_producthspmmhsplithstackhypotigammaigammacrV  	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_exprB  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridrA  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_fused_rms_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr	  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropylinear_cross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_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_hotrd  pairwise_distancepoisson_nll_lossprelurelurelu6rms_normrreluselusilumishscaled_dot_product_attentionsmooth_l1_loss
huber_losssoft_margin_losssoftmaxsoftminsoftplus
softshrinksoftsign
tanhshrinkr  triplet_margin_loss!triplet_margin_with_distance_lossunfoldr   uniform_normal_	constant_kaiming_uniform_nonzerononzero_staticargwherer  vector_normmatrix_normnorm_except_dimnuclear_normr7  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar  	ones_liker!  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2degravelrU  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_itesym_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer'  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr2  HmTmH_backward_hooks_post_accumulate_grad_hooksrF  _cdatarG  rH  _grad_fngrad_fn
grad_dtype_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  rd   re   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizers   namenbytesndim	output_nrr  rr  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_valuesapply_ru   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipuconst_data_ptrdata_ptrrL  r  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalfr  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rm   random_record_streamregister_hook"register_post_accumulate_grad_hookrepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_shortr  
sparse_dimsparse_mask_sparse_mask_projectionsparse_resize_sparse_resize_and_clear_storageuntyped_storager  storage_typesum_to_sizer/  to_dense	_to_dense	to_sparsetolist	to_mkldnntype_asrr  viewview_aszero_
__dlpack____dlpack_device__rA  r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrrC  items__name__
startswithlenextendr  updatedistributedis_availabletorch.distributed	broadcast
all_reducer  all_reduce_coalesced
all_gatherall_gather_singleall_gather_coalescedreduce_scatterreduce_scatter_singleall_to_all_single
all_to_allisendirecvsendrecv)r:   retprivateuse1_backend_nameret2ignoredr  r   r  subnamer
  r   r9  s               r   get_testing_overridesr    sY   6 \\Fx%		-x%2x% 	!!#@x% 	!!#A	x%
 	

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

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

.+x%, 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Z 

Z  		Y!" 		Y#	
, Jr   c                    V ^8  d   QhR\         \        \        \        ,          3,          R\         \         \        \        3,          .\         \        \        3,          3,          /# )r   
dispatcherr   )r   r   r   r   r   )r   s   "r   r   r   ?  sI     ' 'Xc]*+'xB (2r6"223'r   c                   a  R V 3R llpV# )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                t    V ^8  d   QhR\         \        \        3,          R\         \        \        3,          /# r   r   r   )r   r   r   )r   s   "r   r   )wrap_torch_function.<locals>.__annotate__Y  s,     / /HRV$ /"b&)9 /r   c                    <a a \         P                  ! S 4      R  VV V3R ll4       o\        \        \        \
        3,          S4      # )c                d    V ^8  d   QhR\         P                  R\         P                  R\        /# r"   r%   )r   s   "r   r   8wrap_torch_function.<locals>.inner.<locals>.__annotate__[  s)     	) 	)277 	)bii 	)B 	)r   c                     < S! V / VB p\        V4      '       d0   \        \        \        \        \
        3,          S4      V.V O5/ VB # S! V / VB # r  )r   r   r   r   r   r   )r#   r$   relevant_argsr  r   wrappeds   *, r   r  3wrap_torch_function.<locals>.inner.<locals>.wrappedZ  sa    &77M!-00,"b&)73]EIMS  (((r   )	functoolsr   r   r   r   r   )r   r  r  s   f@r   r  "wrap_torch_function.<locals>.innerY  s;    			) 	) 
	) HRV$g..r   rQ  )r  r  s   f r   wrap_torch_functionr#  ?  s    4/ / Lr   c                    V ^8  d   QhR\         \        ,          R\        \        .\        3,          R,          R\        \        ,          /# )r   r  get_type_fnNr   )r   r   r   r
  list)r   s   "r   r   r   i  sC     L LC=L3%+&-L 
#YLr   c                   Vf   \         p\        P                  P                  4       '       g   . # \	        4       p. pV  F  pV! V4      pWR9  g   K  \        VR4      '       g   K'  VP                  \        P                  P                  Jg   KQ  V'       d_   VP                  V4       \        V4      p\        V4       F   w  rx\        WQ! V4      4      '       g   K  Tp M	  VP                  Wd4       K  V0pV.pK  	  V# )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
  r9   _C_is_torch_function_enabledr5   r
  r	  _disabled_torch_function_implrA  r
  	enumerate
issubclassinsert)	r  r%  overloaded_typesoverloaded_argsargarg_typerA  iold_args	   &&       r   _get_overloaded_argsr4  i  s    J  88..00	"%%!#Os# ,"677++8899:
   $$X. O,"+O"<JA!(K,@AA ! #=  &&u2$,: #&%; < r   c          
          V ^8  d   QhR\         \        \        3,          R\        \        ,          R\        P
                  R\        P                  R\        /# )r   
public_apir  r#   r$   r   )r   r   r   r   r   r#   r$   )r   s   "r   r   r     sV     V VR VC=V 77V ii	V
 Vr   c           	        \        V4      p\        \        \        V4      4      p\	        4       '       d:   \        4       ;_uu_ 4       pVP                  WW#4      pRRR4       X\        Jd   V# V F  pVP                  p	\        V	R4      '       dL   V	P                  VJ d<   V	\        P                  P                  Jd   \        P                  ! R\        ^R7       V	! WW#4      pV\        Jg   K  Vu # 	  V P                    RV P"                   2p
RT
 RV Uu. uF  p\        V4      NK  	  up 2p\	        4       '       d   VR\%        4        2,          p\'        V4      h  + '       g   i     EL; iu upi )	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.
stacklevel.zno implementation found for 'z.' on types that implement __torch_function__: z nor in mode )r4  tuplemapr
  r   _pop_mode_temporarilyr	  NotImplementedr
  r8  r9   r(  r*  r+   warnDeprecationWarning
__module__r
  _get_current_function_mode	TypeError)r6  r  r#   r$   r/  typesr  resultoverloaded_argtorch_func_method	func_namer0  r  s   &&*,         r   r   r     se   T +=9O#dO,-E '(( #$$,,ZMF %'M * +==%z22!**n<!)O)OOMMQ"	 #:dC'M+ *. (():+>+>*?@I
'	{ 35DE_cS	_EF	H  '((9;<==
C.I %$$@  Fs   E"E6
"E3	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                    V ^8  d   QhR\         \        \        \        \        ,          3,          \        \        \
        3,          3,          /# r3   )r<  rK  r   r&  r   r   )r   s   "r   r   r   C  s:     Q$ Q$Ed8n	tHcM22% Q$r   c                  T   \         P                  ! \        4      p / pR \        \        P                  3R\        P
                  \        P
                  P                  3R\        P                  P
                  \        \        P                  P
                  4      3R\        P                  P                  \        \        P                  P                  4      3R\        P                  \        \        P                  4      3R\        P                  \        \        P                  4      3R\        P                  \        \        P                  4      3R\        P                  \        \        P                  4      3.pV EF  w  r4pV EF  pRpV\        P                  Jdx   VP                  R	4      '       d   K3  VP                  R
4      '       d   RpMpVP                  R
4      '       d   RpMVV^ ,          P                  4       '       g   RpM6VR8X  d   K  M,\!        WF4      p\!        \"        VR4      V8X  d   K  VR8X  d   K  \!        WF4      pV\        P                  J d   \!        \"        VR4      V8X  d   K  \%        V\&        P(                  4      '       d   EK  \%        V\*        P,                  4      '       d   EK:  \/        V4      '       g   \1        VR4      '       d   V RV R2WP2                  &   V RV R2WP4                  &   V'       d   EK  VP2                  \7        4       9   dC   Rp	VP2                  \9        4       9   d%   \;        V	P=                  WHP>                  4      4      hEK  W,          PA                  VP2                  4       EK  \/        V4      '       g   EK#  V RV 2W&   V'       d   EK7  V\7        4       9   d9   Rp	V\9        4       9   d%   \;        V	P=                  WHP>                  4      4      hEK~  W,          PA                  V4       EK  	  EK  	  W3# )r9   ztorch.functionalztorch.nn.functionalztorch.nn.initztorch.Tensorztorch.linalgz	torch.fftztorch.specialFrq  rp  T
unique_dimN__weakref__r2  r;  z.__get__z.__set__zk{}.{} is in the tuple returned by torch._overrides.get_ignored_functions but still has an explicit override)!collectionsdefaultdictr&  r9   __all__r   r   dirr   r:   r  r   r	  r
  endswithislowerr
  object
isinstancerE  
ModuleType
__future___Featurer  r
  r2  __set__rC  r  AssertionErrorr   r
  r  )
overridable_funcsrA  tested_namespacesnamespace_str	namespacens_funcsrI  r(   r   r  s
             r   _get_overridable_functionsr`  B  s/    $//5E	%'	U--u/?/?/G/GH	 3 3S9L9L5MN	%((--UXX]]);<	s5<<'89	s5<<'89	eiiUYY0	%--U]]);<	 /@*(!IF,''--))#..!F'',,!F"1--//!F,. / y469d3t;-90DELL(WVY-MQU-U$ 0 011$
 3 344D>>gdI&>&>)6q8&Lll#)6q8&Lll#<<#8#::=  ||'<'>>,SZZ	==-QRR%+224<<@D>>*O1YK8EK ,..9  022(I}})MNN(//5A " /@D ##r   c                \    V ^8  d   QhR\         \        \        \        ,          3,          /# r3   )rK  r   r&  r   )r   s   "r   r   r     s!     	+ 	+4T(^(;#< 	+r   c                 $    \        4       ^ ,          # )zList 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`  rQ  r   r   get_overridable_functionsrc    s     &'**r   c                    \        V \        P                  P                  \        P                  P                  34      '       d   \        V 4      # \        4       ^,          P                  V 4      # )zGet 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.
)rU  r9   _ops
OpOverloadOpOverloadPacketr   r`  get)fs   &r   resolve_namerj    sL      !ejj++UZZ-H-HIJJ1v%'*..q11r   c                :    V ^8  d   QhR\         \        ,          /# r3   r4   )r   s   "r   r   r     s      S] r   c                 Z    \        4       p \        V \        P                  ,          4      pV# )z<Returns a set of the overridable methods on ``torch.Tensor``)rc  r5   r9   r:   )r[  methodss     r   _get_tensor_methodsrn    s&     23#ELL12GNr   c                0    V ^8  d   QhR\         R\        /# r  )r   r
  )r   s   "r   r   r     s     G Gx GD Gr   c                J    V \        4       9   ;'       g    V P                  R8H  # )a7  
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
r2  )rn  r
  )r   s   &r   is_tensor_method_or_propertyrq    s$    . &((FFDMMY,FFr   c                `    \        V 4      \        P                  J ;'       g    \        V R4      # )a  
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
  r9   r:   r
  )inps   &r   is_tensor_likert    s(    D 9$JJ5I(JJr   c                   h   a  ] tR tRt o RtV 3R lR ltRR ltR tR t]	R	 4       t
V 3R
 ltRtV tR# )TorchFunctionModei   a  
A ``TorchFunctionMode`` allows you to override the meaning of all
``__torch_function__`` overridable functions within a dynamic scope,
without having to actually create a tensor subclass or manually
monkey-patch functions in the PyTorch API.  Some common situations
where you should use a mode:

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

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

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

Independent subclasses of :class:`TorchFunctionMode` are compositional:
modes can be pushed onto a stack using ``with MyMode():``.
When you call functions in the PyTorch API inside your
``__torch_function__`` implementation, by default, they will forward on to
the next mode on the mode stack.  If you want recursively call back into
your current ``__torch_function__`` implementation, either explicitly
invoke ``self.__torch_function__(...)``, or use the context manager
``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch
API self-referential (beware of infinite loops, in this case!)
c                   < V ^8  d   QhRR/# )r   r   NrQ  )r   __classdict__s   "r   r   TorchFunctionMode.__annotate__"  s      $ r   c                    R # r  rQ  rH  s   &r   r  TorchFunctionMode.__init__"  s    r   Nc                    \         hr  )NotImplementedErrorr  r   rE  r#   r$   s   &&&&&r   r	  $TorchFunctionMode.__torch_function__%  s    !!r   c                    \        V 4       V # r  )
_push_moderH  s   &r   	__enter__TorchFunctionMode.__enter__(  s    4r   c                    \        4        R # r  )	_pop_mode)r  exc_typeexc_valexc_tbs   &&&&r   __exit__TorchFunctionMode.__exit__,  s    r   c                F    \         P                  ! R ^R7       V ! V/ VB pV# )zP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`r9  )r+   r@  )clsr#   r$   instances   &*, r   pushTorchFunctionMode.push/  s*    ^	
 ''r   c                $   < V ^8  d   Qh/ R;R&   # )r   rv  r  rQ  )r   rx  s   "r   r   ry     s     > ? r   rQ  rQ  N)r
  rB  __qualname____firstlineno____doc__r  r	  r  r  classmethodr  __annotate_func____static_attributes____classdictcell__rx  s   @r   rv  rv     s@     B "  a  r   rv  c                  L    \        4       p V ^ 8  d   \        V ^,
          4      # R# r  )r   r   )	stack_lens    r   rC  rC  9  s%    )+I4=M!)a-0KtKr   c                  h    \        4       p \        V 4       Uu. uF  p\        V4      NK  	  up# u upi r  )r   r   r   )r  r2  s     r    _get_current_function_mode_stackr  >  s/    )+I/4Y/?@/?!"1%/?@@@s   /c                     \        V 4       R # r  )r   )r  s   &r   r  r  C  s
    !$'r   c                      \        4       p V # r  )r   olds    r   r  r  G  s    
#
%CJr   c               #   b   "   \        4       p  V x  \        V 4       R #   \        T 4       i ; i5ir  )r  r  r  s    r   r>  r>  L  s$     
+C	3
3s   / /,/c                   *   a  ] tR tRt o RR ltRtV tR# )BaseTorchFunctionModeiU  Nc                    Vf   / pV! V/ VB # r  rQ  r~  s   &&&&&r   r	  (BaseTorchFunctionMode.__torch_function__V  s    >FT$V$$r   rQ  r  )r
  rB  r  r  r	  r  r  r  s   @r   r  r  U  s     % %r   r  c               #   \  "   \         P                  P                  4       p  \         P                  P                  \         P                  P                  P
                  4       R x  \         P                  P                  V 4       R #   \         P                  P                  T 4       i ; i5ir  )r9   r(  _get_torch_function_state_set_torch_function_state_TorchFunctionStateENABLED)	old_states    r   _enable_torch_functionr  \  se     224I6**588+G+G+O+OP**95**95s   B,AB '!B,!B))B,c               #      "   \         P                  P                  4       ;_uu_ 4         R x   R R R 4       R #   i ; i  + '       g   i     R # ; i5ir  )r9   r(  _RestorePythonTLSSnapshotrQ  r   r   enable_reentrant_dispatchr  f  s?      
	+	+	-	-		 
.	- 	 
.	-	-s%   'A?:
A<?A	
	Ac                B    \         P                  P                  WW#4      # )a	  Skip one level of ``__torch_function__`` dispatch and call the function.

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

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

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

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

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

    >>> from torch.overrides import TorchFunctionMode
    >>> class LoggingMode(TorchFunctionMode):
    ...     def __init__(self):
    ...         self.depth = 0
    ...
    ...     def __torch_function__(self, func, types, args, kwargs=None):
    ...         print(f"{'  ' * self.depth}Calling {func.__name__}")
    ...         self.depth += 1
    ...         with self:
    ...             r = torch.overrides.redispatch_function(
    ...                 func, types, args, kwargs
    ...             )
    ...         self.depth -= 1
    ...         return r
    >>> a = torch.tensor([3.0])
    >>> b = torch.tensor([4.0])
    >>> with LoggingMode():
    ...     result = scaled_mul(a, b)
    Calling scaled_mul
      Calling mul
      Calling add
    >>> result
    tensor([13.])
)r9   r(  _skip_one_hop_torch_function)r   rE  r#   r$   s   &&&&r   redispatch_functionr  v  s    F 8800dKKr   )rC  rc  r  r   r   rj  rt  rq  r#  r  r  )z.*is deprecated, please use.*r9   r  )<r  rW  rN  
contextlibr!  r?  rE  r+   collections.abcr   r   r   typingr   r   r   typing_extensionsr	   r9   torch._Cr
   r   r   r   r   r   r   r   r   rP  r   r   r1   cacherC  rI  r  r#  r4  r   r   r   r   r`  rc  rj  rn  rq  rt  rv  rC  r  r  r  contextmanagerr>  r  r  r  r  rQ  r   r   <module>r     s  ,     
   .  % % ' 
 
 
 t_T] F _  _D	  2 U  Up$'TL^Vr ! . '	  * 	  Q$ Q$h 	+ 	+ 2 2(   G G2"KJ6 6rL
A
(
  %- % 6 6  CLr   