
    3jZ                    N   S r SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJ	r	J
r
  SSKJr  SSKJrJrJr  SSKJr  SSKrSSKJrJrJrJrJrJrJrJrJr  / SQr\" S	5      r\" S
5      r  S0S\	\\4   S\ S\ S\	\\4   4S jjr!\RD                  \!S\#\	   4S j5       5       r$\RD                  S\#\	   4S j5       r%\RD                  \!S\&\	\	4   4S j5       5       r'S\	\\
\   4   S\	\	\\4   /\	\\4   4   4S jr( S1S\
\   S\	\/\)4   S-  S\*\   4S jjr+S\	\\4   S\
\   S\RX                  S\RZ                  S\4
S jr.\" \S5      r/\" \S5      r0\" \S5      r1\RD                  S\2\&\\*\	   4   \&\	\ 4   4   4S j5       r3\!S\&\\*\	   4   4S  j5       r4\!S! 5       r5\RD                  S\#\	   4S" j5       r6\!S\	S\74S# j5       r8S$ r9 " S% S&5      r:S' r;S( r<S) r=S* r>\R~                  S+ 5       r@ " S, S-\:5      rA\R~                  S. 5       rB\R~                  S/ 5       rCg)2aE  
Python implementation of ``__torch_function__``

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

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

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

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

Returns
-------
function
    The wrapped function.
argskwargsr"   c                     > [         R                  " 5          [         R                  " S[        TTS9  T" U 0 UD6sS S S 5        $ ! , (       d  f       g = f)Nignore)categorymessager!   )warningscatch_warningsfilterwarningsUserWarning)r$   r%   r   r!   r    s     I/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/overrides.pywrapper'_disable_user_warnings.<locals>.wrapper[   sA    $$&##;f ((	 '&&s   #A
A)r   r   r$   r%   r   )r   r    r!   r/   s   ``` r.   _disable_user_warningsr1   C   sD    0 4[)rww )")) ) ) ) ) N    c                  %   [         R                  n 1 [         R                  i[         R                  i[         R                  i[         R
                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                   i[         R"                  i[         R$                  i[         R&                  i[         R(                  i[         R*                  i[         R,                  i[         R.                  i[         R0                  i[         R2                  i[         R4                  i[         R6                  i[         R8                  i[         R:                  i[         R<                  i[         R>                  i[         R@                  i[         RB                  i[         RD                  i[         RF                  i[         RH                  i[         RJ                  i[         RL                  i[         RN                  i[         RP                  i[         RR                  i[         RT                  i[         RV                  i[         RX                  i[         RZ                  i[         R\                  i[         R^                  i[         R`                  i[         Rb                  i[         Rd                  i[         Rf                  i[         Rh                  i[         Rj                  i[         Rl                  i[         Rn                  i[         Rp                  i[         Rr                  i[         Rt                  i[         Rv                  i[         Rx                  i[         Rz                  i[         R|                  i[         R~                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  R                  i[         R                  R                  i[         R                  R                  i[         R                  R                  i[         R                  i[         R                  R                  i[         R                  R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  i[         R                  R                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR
                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                  i[         GR                   GR                  GR                   i[         GR                   GR                  GR"                  i[         GR                   GR$                  GR&                  i[         GR                   GR$                  GR(                  i[         GR                   GR$                  R                  i[         GR                   GR$                  GR*                  i[         GR                   GR$                  R                  i[         GR                   GR$                  GR,                  i[         GR                   GR$                  GR.                  i[         GR                   GR$                  GR0                  i[         GR                   GR$                  GR2                  i[         GR                   GR$                  GR4                  i[         GR                   GR$                  GR6                  i[         GR                   GR$                  GR8                  i[         GR:                  GR<                  iG[        iG[        i[         GR>                  i[         GR@                  i[         GRB                  i[         GRD                  i[         GRF                  i[         GRH                  i[         GRJ                  i[         GRL                  i[         GRN                  i[         GRP                  i[         GRR                  i[         GRT                  i[         GRV                  i[         GRX                  i[         GRZ                  i[         GR\                  i[         GR^                  i[         GR`                  i[         GRb                  i[         GRd                  i[         GRf                  i[         GRh                  i[         GRj                  i[         GR                   GR                  GRl                  i[         GRn                  i[         GRp                  i[         GRr                  i[         GRt                  i[         GRv                  i[         GRx                  i[         GRz                  i[         GR|                  i[         GR~                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  i[         GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                  GR                  iU GR                  GR                  iU GR                  iU GR                  iU GR                  iU GR                  iU GR                   iU GR                  iU GR                  iU GR                  iU GR<                  iU GR                  iU GR
                  iU GR                  inG[        GR                  S:  a  UGR                  U GR                  5        U$ )a  
Return public functions that cannot be overridden by ``__torch_function__``.

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

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

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

H 	

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

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

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

oHp 	

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

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

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

IHJ 	

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

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

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

Not ALL property accessors have this property; for example ``Tensor.T`` actually
just creates a new transposed tensor on the fly, and so we SHOULD interpose on
these calls (you need to check the implementation of the function to see if
this is the case or not).  Additionally, if a property accessor doesn't return a Tensor,
it doesn't have to be on this list (though it is harmless if it is).
)r6   r7   _baser.  grad_grad)r7   s    r.   get_default_nowrap_functionsrD    s>    $ \\F r2   c                     [         R                  n 0 [         R                  GSS j_[         R                  GSS j_[         R                  S _[         R
                  S _[         R                  GSS j_[         R                  S _[         R                  GSS j_[         R                  GSS	 j_[         R                  GSS
 j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                   GSS j_[         R"                  GSS j_[         R$                  S _0 [         R&                  GSS j_[         R(                  GSS j_[         R*                  GSS j_[         R,                  GSS j_[         R.                  GSS j_[         R0                  GSS j_[         R2                  GSS j_[         R4                  GSS j_[         R6                  S _[         R8                  S _[         R:                  GSS j_[         R<                  GSS j_[         R>                  S  _[         R@                  GSS! j_[         RB                  GSS" j_[         RD                  GSS# j_[         RF                  GSS$ j_E0 [         RH                  GSS% j_[         RJ                  GSS& j_[         RL                  GSS' j_[         RN                  GSS( j_[         RP                  GSS) j_[         RR                  S* _[         RT                  S+ _[         RV                  S, _[         RX                  GSS. j_[         RZ                  GSS/ j_[         R\                  S0 _[         R^                  S1 _[         R`                  S2 _[         Rb                  S3 _[         Rd                  S4 _[         Rf                  S5 _[         Rh                  S6 _E0 [         Rj                  S7 _[         Rl                  GSS8 j_[         Rn                  S9 _[         Rp                  GSS; j_[         Rr                  GSS< j_[         Rt                  GSS= j_[         Rv                  GSS> j_[         Rx                  GSS? j_[         Rz                  GSS@ j_[         R|                  GSSA j_[         R~                  GSSB j_[         R                  GSSC j_[         R                  SD _[         R                  GSSE j_[         R                  SF _[         R                  SG _[         R                  GSSH j_E0 [         R                  SI _[         R                  GSSJ j_[         R                  GSSK j_[         R                  GSSL j_[         R                  GSSM j_[         R                  GSSN j_[         R                  GSSP j_[         R                  SSQ.SR j_[         R                  SS _[         R                  GSST j_[         R                  R                  GSSU j_[         R                  R                  GSSV j_[         R                  GSSW j_[         R                  GSSX j_[         R                  SY _[         R                  GSSZ j_[         R                  GSS[ j_E0 [         R                  GSS\ j_[         R                  GSS] j_[         R                  GSS^ j_[         R                  GSS_ j_[         R                  GSS` j_[         R                  Sa _[         R                  GSSb j_[         R                  Sc _[         R                  GSSd j_[         R                  Se _[         R                  R                  GSSf j_[         R                  GSSg j_[         R                  GSSh j_[         R                  GSSi j_[         R                  GSSj j_[         R                  GSSk j_[         R                  GSSl j_E0 [         R                  GSSm j_[         R                  GSSn j_[         R                  So _[         R                  GSSp j_[         R                  GSSq j_[         R                  GSSr j_[         R                  GSSs j_[         R                  St _[         R                  GSSu j_[         R                  GSSv j_[         R                  GSSw j_[         R                  GS Sx j_[         R                  Sy _[         R                  GSSz j_[         R                  R                  GSS| j_[         R                  GSS} j_[         R                  GSS~ j_E0 [         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  GSS j_[         R                  S _[         R                  S _[         R                  R                  S _[         GR                   S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR
                  GSS j_[         R                  GR
                  GSS j_[         GR                  GSS j_E0 [         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  S _[         GR                   S _[         GR"                  GSS j_[         R                  GR$                  GSS j_[         R                  GR&                  GSS j_[         R                  GR(                  GSS j_[         R                  GR*                  GSS j_[         GR,                  S _[         GR.                  GS	S j_E0 [         GR0                  GS
S j_[         GR2                  GSS j_[         GR4                  GSS j_[         GR6                  S _[         GR8                  GSS j_[         GR:                  GSS j_[         GR<                  GSS j_[         GR>                  GSS j_[         GR@                  GSS j_[         GRB                  GSS j_[         GRD                  S _[         GRF                  S _[         GRH                  GSS j_[         GRJ                  S _[         GRL                  S _[         GRN                  S _[         GRP                  S _E0 [         GRR                  S _[         GRT                  S _[         GRV                  S _[         GRX                  S _[         GRZ                  S _[         GR\                  GR^                  GSS j_[         GR\                  GR`                  GSS j_[         GR\                  GRb                  GSS j_[         GR\                  GRd                  GSS j_[         GR\                  GRf                  GSS j_[         GR\                  GRh                  GSS j_[         GR\                  GRj                  GSS j_[         GR\                  GRl                  GSS j_[         GR\                  GRn                  GSS j_[         GR\                  GRp                  GSS j_[         GR\                  GRr                  GSS j_[         GR\                  GRt                  GSS j_E0 [         GR\                  GRv                  GSS j_[         GR\                  GRx                  GSS j_[         GR\                  GRz                  GSS j_[         GR\                  GR|                  GSS j_[         GR\                  GR~                  GSS j_[         GR\                  GR                  GSS j_[         GR\                  GR                  GSS j_[         GR\                  GR\                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  S _[         GR                  S _[         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  S _[         GR                  GSS j_E0 [         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  SS[         GR                  SS4S j_[         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_E0 [         GR                  S _[         GR                  S _[         GR                  S _[         GR                  GSS j_[         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         R                  GR                  S _[         GR                  GSS j_E0 [         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  GSS j_[         GR                  S _[         GR                  S _[         GR                  GSS j_[         GR                  GSS j_[         GR                  GS  _[         GR                  GSGS j_[         GR                  GS _[         GR                  GSGS j_[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_E0 [         GR                  GSGS j_[         GR                   GS _[         GR                  GS	 _[         GR                  GSGS
 j_[         R                  GR                  GSGS j_[         R                  GR                  GSGS j_[         GR
                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         GR                  GS _E0 [         GR                   GSGS j_[         GR"                  GSGS j_[         GR$                  GS _[         GR&                  GSGS j_[         R                  GR(                  GSGS j_[         R                  GR*                  GSGS j_[         R                  GR,                  GSGS j_[         GR.                  GSGS j_[         GR0                  GSGS  j_[         GR2                  GSGS! j_[         GR4                  GSGS" j_[         GR6                  GSGS# j_[         GR8                  GSGS$ j_[         GR:                  GSGS% j_[         GR<                  GSGS& j_[         GR>                  GSGS' j_[         GR@                  GSGS( j_E0 [         GRB                  GSGS) j_[         GRD                  GSGS* j_[         GRF                  GSGS+ j_[         GRH                  GSGS, j_[         GRJ                  GSGS- j_[         GRL                  GS. _[         GRN                  GSGS/ j_[         GRP                  GSGS0 j_[         GRR                  GSGS1 j_[         GRT                  GSGS2 j_[         GRV                  GSGS3 j_[         GRX                  GSGS4 j_[         GRZ                  GSGS5 j_[         GR\                  GS6 _[         GR^                  GSGS7 j_[         GR`                  GSGS8 j_[         GRb                  GSGS9 j_E0 [         GRd                  GSGS: j_[         GRf                  GSGS; j_[         GRh                  GSGS< j_[         GRj                  GS= _[         GRl                  GS> _[         GRn                  GSGS? j_[         GRp                  GSGS@ j_[         R                  GRd                  GSGSA j_[         R                  GRr                  GSGSB j_[         R                  GRt                  GSGSC j_[         R                  GRf                  GSGSD j_[         R                  GRp                  GSGSE j_[         GRv                  GSF _[         R                  GRv                  GSGSG j_[         R                  GRx                  GSGSH j_[         R                  GRz                  GSGSI j_[         GR|                  GSJ _E0 [         R                  GR|                  GSK _[         GR~                  GSGSL j_[         GR                  GSGSM j_[         GR                  GSGSN j_[         GR                  GS GSO j_[         GR                  GS GSP j_[         GR                  GS GSQ j_[         GR                  GS!GSR j_[         GR                  GSGSS j_[         GR                  GS"GST j_[         GR                  GSGSU j_[         GR                  GSGSV j_[         GR                  GSW _[         GR                  GSGSX j_[         GR                  GSGSY j_[         GR                  GSGSZ j_[         GR                  GS[ _E0 [         GR                  GS\ _[         GR                  GS] _[         GR                  GS^ _[         GR                  GS_ _[         GR                  GS` _[         GR                  GSa _[         GR                  GSGSb j_[         GR                  GS#GSc j_[         GR                  GSd _[         GR                  GSe _[         GR                  GSGSf j_[         GR                  GSGSg j_[         GR                  GSGSh j_[         GR                  GSGSi j_[         GR                  GSGSj j_[         GR                  GSk _[         GR                  GSl _E0 [         GR                  GS$GSm j_[         GR                  GSn _[         GR                  GSo _[         GR                  GSp _[         GR                  GS%GSq j_[         GR                  GS&GSr j_[         GR                  GSs _[         GR                  GS'GSt j_[         GR                  GSu _[         GR                  GSGSv j_[         GR                  GSGSw j_[         GR                  GSGSx j_[         GR                  GSGSy j_[         GR                  GSGSz j_[         GR                  GR                  GR                  GS{ _[         GR                  GR                  GR                  GS| _[         GR                  GR                  R
                  GSGS} j_E0 [         GR                  GR                  GR                  GSGS~ j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  R*                  GS(GS j_[         GR                  GR                  GR                  GS)GS j_[         GR                  GR                  GR                  GS)GS j_[         GR                  GR                  R\                  GS*GS j_[         GR                  GR                  Rn                  GSGS j_[         GR                  GR                  GR                  GS+GS j_[         GR                  GR                  Rp                  GSGS j_[         GR                  GR                  R                  GSGS j_[         GR                  GR                  R                  GSGS j_[         GR                  GR                  GR                  GS,GS j_[         GR                  GR                  R                  GSGS j_E0 [         GR                  GR                  GR                  GS-GS j_[         GR                  GR                  GR                  GS-GS j_[         GR                  GR                  GR                  GS-GS j_[         GR                  GR                  GR                  GS-GS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR.                  GS	GS j_[         GR                  GR                  GR0                  GS.GS j_[         GR                  GR                  GRX                  GS(GS j_[         GR                  GR                  GR                   GS/GS j_[         GR                  GR                  GR                  GS0GS j_[         GR                  GR                  GR                  GS0GS j_[         GR                  GR                  GR                  GS0GS j_[         GR                  GR                  GR                  GS0GS j_[         GR                  GR                  GR
                  GS1GS j_[         GR                  GR                  GR                  GS2GS j_[         GR                  GR                  GR                  GS3GS j_[         GR                  GR                  GR                  GS4GS j_E0 [         GR                  GR                  GR                  GS%GS j_[         GR                  GR                  GR                  GS5GS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GS6GS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                   GS7GS j_[         GR                  GR                  GR                  GS8GS j_[         GR                  GR                  GR"                  GSGS j_[         GR                  GR                  GR                  GS9GS j_[         GR                  GR                  GR.                  GS%GS j_[         GR                  GR                  GR                  GS:GS j_[         GR                  GR                  GR                  GSGS j_[         GR                  GR                  GR                  GS;GS j_[         GR                  GR                  GR@                  GS<GS j_[         GR                  GR                  GR                   GS _[         GR                  GR                  GR"                  GSGS j_[         GR                  GR                  GR$                  GSGS j_E0 [         GR                  GR                  GR&                  GSGS j_[         GR                  GR                  GRh                  GSGS j_[         GR                  GR                  GR                  GS!GS j_[         GR                  GR                  GR                  GS!GS j_[         GR                  GR                  GR                  GS!GS j_[         GR                  GR                  GR(                  GS!GS j_[         GR                  GR                  GR                  GS!GS j_[         GR                  GR                  GR*                  GS!GS j_[         GR                  GR                  GR,                  GS=GS j_[         GR                  GR                  GR.                  GS=GS j_[         GR                  GR                  GR0                  GS=GS j_[         GR                  GR                  GR2                  GS9GS j_[         GR                  GR                  GR4                  GS>GS j_[         GR                  GR                  GR6                  GS?GS j_[         GR                  GR                  GR8                  GS@GS j_[         GR                  GR                  GR:                  GS+GS j_[         GR                  GR                  GR<                  GSAGS j_E0 [         GR                  GR                  GR>                  GSBGS j_[         GR                  GR                  GR@                  GS3GS j_[         GR                  GR                  GRB                  GSCGS j_[         GR                  GR                  GRD                  GSDGS j_[         GR                  GR                  GRF                  GSEGS j_[         GR                  GR                  GRH                  GS _[         GR                  GR                  GRJ                  GSGS j_[         GR                  GR                  GRL                  GSGS j_[         GR                  GR                  GRN                  GSFGS j_[         GR                  GR                  GRP                  GSGGS j_[         GR                  GR                  GRR                  GSGS j_[         GR                  GR                  GRT                  GSGS j_[         GR                  GR                  GRV                  GSGS j_[         GR                  GR                  GRX                  GSHGS j_[         GR                  GR                  GRZ                  GSIGS j_[         GR                  GR                  GR\                  GSJGS j_[         GR                  GR                  GR^                  GS@GS j_E0 [         GR                  GR                  GR`                  GS<GS j_[         GR                  GR                  GRb                  GS<GS j_[         GR                  GR                  GRd                  GSKGS j_[         GR                  GR                  GRf                  GSGS j_[         GR                  GR                  GRh                  GS _[         GR                  GR                  GRj                  GS _[         GR                  GR                  GRl                  GSGS j_[         GR                  GR                  GRn                  GSLGS j_[         GR                  GR                  GRp                  SSOSS:GS.GS j_[         GR                  GR                  GRr                  GS/GS j_[         GR                  GRt                  GRv                  GSMGS j_[         GR                  GRt                  GRx                  GSMGS j_[         GR                  GRt                  GRz                  GS _[         GR                  GRt                  GR|                  GSNGS j_[         GR~                  GSGS j_[         GR                  S{GS.GS j_[         GR                  GS _E0 [         GR                  GSOGS j_[         R                  GR                  GSPGS j_[         R                  GR                  GSQGS j_[         R                  GR                   GSRGS j_[         GR                  GSSGS j_[         GR                  GSOGS j_[         GR                  GS _[         GR                  GS _[         GR                  GSTGS j_[         GRD                  GSDGS j_[         GR                  GS _[         GR                  GSUGS j_[         GR                  GSGS j_[         GR                  GSVGS j_[         R                  GR                  GSWGS j_[         GR                  GS _[         GR                  GS _E0 [         GR                  GSGS j_[         GRF                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GRH                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GS  _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         R                  GR                  GSXGS j_[         GR                  GSYGS j_E0 [         GR                  GSYGS j_[         GR                  GS	 _[         GR                  GS
 _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                    GSZGS j_[         GR                    GS[GS j_[         GR                    GS\GS j_[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         GR                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         R                  GR                  GSGS j_E0 [         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         GRJ                  GSGS j_[         GR                  GSGS j_[         GR                  GSGS  j_[         GR                  GSGS! j_[         GR                  GS" _[         GRN                  GSFGS# j_[         GR                  GS$ _[         GR                  GSGS% j_[         GR                  GS& _[         GR                  GSGS' j_[         GR                  GSGS( j_[         GR                  GS]GS* j_[         GR                  GSGS+ j_[         GR                  GSGS, j_E0 [         GR                  GS- _[         GRP                  GSGGS. j_[         GR                   GSGS/ j_[         GR                  GSGS0 j_[         GR                  GSGS1 j_[         GR                  GS2 _[         GR                  GS3 _[         GR
                  GSGS4 j_[         GR                  GSGS5 j_[         GR                  GS^GS6 j_[         GR                  GS7 _[         GR                  GS8 _[         GR                  GS_GS9 j_[         GR                  GS_GS: j_[         GRR                  GSGS; j_[         GR                  GSGS< j_[         GR                  GSGS= j_E0 [         GR                  GSGS> j_[         GR                  GSGS? j_[         GR                   GSGS@ j_[         GR"                  GSGSA j_[         GR$                  GSGSB j_[         GR&                  GSC _[         R                  GR&                  GSD _[         GR(                  GSGSE j_[         GR*                  GSGSF j_[         GR`                  GSGSG j_[         R                  GR,                  GSGSH j_[         R                  GR.                  GSGSI j_[         GR0                  GS`SSGSJ.GSK jj_[         GR2                  GSGSL j_[         GR4                  GSGSM j_[         GR6                  GSGSN j_[         GR8                  GSGSO j_E0 [         GR:                  GSGSP j_[         GR<                  GSGSQ j_[         GR>                  GSGSR j_[         GR@                  GSGSS j_[         GRB                  GSGST j_[         GRD                  GSaGSU j_[         GRF                  GSGSV j_[         GRH                  GSGSW j_[         GRJ                  GSGSX j_[         GRL                  GSY _[         GRN                  GSZ _[         GRP                  GS[ _[         GRR                  GS\ _[         GRT                  GS] _[         GRV                  GS^ _[         GRX                  GS_ _[         GRZ                  GS` _E0 [         GR\                  GSa _[         GR^                  GSb _[         GR`                  GSc _[         GRb                  GSd _[         GRd                  GSe _[         GRf                  GSf _[         GRh                  GSg _[         GRj                  GSh _[         GRl                  GSi _[         GRn                  GSGSj j_[         GRp                  GSbGSk j_[         GRr                  GScGSl j_[         R                  GRp                  GSGSm j_[         R                  GRt                  GSGSn j_[         GRv                  GSo _[         GRx                  GSp _[         GRz                  GR|                  GSq _E0 [         GRz                  GR~                  GSr _[         GRz                  GR                  GSs _[         GRz                  GR                  GSt _[         GRz                  GR                  GSu _[         GRz                  GR                  GSGSv j_[         GRz                  GR                  GSGSw j_[         GRz                  GR                  GSGSx j_[         GRz                  GR                  GSGSy j_[         GRz                  GR                  GSz _[         GRz                  GR                  GS{ _[         GRz                  GR8                  GS| _[         GRz                  GR:                  GS} _[         GRz                  GR                  GS~ _[         GRz                  GR<                  GS _[         GRz                  GR@                  GS _[         GRz                  GR                  GS _[         GRz                  GRB                  GS _E0 [         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GR                  GS _[         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GRD                  GS _[         GRz                  GR                  GS _[         GRz                  GR@                  GSGS j_[         GRz                  GRX                  GS _[         GRz                  GRZ                  GSGS j_[         GRz                  GR                  GS _E0 [         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GSGS j_[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GS _[         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GR                  GSGS j_[         GRz                  GR"                  GS _[         GRz                  GR`                  GSGS j_E0 [         GRz                  GR                  GS _[         GRz                  GR                  GSGS j_[         GRz                  GRN                  GSGS j_[         GRz                  GR                  GSGS j_[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GSGS j_[         R                  GR                  GSGS j_[         R                  GR                  GSGS j_[         GR                  GSdGS j_[         GR                  GSGS j_[         GRl                  GSGS j_[         GR                  GS _[         GR                  GS#GS j_[         GR                  GS _E0 [         GR                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GSeGS j_[         R                  GR                  GSTGS j_[         GR                  GSGS j_[         GRn                  GSLGS j_[         GR                  GSGS j_[         GR                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GS _[         GR                  GSfGS j_[         GR                  GSGS j_[         GR                  GS _[         GR                   GSGS j_[         GR                  GSGS j_E0 [         GR                  GSGS j_[         GR                  GSGS j_[         R                  GR                  GSGS j_[         GR
                  GSGS j_[         GR                  GSGS j_[         GR                  GS _[         GR                  GSGS j_[         GR                  GSGS j_[         GR                  GS _[         GR                  GS _[         GR                  GSGS j_[         GR                  GS _[         GR                  GS _[         GR                  GS _[         GR                   GS _[         GR"                  GS _[         GR$                  GS _E0 [         GR&                  GSGS j_[         GR(                  GS _[         GR*                  GSGS j_[         GR,                  SGS.GS j_[         GR.                  GS _[         GR0                  GS _[         GR2                  GS _[         GR4                  GS _[         GR6                  GS _[         GR8                  GS_GS j_[         GR:                  GSGS j_[         GR<                  GSGS j_[         GR>                  GS _[         GR@                  GS _[         GRB                  GS _[         GRD                  GS _[         GRF                  GS _E0 [         GRH                  GS _[         GRJ                  GS _[         GRL                  GS _[         GRN                  GS _[         GRP                  GS _[         GRR                  GS _[         GRT                  GS _[         GRV                  GSGS j_[         GRX                  GS _[         GRZ                  GS _[         GR\                  GS _U GR^                  GS _U GR`                  GS _U GRb                  GS _U GRd                  GS _U GRf                  GS _U GRh                  GS _E0 U GRj                  GS _U GRl                  GS _U GRn                  GS _U GRp                  GS _U GRr                  GS _U GRt                  GS  _U GRv                  GS _U GRx                  GS _U GRz                  GS _U GR|                  GS _U GR~                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS	 _U GR                  GS
 _U GR                  GS _E0 U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  SGS.GS j_U GR                  GS _U GR                  GS _U GR                  GR                  GS _U GR                  GR                  GS _E0 U GR                  GR                  GS _U GR                  GR                  GS _U GR                  GR                  GS  _U GR                  GR                  GS! _U GR                  GR                  GS" _U GR                  GR                  GS# _U GR                  GR                  GS$ _U GR                  GR                  GS% _U GR                  GR                  GS& _U GR                  GR                  GS' _U GR                  GR                  GS( _U GR                  GR                  GS) _U GR                  GS* _U GR                  GS+ _U GR                  GS, _U GR                  GR                  GS- _U GR                  GR                  GS. _E0 U GR                  GR                  GS/ _U GR                  GR                  GS0 _U GR                  GR                  GS1 _U GR                  GR                  GS2 _U GR                  GR                  GS3 _U GR                  GR                  GS4 _U GR                  GR                  GS5 _U GR                  GR                  GS6 _U GR                  GR                  GS7 _U GR                  GR                  GS8 _U GR                  GR                  GS9 _U GR                  GR                  GS: _U GR                  GR                  GS; _U GR                  GR                  GS< _U GR                  GR                  GS= _U GR                  GR                  GS> _U GR                  GR                  GS? _E0 U GR                  GR                  GS@ _U GR                  GR                  GSA _U GR                  GR                  GSB _U GR                  GR                  GSC _U GR                  GR                  GSD _U GR                  GR                  GSE _U GR                   GR                  GSF _U GR                  GR                  GSG _U GR                  GR                  GSH _U GR                  GR                  GSI _U GR                  GR                  GSJ _U GR                  GR                  GSK _U GR                  GR                  GSL _U GR
                  GR                  GSM _U GR                  GSGSN j_U GR                  GSO _U GR                  GSP _E0 U GR                  GSQ _U GR                  GSR _U GR                  GSS _U GR                  GST _U GR                  GSU _U GR                  GSV _U GR                  GSW _U GR                   GSX _U GR"                  GSY _U R                  GSZ _U GR$                  GS[ _U GR&                  GS\ _U GR(                  GS] _U GR*                  GS^ _U GR,                  GS_ _U GR.                  GS0GS` j_U GR0                  [         GR2                  4GSa j_E0 U GR4                  [         GR2                  4GSb j_U GR6                  [         GR2                  4GSc j_U GR8                  [         GR2                  4GSd j_U GR:                  GS)SGSe.GSf jj_U GR<                  GSg _U GR>                  GSh _U GR@                  [         GRB                  4GSi j_U GRD                  GSGSj j_U GRF                  [         GR2                  4GSk j_U GRH                  [         GR2                  4GSl j_U GRJ                  [         GR2                  4GSm j_U GRL                  [         GR2                  4GSn j_U GRN                  [         GR2                  4GSo j_U GRP                  GSp _U GRR                  GSq _U GR                  GSGSr j_U GRT                  GSs _E0 U GRV                  GSGSt j_U GRX                  [         GR2                  4GSu j_U GRZ                  [         GR2                  4GSv j_U GR\                  GSw _U GR^                  GSx _U GR`                  GSy _U GRb                  GSSGSe.GSz jj_U GRd                  GS{ _U GRf                  GS| _U GRh                  [         GR2                  4GS} j_U GRj                  [         GR2                  4GS~ j_U GRl                  SGSe.GS j_U GR                  GS _U GRn                  [         GR2                  4GS j_U GRp                  [         GR2                  4GS j_U GRr                  GS _U GRt                  GS _E0 U GRv                  [         GR2                  4GS j_U GRx                  GS _U GRz                  GS _U GR                  GS _U GR|                  GS _U GR~                  GS _U GR                  GS _U GR                  GS _U GR                  GSgSGSe.GS jj_U GR@                  GS _U GR                  [         GR2                  4GS j_U GR                  GS _U GR                  GS _U GR                  GSGS j_U GR                  GSGS j_U GR.                  GS _U GR                  GS _E0 U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GRx                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GSGS j_U GR                  GS _U GR                  GSSGSe.GS jj_U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _E0 U GR                  GSGS j_U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GShGS j_U GR                  GS _U GR                  GS _U GR                  [         GR2                  4GS j_U GR                  GS _U GR                  GS_GS j_U GR                  GS _U GR                  GS _U GR                  GSGS j_U GR                  GS _E0 U GR                  GS _U GR<                  GSGS j_U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GR                  SS[         GR2                  4GS j_U GR                  GSSGS.GS jj_U GR                  GSGS j_U GR                  GS _U GR                  GS _U GR                  GS _U GR                  GS _U GRr                  GS _U GRv                  GS)GS j_EU GR                  GS U GR                  GS U GR                  GS U GR                  GS U GR                  GSiGS jU GR                  GS U GR                  GS [         R                  GR                  GSGS j0En[         GR                  GR                  GR                  nG[        X5      (       a5  GSGS jUG[        X5      '   GS UG[        U GSU 35      GR                  '   0 nG[        5       nUGR                  5        GH  u  pVUGR                  UGR                  GS-   GSUGR                  -   GS-   GSUGR                  -   GS-   GSUGR                  -   GS-   /nUGR                  GR                  GS5      (       aH  UGR                  G[	        GS5      S nUGR                  GSU-   GS-   GSU-   GS-   GSU-   GS-   /5        U H5  n	G[        X	S5      n
G[        U
5      (       d  M#  X;  d  M*  X;  d  M1  XcU
'   M7     GM	     UGR                  U5        [         GR                  GR                  5       (       Ga]  S-SGKGJn  UGR                  0 UGR                  GS0GS j_UGR                  GSGS j_UGR                  GSjGS j_UGR                  GSGS j_UGR                  GSGS j_UGR                   GSGS j_UGR"                  GSGS j_UGR                  GSjGS j_UGR                  GSjGS j_UGR$                  GSGS j_UGR&                  GSGS j_UGR(                  GSGS j_UGR*                  GSGS j_UGR,                  GSkGS j_UGR.                  GSkGS j_UGR0                  GSkGS j_UGR2                  GSkGS j_5        U$ (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                     gN inputouts     r.   <lambda>'get_testing_overrides.<locals>.<lambda>      2r2   c                     grG  rI  rJ  s     r.   rM  rN        r2   c                     grG  rI  rK  output_sizes     r.   rM  rN        br2   c                     grG  rI  )inputsrT  s     r.   rM  rN        rr2   c                     grG  rI  rJ  s     r.   rM  rN        Br2   c                     grG  rI  rK  s    r.   rM  rN        Rr2   c                     grG  rI  rJ  s     r.   rM  rN        br2   c                     grG  rI  rJ  s     r.   rM  rN        Rr2   c                     grG  rI  rJ  s     r.   rM  rN        rr2   c                     grG  rI  rK  otherrL  s      r.   rM  rN        "r2   c                     grG  rI  rK  batch1batch2alphabetarL  s         r.   rM  rN        rr2   c                     grG  rI  rK  tensor1tensor2valuerL  s        r.   rM  rN        "r2   c                     grG  rI  rp  s        r.   rM  rN    rt  r2   c                     grG  rI  rK  mat1mat2rm  rl  rL  s         r.   rM  rN    rt  r2   c                     grG  rI  )rK  matvecrm  rl  rL  s         r.   rM  rN        r2   c                     grG  rI  )rK  vec1vec2rm  rl  rL  s         r.   rM  rN        r2   c                     grG  rI  thetasizealign_cornerss      r.   rM  rN    r}  r2   c                     grG  rI  rK  dims     r.   rM  rN    rO  r2   Fc                     grG  rI  rK  rf  rtolatol	equal_nans        r.   rM  rN        VXr2   c                     grG  rI  rK  ptraininplaces       r.   rM  rN        Br2   c                     grG  rI  r  s     r.   rM  rN    rZ  r2   c                     grG  rI  r  s     r.   rM  rN    rZ  r2   c                     grG  rI  rK  r  keepdimrL  s       r.   rM  rN    r}  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN        Br2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rc  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  )rK  msgs     r.   rM  rN    rQ  r2   c                     grG  rI  rJ  s     r.   rM  rN    r_  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  rJ  s     r.   rM  rN    rc  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN    r_  r2   c                     grG  rI  re  s      r.   rM  rN        Br2   c                     grG  rI  re  s      r.   rM  rN        br2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  rJ  s     r.   rM  rN    rc  r2   c                      grG  rI  tensorss    r.   rM  rN    rO  r2   c                      grG  rI  r  s    r.   rM  rN    rO  r2   c                      grG  rI  r  s    r.   rM  rN    rO  r2   r   c                     grG  rI  )rK  kernel_sizer  padding	ceil_modecount_include_pads         r.   rM  rN        vxr2   c                     grG  rI  ri  s         r.   rM  rN        PRr2   c	                     grG  rI  )	rK  weightbiasrunning_meanrunning_vartrainingmomentumepscudnn_enableds	            r.   rM  rN        y{r2   c                     grG  rI  )grad_outrK  meaninvstdr  sum_dy
sum_dy_xmucount_tensors           r.   rM  rN    r  r2   c                     grG  rI  )r  rK  r  r  r  input_gweight_gbias_gs           r.   rM  rN        sur2   c                     grG  rI  )rK  r  r  r  r  r  s         r.   rM  rN    rn  r2   c                     grG  rI  rK  r  r  r  r  r  r  counts           r.   rM  rN        tvr2   c                     grG  rI  r  s           r.   rM  rN    	      ACr2   c                     grG  rI  rK  r  s     r.   rM  rN        2r2   c                     grG  rI  )rK  r  r  r  s       r.   rM  rN        Z\r2   c                     grG  rI  )rK  	generatorrL  s      r.   rM  rN        r2   c                     grG  rI  input1input2r  r  s       r.   rM  rN        Rr2   r  c                     grG  rI  rK  targetr  size_averagereduce	reduction
pos_weights          r.   rM  rN        rtr2   c                     grG  rI  )rK  weights	minlengths      r.   rM  rN    r  r2   c                     grG  rI  )r  probr  s      r.   rM  rN        Br2   c                     grG  rI  re  s      r.   rM  rN        "r2   c                     grG  rI  rJ  s     r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN        r2   c                     grG  rI  re  s      r.   rM  rN     r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN        "r2   c                      grG  rI  r  s    r.   rM  rN    rO  r2   c                     grG  rI  rK  ry  	out_dtyperL  s       r.   rM  rN    r  r2   c                      grG  rI  r  s    r.   rM  rN    rg  r2   c                     grG  rI  selfr  s     r.   rM  rN    rc  r2   c                     grG  rI  )rK  
boundaries	out_int32rightrL  s        r.   rM  rN        []r2   c                      grG  rI  r  s    r.   rM  rN    rc  r2   c                     grG  rI  r  r  rL  s      r.   rM  rN  	  r  r2   c                     grG  rI  r  s      r.   rM  rN  
      rr2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  )x1x2r  compute_modes       r.   rM  rN        _ar2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2         ?c                     grG  rI  rK  rl  r  s      r.   rM  rN    r  r2   )rL  c                     grG  rI  )rL  matricess     r.   rM  rN        r2   c                     grG  rI  rK  groupss     r.   rM  rN        Rr2   c                     grG  rI  rK  upperrL  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    r  r2   c                     grG  rI  rK  check_errorsrL  s      r.   rM  rN        br2   c                     grG  rI  r!  s      r.   rM  rN        Rr2   c                     grG  rI  )r  r  r"  rL  s       r.   rM  rN        Br2   c                     grG  rI  )rK  numeln_binsratio	bit_widths        r.   rM  rN        WYr2   c                     grG  rI  rK  chunksr  s      r.   rM  rN    rg  r2   c                     grG  rI  rK  minmaxrL  s       r.   rM  rN    r  r2   c                     grG  rI  r6  s       r.   rM  rN        r2   c                     grG  rI  )rK  r7  rL  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r8  rL  s      r.   rM  rN    r  r2   c                     grG  rI  r  rL  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  
correctionfweightsaweightss       r.   rM  rN        Rr2   c                     grG  rI  r\  s    r.   rM  rN        2r2   c                     grG  rI  )rK  rwith_replacements      r.   rM  rN        rr2   c                     grG  rI  )realimags     r.   rM  rN         "r2   c                     grG  rI  re  s      r.   rM  rN  !  r  r2   c                     grG  rI  )absangs     r.   rM  rN  "      br2   c                     grG  rI  )rK  ords     r.   rM  rN  #  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  $  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  %  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  &  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  '  r  r2   c                     grG  rI  )rK  padrs  s      r.   rM  rN  (      2r2   c                     grG  rI  rK  r  r  r  r  dilationr  s          r.   rM  rN  )      bdr2   c                     grG  rI  r]  s          r.   rM  rN  *  r_  r2   c                     grG  rI  r]  s          r.   rM  rN  +  r_  r2   c	                     grG  rI  )	rK  r  r  r  r  r^  
transposedoutput_addingr  s	            r.   rM  rN  ,      uwr2   c                     grG  rI  )rK  r  r  rZ  s       r.   rM  rN  -  r[  r2   c                     grG  rI  rK  r  r  r  r  output_paddingr  r^  s           r.   rM  rN  .  	      Ar2   c                     grG  rI  rh  s           r.   rM  rN  /  rj  r2   c                     grG  rI  rh  s           r.   rM  rN  0  rj  r2   c                     grG  rI  r\  s    r.   rM  rN  1  rR  r2   c                     grG  rI  rJ  s     r.   rM  rN  2  rO  r2   c                     grG  rI  r  r  r  marginr  r  r  s          r.   rM  rN  3  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  4  rZ  r2   c                     grG  rI  )r  r  r  r  s       r.   rM  rN  5  r  r2   c                     grG  rI  r\  s    r.   rM  rN  6  rO  r2   c                     grG  rI  rK  rf  r  rL  s       r.   rM  rN  7  rU  r2   rH  c                     grG  rI  rv  s       r.   rM  rN  8      2r2   c                     grG  rI  	log_probstargetsinput_lengthstarget_lengthsblankr  zero_infinitys          r.   rM  rN  :  r  r2   c                     grG  rI  rK  r  rL  s      r.   rM  rN  <  r  r2   c                     grG  rI  r  s      r.   rM  rN  =  r  r2   c                     grG  rI  rK  r  rL  rb   s       r.   rM  rN  >  r:  r2   c                     grG  rI  r  s       r.   rM  rN  ?  rX  r2   c                     grG  rI  yxr  s      r.   rM  rN  @  rU  r2   c                     grG  rI  r  s      r.   rM  rN  A  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  B  rc  r2   c                     grG  rI  r\  s    r.   rM  rN  C      r2   c                     grG  rI  r\  s    r.   rM  rN  D      r2   c                     grG  rI  r\  s    r.   rM  rN  E  r  r2   c                     grG  rI  r\  s    r.   rM  rN  F  r  r2   c                     grG  rI  rK  diagonalrL  s      r.   rM  rN  G  r  r2   c                     grG  rI  r  s      r.   rM  rN  H  rU  r2   c                     grG  rI  )rK  offsets     r.   rM  rN  I  rQ  r2   c                     grG  rI  )rK  nr  prependappendrL  s         r.   rM  rN  J      TVr2   c                     grG  rI  rK  r  dim1dim2s       r.   rM  rN  K  r:  r2   c                     grG  rI  r  s       r.   rM  rN  L  r  r2   c                     grG  rI  )rK  srcr  r  r  s        r.   rM  rN  M  rC  r2   c                     grG  rI  )r  r  r  r  storage_offsets        r.   rM  rN  N  r1  r2   c                     grG  rI  rJ  s     r.   rM  rN  O  rc  r2   c                     grG  rI  )rK  rf  r  s      r.   rM  rN  P  r_  r2   c                     grG  rI  rK  rf  rounding_moderL  s       r.   rM  rN  Q      br2   c                     grG  rI  r  s       r.   rM  rN  R  r  r2   c                     grG  rI  re  s      r.   rM  rN  S  rg  r2   c                     grG  rI  r  s       r.   rM  rN  T  rU  r2   c                     grG  rI  rK  ry  r   s      r.   rM  rN  U  r  r2   c                     grG  rI  )rx  ry  s     r.   rM  rN  V      rr2   c                     grG  rI  rK  indices_or_sectionss     r.   rM  rN  W  r  r2   c                     grG  rI  r>  s     r.   rM  rN  X  rQ  r2   c                     grG  rI  rJ  s     r.   rM  rN  Y  rg  r2   c                     grG  rI  rJ  s     r.   rM  rN  Z  r  r2   c                     grG  rI  rK  UPLOrL  s      r.   rM  rN  [  r  r2   c                     grG  rI  r  s      r.   rM  rN  \  r  r2   c                     grG  rI  )equationoperandss     r.   rM  rN  ]  rg  r2   c                     grG  rI  rK  r  padding_idxmax_norm	norm_typescale_grad_by_freqr   s          r.   rM  rN  _      z|r2   c
                     grG  rI  )
rK  r  offsetsr  r  r  moder   per_sample_weightsr  s
             r.   rM  rN  b  s	      hjr2   c                     grG  rI  rK  rb   rp   ra   requires_grads        r.   rM  rN  d      cer2   c                     grG  rI  re  s      r.   rM  rN  e      r2   c                     grG  rI  rK  rf  s     r.   rM  rN  f  rM  r2   c                     grG  rI  rJ  s     r.   rM  rN  g  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  h  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  i  r_  r2   c                     grG  rI  rJ  s     r.   rM  rN  j  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  k  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  l  ra  r2   c                     grG  rI  )rK  scale
zero_pointaxis	quant_min	quant_maxs         r.   rM  rN  m      mor2   c                     grG  rI  )rK  r  r  r  r  s        r.   rM  rN  n      fhr2   c                     grG  rI  )r  observer_onfake_quant_onaveraging_construnning_minrunning_maxr  r  r  r  ch_axisper_row_fake_quantsymmetric_quants                r.   rM  rN  p  s	      ACr2   c                     grG  rI  rK  packed_weightr  outputs       r.   rM  rN  r  r  r2   c                     grG  rI  r  s       r.   rM  rN  s      dfr2   c                     grG  rI  rK  r  packedcol_offsetsweight_scaleweight_zero_pointr  s          r.   rM  rN  t      {}r2   c                     grG  rI  r  s          r.   rM  rN  v      ^`r2   c                     grG  rI  r\  s    r.   rM  rN  x  r[  r2   c                     grG  rI  r\  s    r.   rM  rN  y  r  r2   c                     grG  rI  )rK  abs      r.   rM  rN  z  r:  r2   c                     grG  rI  rK  r  r  s      r.   rM  rN  {  r  r2   c                     grG  rI  r  s      r.   rM  rN  |  r  r2   c                     grG  rI  rK  r  r  norms       r.   rM  rN  }  r  r2   c                     grG  rI  r  s       r.   rM  rN  ~  r  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  rK  sr  r  s       r.   rM  rN    r}  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    rx  r2   c                     grG  rI  r  s       r.   rM  rN    rx  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r)  r2   c                     grG  rI  r  s       r.   rM  rN    rI  r2   c                     grG  rI  r  s       r.   rM  rN    r}  r2   c                     grG  rI  r  s       r.   rM  rN    r}  r2   c                     grG  rI  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r:  r2   c                     grG  rI  rJ  s     r.   rM  rN    rO  r2   c                     grG  rI  )rK  	start_dimend_dims      r.   rM  rN    rU  r2   c                     grG  rI  rK  dimss     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    rn  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  rK  exponentrL  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  )rK  
fill_valuerL  rb   rp   ra   r  s          r.   rM  rN    s	      BDr2   c                     grG  rI  )rK  r  	dep_tokens      r.   rM  rN    r  r2   c                     grG  rI  )LU_data	LU_pivotsunpack_dataunpack_pivotss       r.   rM  rN    r  r2   c                     grG  rI  )rK  r  indexrL  sparse_grads        r.   rM  rN    rC  r2   c                     grG  rI  re  s      r.   rM  rN    rg  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  rJ  s     r.   rM  rN    rM  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rK  r  rL  s      r.   rM  rN    r  r2   c                     grG  rI  rA  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  spacingr  
edge_orders       r.   rM  rN    r+  r2   c                     grG  rI  rK  gridinterpolation_modepadding_moder  s        r.   rM  rN        acr2   c                     grG  rI  rG  s        r.   rM  rN    r  r2   c                     grG  rI  rG  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  
num_groupsr  r  r  r  s         r.   rM  rN        kmr2   c	                     grG  rI  	rK  hxparams
has_biases
num_layersdropoutr  bidirectionalbatch_firsts	            r.   rM  rN        qsr2   c                     grG  rI  rK  rS  w_ihw_hhb_ihb_hhs         r.   rM  rN    r+  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rK  lambds     r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  r  rL  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  valuesrL  s      r.   rM  rN    r  r2   c                     grG  rI  rK  r  rq  r  r  r  s         r.   rM  rN        xzr2   c                     grG  rI  )rK  binsr7  r8  rL  s        r.   rM  rN    r)  r2   c                     grG  rI  )rK  rm  r7  r8  r  densityrL  s          r.   rM  rN    rP  r2   c                     grG  rI  )rK  rm  r   r  ro  s        r.   rM  rN    r1  r2   c                     grG  rI  rK  taus     r.   rM  rN    r  r2   c                     grG  rI  )rx  ry  rL  s      r.   rM  rN    rg  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r>  s     r.   rM  rN    rQ  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  rK  r  r7  sources       r.   rM  rN    r[  r2   c                     grG  rI  r|  s       r.   rM  rN    r  r2   c                     grG  rI  )rK  indicesrh  
accumulates       r.   rM  rN    rt  r2   c                     grG  rI  )rK  r  r7  rL  s       r.   rM  rN    r:  r2   c                     grG  rI  )rK  r  r7  rs  s       r.   rM  rN    r[  r2   c                     grG  rI  )rK  r  r7  r}  r  include_inputs         r.   rM  rN    r  r2   c                     grG  rI  r\   s    r.   rM  rN    r  r2   c                     grG  rI  )eteassume_uniqueinverts       r.   rM  rN    r)  r2   c                     grG  rI  r  s    r.   rM  rN    r  r2   c                     grG  rI  r  s    r.   rM  rN    r]  r2   c                     grG  rI  rJ  s     r.   rM  rN    rQ  r2   c                     grG  rI  rJ  s     r.   rM  rN    rQ  r2   c	                     grG  rI  )	rK  r  r  r  r  use_input_statsr  r  r  s	            r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rR  r2   c                     grG  rI  rJ  s     r.   rM  rN    rc  r2   c                     grG  rI  rJ  s     r.   rM  rN    rg  r2   c                     grG  rI  r%  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r]  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rZ  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s        r.   rM  rN        UWr2   c                     grG  rI  r\  s    r.   rM  rN    rE  r2   c
                     grG  rI  )
rK  n_fft
hop_length
win_lengthwindowcenter
normalizedonesidedlengthreturn_complexs
             r.   rM  rN    	      bdr2   c                     grG  rI  rK  r  r  r  r  
log_targets         r.   rM  rN    s    prr2   c                     grG  rI  r  s     r.   rM  rN        r2   c                     grG  rI  )rK  kr  r  rL  s        r.   rM  rN    r+  r2   c                     grG  rI  )rK  	hermitianr&  rL  s       r.   rM  rN    rK  r2   c                     grG  rI  )rK  r  rL  s      r.   rM  rN    rt  r2   c                     grG  rI  )LDpivotsBr  rL  s        r.   rM  rN        QSr2   c                     grG  rI  )rK  normalized_shaper  r  r  r  s         r.   rM  rN    rZ  r2   c                     grG  rI  re  s      r.   rM  rN    rg  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  endr  rL  s       r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    r_  r2   c                     grG  rI  )rK  r  r  Xr  iKnitertollargestmethodtrackerortho_iparamsortho_fparamsortho_bparamss                 r.   rM  rN    s	      IKr2   c                     grG  rI  rJ  s     r.   rM  rN    rO  r2   c                     grG  rI  rK  r  rb   s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  rL  s      r.   rM  rN    rZ  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    ra  r2   c                     grG  rI  )rK  namesr  rL  s       r.   rM  rN    rI  r2   c	                     grG  rI  )	databatch_sizesrS  rT  rU  rV  rW  r  rX  s	            r.   rM  rN     rZ  r2   c                     grG  rI  r\  s         r.   rM  rN    rC  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  )Apivot	get_infosrL  s       r.   rM  rN    rx  r2   c                     grG  rI  )r  r2  r3  rL  s       r.   rM  rN    r:  r2   c                     grG  rI  rp  s          r.   rM  rN    rj  r2   c                     grG  rI  )rK  maskrs  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r}  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  rL  s      r.   rM  rN  	  r[  r2   c                     grG  rI  re  s      r.   rM  rN  
  r  r2   c                     grG  rI  rK  r  rL  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r&  rL  s       r.   rM  rN    r	  r2   c                     grG  rI  )LUr  r  leftadjointrL  s         r.   rM  rN        Y[r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  s     r.   rM  rN    ra  r2   c                     grG  rI  rK  r  rL  s      r.   rM  rN    rU  r2   c                     grG  rI  )rK  r  r  s      r.   rM  rN        2r2   c                     grG  rI  r>  s     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  rJ  s     r.   rM  rN    rO  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  rK  r  r  r  r^  r  s         r.   rM  rN        jlr2   c                     grG  rI  r	  s         r.   rM  rN    r
  r2   c                     grG  rI  r	  s         r.   rM  rN    r
  r2   c                     grG  rI  rK  r  r  r  r^  return_indicesr  s          r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rZ  r2   c                     grG  rI  )rK  r  r  rb   rL  s        r.   rM  rN     r  r2   c                     grG  rI  r  s     r.   rM  rN  !  r_  r2   c                     grG  rI  r  s     r.   rM  rN  "  r  r2   c                      grG  rI  )r  r%   s     r.   rM  rN  #  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  $  rO  r2   c                     grG  rI  re  s      r.   rM  rN  %  r  r2   c                     grG  rI  re  s      r.   rM  rN  &  r  r2   c                     grG  rI  )rK  r  r  r  r  r  exponential_average_factorepsilons           r.   rM  rN  (  r  r2   c	                     grG  rI  	rK  r  r  r  r  r^  r  	benchmarkdeterministics	            r.   rM  rN  *  r  r2   c	                     grG  rI  )	rK  r  zrl  r  r  r  r^  r  s	            r.   rM  rN  +  r  r2   c                     grG  rI  r]  s          r.   rM  rN  ,  r  r2   c
                     grG  rI  )
rK  r  r  r  ri  r  r^  r  r  r  s
             r.   rM  rN  .  re  r2   c	                     grG  rI  r  s	            r.   rM  rN  1      egr2   c                     grG  rI  )rK  r  weight_stride0rS  cxr  hidden_sizerV  rY  rW  r  rX  r  dropout_states                 r.   rM  rN  4  r  r2   c                     grG  rI  r  s       r.   rM  rN  6  r:  r2   c                     grG  rI  r  s       r.   rM  rN  7  rx  r2   c                     grG  rI  rK  r}  destinations      r.   rM  rN  8  r  r2   c                     grG  rI  r-  s      r.   rM  rN  9  r[  r2   c                     grG  rI  )rK  
descendingrL  s      r.   rM  rN  :  rX  r2   c                     grG  rI  re  s      r.   rM  rN  ;  rg  r2   c                     grG  rI  re  s      r.   rM  rN  <  r  r2   c                     grG  rI  )rK  num_samplesreplacementrL  s       r.   rM  rN  =      SUr2   c                     grG  rI  )rK  r|  rL  s      r.   rM  rN  >  rc  r2   c                     grG  rI  rK  r  s     r.   rM  rN  ?  r  r2   c                     grG  rI  )rK  r  startr  s       r.   rM  rN  @  r  r2   c                     grG  rI  )rK  nanposinfneginfrL  s        r.   rM  rN  A  r  r2   c                     grG  rI  )rK  r  r  r  r  r  r  r  s           r.   rM  rN  B  rZ  r2   c                     grG  rI  )rK  r  r  r  r  r  s         r.   rM  rN  C      ]_r2   c                     grG  rI  r  s      r.   rM  rN  D  r  r2   c                     grG  rI  rK  r  r  r  r  s        r.   rM  rN  E  r  r2   c                     grG  rI  rK  r  r  r  s       r.   rM  rN  F  r1  r2   c                     grG  rI  )rK  r  r  NCHxWgroupr  s           r.   rM  rN  G  r  r2   c                     grG  rI  )rK  r  r  r  rb   s        r.   rM  rN  H  r7  r2   c                     grG  rI  r  s     r.   rM  rN  I  r  r2   c                     grG  rI  re  s      r.   rM  rN  J  r  r2   c                     grG  rI  re  s      r.   rM  rN  K  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  L  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  M  rQ  r2   c                     grG  rI  re  s      r.   rM  rN  N  r  r2   c                     grG  rI  rS  s     r.   rM  rN  O  r+  r2   c                     grG  rI  rS  s     r.   rM  rN  P  r+  r2   c                     grG  rI  rK  rT  r  s      r.   rM  rN  Q  r_  r2   c                     grG  rI  rX  s      r.   rM  rN  R      oqr2   c                     grG  rI  rX  s      r.   rM  rN  S  r_  r2   c                     grG  rI  rX  s      r.   rM  rN  T  rZ  r2   c                     grG  rI  rX  s      r.   rM  rN  U  r_  r2   c                     grG  rI  rX  s      r.   rM  rN  V  rZ  r2   c                     grG  rI  r  s      r.   rM  rN  W  r  r2   c                     grG  rI  rK  r  r  r  s       r.   rM  rN  X  r  r2   c                     grG  rI  rK  r  r  r  r  r  divisor_overrides          r.   rM  rN  Z  	      @Br2   c                     grG  rI  rc  s          r.   rM  rN  ]  re  r2   c                     grG  rI  )rK  r  r  r  r  r  r  r  s           r.   rM  rN  `  r  r2   c                     grG  rI  r  s       r.   rM  rN  b  r  r2   c                     grG  rI  rK  r  r  r  r  r  s         r.   rM  rN  d  rK  r2   c                     grG  rI  r  s          r.   rM  rN  g  r  r2   c                     grG  rI  r  s      r.   rM  rN  i  rt  r2   c                     grG  rI  rp  s          r.   rM  rN  k      gir2   c                     grG  rI  )rK  r  r  r  ignore_indexr  r  label_smoothings           r.   rM  rN  n  	      JLr2   c                     grG  rI  rz  s          r.   rM  rN  q  r  r2   c                     grG  rI  ra  s       r.   rM  rN  s      XZr2   c                     grG  rI  ra  s       r.   rM  rN  t  r  r2   c                     grG  rI  ra  s       r.   rM  rN  u  r  r2   c                     grG  rI  ra  s       r.   rM  rN  v  r  r2   c                     grG  rI  r  s      r.   rM  rN  w  r  r2   c                     grG  rI  r  s          r.   rM  rN  y  r  r2   c                     grG  rI  )rK  r  r  r  r  r  r  r   r  include_last_offsetr  s              r.   rM  rN  |  s	      HJr2   c                     grG  rI  ra  s       r.   rM  rN  ~  rn  r2   c                     grG  rI  )rK  rT  r  r^  r  r  s         r.   rM  rN    rP  r2   c                     grG  rI  rK  r  rT  output_ratior  _random_sampless         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    rk  r2   c                     grG  rI  )rK  r  varr   r  r  s         r.   rM  rN    r  r2   c                     grG  rI  )rK  approximates     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  rH  r  rJ  r  s        r.   rM  rN    rk  r2   c                     grG  rI  )rK  rO  r  r  r  s        r.   rM  rN    r$  r2   c                     grG  rI  )logitsrs  hardr  r  s        r.   rM  rN    rK  r2   c                     grG  rI  rd  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  min_valmax_valr  s       r.   rM  rN    r  r2   c                     grG  rI  rj  s         r.   rM  rN        `br2   c                     grG  rI  )rK  r  r  r  r  r  r  r  s           r.   rM  rN    s	      GIr2   c                     grG  rI  )rK  r  scale_factorr  r  recompute_scale_factor	antialiass          r.   rM  rN    s	      KMr2   c                     grG  rI  r  s         r.   rM  rN    rj  r2   c                     grG  rI  rK  r  r  r  r  r  s         r.   rM  rN    r  r2   c                     grG  rI  rF  s        r.   rM  rN    rP  r2   c                     grG  rI  )rK  negative_sloper  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  s      r.   rM  rN    r)  r2   c                     grG  rI  )rK  r  rl  rm  r  s        r.   rM  rN    r$  r2   c                     grG  rI  rK  r  _stacklevelrb   s       r.   rM  rN    s    \^r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  rK  r  r  r  r  s        r.   rM  rN    rP  r2   c                     grG  rI  r  s        r.   rM  rN    rP  r2   c                     grG  rI  r  s        r.   rM  rN    rP  r2   c                     grG  rI  rp  s          r.   rM  rN    rn  r2   c                     grG  rI  rK  r  r  r  r^  r  r  s          r.   rM  rN    r  r2   c                     grG  rI  r  s          r.   rM  rN    r  r2   c                     grG  rI  r  s          r.   rM  rN    r  r2   c                     grG  rI  r  s          r.   rM  rN    r  r2   c                     grG  rI  r  s          r.   rM  rN    r  r2   c                     grG  rI  r  s          r.   rM  rN    r  r2   c                     grG  rI  rK  r  r  r  r  rT  s         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    rk  r2   c                     grG  rI  r  s         r.   rM  rN    r  r2   c                     grG  rI  )querykeyrs  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.   rM  rN    s	      ]_r2   c                     grG  rI  )rK  r  r  rq  r  r  r  r  s           r.   rM  rN    r  r2   c                     grG  rI  rK  r  r  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  rj  s         r.   rM  rN    rK  r2   c                     grG  rI  )rK  r  r  r  rp  r  r  s          r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  r  rL  s        r.   rM  rN    r  r2   c                     grG  rI  )r\   num_classess     r.   rM  rN    r  r2   c                     grG  rI  )rK  rZ  r  rs  s       r.   rM  rN    r'  r2   c                     grG  rI  r  r  r  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  r  	log_inputr   r  r  r  r  s           r.   rM  rN    r  r2   c                     grG  rI  rK  r  s     r.   rM  rN    r  r2   c                     grG  rI  rK  r  s     r.   rM  rN    rX  r2   c                     grG  rI  r  s     r.   rM  rN    r:  r2   c                     grG  rI  rH  s       r.   rM  rN    rC  r2   c                     grG  rI  rK  lowerr"  r  r  s        r.   rM  rN    s    wyr2   c                     grG  rI  r  s     r.   rM  rN    rX  r2   c                     grG  rI  r  s     r.   rM  rN    rX  r2   c                     grG  rI  r  s     r.   rM  rN    rX  r2   c                     grG  rI  )r  r  rs  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  r  r  rm  s         r.   rM  rN    rj  r2   c                     grG  rI  )rK  r  r  deltar  s        r.   rM  rN    s    hjr2   c                     grG  rI  r  s        r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    ru  r2   c                     grG  rI  r  s       r.   rM  rN    ru  r2   c                     grG  rI  )rK  rm  	thresholds      r.   rM  rN    rt  r2   c                     grG  rI  rd  s     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  rK  r  rs  r  s       r.   rM  rN    r  r2   c
                     grG  rI  
anchorpositivenegativerq  r  r  swapr  r  r  s
             r.   rM  rN    rr  r2   )distance_functionrq  r  r  c                    grG  rI  )r  r  r  r  rq  r  r  s          r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r^  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  )r\   r  r  r  s       r.   rM  rN    rC  r2   c                     grG  rI  )r\   r  stdr  s       r.   rM  rN    r  r2   c                     grG  rI  )r\   vals     r.   rM  rN    r  r2   c                     grG  rI  )r\   r  r  nonlinearityr  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  as_tuples     r.   rM  rN    r  r2   )r.  c                    grG  rI  )rK  r  r.  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rR  r2   c                     grG  rI  rK  r  r  r  rL  rb   s         r.   rM  rN    r  r2   c                     grG  rI  rK  rT  r  r  rL  rb   s         r.   rM  rN    r_  r2   c                     grG  rI  r  s         r.   rM  rN    r  r2   c                     grG  rI  r  s         r.   rM  rN    s     13r2   c                     grG  rI  )vpowr  s      r.   rM  rN    r  r2   c                     grG  rI  r  s         r.   rM  rN     r_  r2   c                     grG  rI  r\  s    r.   rM  rN    rE  r2   c                     grG  rI  rr  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  r  input3r  	transposes        r.   rM  rN    rn  r2   c                     grG  rI  r  s        r.   rM  rN    r  r2   c                     grG  rI  r  r  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  qr  r  s       r.   rM  rN    rI  r2   c                     grG  rI  r:  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  rconds     r.   rM  rN    r  r2   c                     grG  rI  )rK  r!  r  s      r.   rM  rN  	  rI  r2   c                     grG  rI  )rK  upscale_factors     r.   rM  rN  
  r[  r2   c                     grG  rI  )rK  downscale_factors     r.   rM  rN    rX  r2   c                     grG  rI  )rK  r  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  r   r  r  s         r.   rM  rN    r1  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    rQ  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  r  s        r.   rM  rN    r_  r2   c                     grG  rI  r(  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  rb   s     r.   rM  rN    r_  r2   c                     grG  rI  )rK  r7  r}  r  s       r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rQ  r2   c                     grG  rI  r\  s    r.   rM  rN    rg  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r]  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  )rK  somerL  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  rL  s      r.   rM  rN    r  r2   c                     grG  rI  rK  r  r  r  interpolationrL  s         r.   rM  rN    r  r2   c                     grG  rI  r9  s         r.   rM  rN    rn  r2   c                     grG  rI  )rK  scaleszero_pointsr  rb   s        r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  rb   s       r.   rM  rN    r  r2   c                     grG  rI  )rK  rb   reduce_ranges      r.   rM  rN     r'  r2   c                     grG  rI  )rK  r  r  r  r  r  output_scaleoutput_zero_points           r.   rM  rN  !  rZ  r2   c                     grG  rI  rK  rS  r]  r^  r_  r`  	packed_ih	packed_hhcol_offsets_ihcol_offsets_hhscale_ihscale_hhzero_point_ihzero_point_hhs                 r.   rM  rN  #  	      _ar2   c                     grG  rI  rF  s                 r.   rM  rN  &  rO  r2   r      c                     grG  rI  r	  s         r.   rM  rN  )  s     "r2   c                     grG  rI  r	  s         r.   rM  rN  .  s     !#r2   c                     grG  rI  r	  s         r.   rM  rN  4  s     !#r2   c                     grG  rI  rF  s                 r.   rM  rN  ;  rO  r2   c                     grG  rI  rF  s                 r.   rM  rN  >  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  @  rc  r2   c                     grG  rI  r\  s    r.   rM  rN  A  rE  r2   c                     grG  rI  rJ  s     r.   rM  rN  B  rZ  r2   c                     grG  rI  re  s      r.   rM  rN  C  r  r2   c                     grG  rI  rv  s       r.   rM  rN  D  r  r2   c                     grG  rI  r\  s    r.   rM  rN  E  rM  r2   c                     grG  rI  r\  s    r.   rM  rN  F  ra  r2   c                     grG  rI  rJ  s     r.   rM  rN  G  rg  r2   c                     grG  rI  r  s     r.   rM  rN  H  r  r2   c                     grG  rI  re  s      r.   rM  rN  I  r  r2   c                     grG  rI  )rK  r  r  maxnormrL  s        r.   rM  rN  J  rX  r2   c                     grG  rI  r  s     r.   rM  rN  K  r  r2   c                     grG  rI  )rK  shapes     r.   rM  rN  L  rZ  r2   c                     grG  rI  rH  s       r.   rM  rN  M  rn  r2   c	                     grG  rI  rR  s	            r.   rM  rN  N  r  r2   c                     grG  rI  r\  s         r.   rM  rN  O  r  r2   c	                     grG  rI  rR  s	            r.   rM  rN  P  r  r2   c                     grG  rI  r\  s         r.   rM  rN  Q  r  r2   c                     grG  rI  )rK  shiftsr!  s      r.   rM  rN  R  r  r2   r   rS  c                     grG  rI  )rK  r  r!  s      r.   rM  rN  S  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  T  ra  r2   c                     grG  rI  r>  s     r.   rM  rN  U  r  r2   c                     grG  rI  )r  r  compressed_indices_dtypes      r.   rM  rN  V  r'  r2   c                     grG  rI  r  s        r.   rM  rN  W  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  X  ra  r2   c                     grG  rI  )rK  rf  rl  s      r.   rM  rN  Y  rg  r2   c                     grG  rI  rw  s         r.   rM  rN  Z  r  r2   c                     grG  rI  rK  r  r7  r  s       r.   rM  rN  [  r  r2   c                     grG  rI  rz  s       r.   rM  rN  \  r  r2   c                     grG  rI  )rK  r  r7  r  r  include_selfs         r.   rM  rN  ]  ru  r2   c                     grG  rI  )sorted_sequencerK  r  r  rL  s        r.   rM  rN  ^  r  r2   c                     grG  rI  )r  r  lengthsr  r  r  unsafes          r.   rM  rN  _  r  r2   c                     grG  rI  )rK  r  r7  s      r.   rM  rN  `  rQ  r2   c                     grG  rI  )rK  r  r  r7  s       r.   rM  rN  a  r  r2   c                     grG  rI  rK  r  r  r<  r  steps         r.   rM  rN  b  r  r2   c                     grG  rI  r  s         r.   rM  rN  c  r  r2   c                     grG  rI  r  s     r.   rM  rN  d  r  r2   c                     grG  rI  rJ  s     r.   rM  rN  e  rc  r2   c                     grG  rI  rJ  s     r.   rM  rN  f  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  g  rc  r2   c                     grG  rI  rJ  s     r.   rM  rN  h  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  i  rO  r2   c                     grG  rI  rJ  s     r.   rM  rN  j  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  k  rZ  r2   c                     grG  rI  r\  s    r.   rM  rN  l  r]  r2   c                     grG  rI  r\  s    r.   rM  rN  m  rZ  r2   c                     grG  rI  r  s      r.   rM  rN  n  r  r2   c                     grG  rI  r  s      r.   rM  rN  o  r  r2   c                     grG  rI  r  s      r.   rM  rN  p  r  r2   c                     grG  rI  )r  r  r  rL  s       r.   rM  rN  q  rU  r2   c                     grG  rI  )r  r  r  r&  rL  s        r.   rM  rN  r  r  r2   )stablerL  c                    grG  rI  )rK  r  r1  r  rL  s        r.   rM  rN  s  r1  r2   c                     grG  rI  r\   split_size_or_sectionsr  s      r.   rM  rN  t  rx  r2   c                     grG  rI  r  s      r.   rM  rN  u  r'  r2   c                     grG  rI  rJ  s     r.   rM  rN  v  rZ  r2   c                     grG  rI  rJ  s     r.   rM  rN  w  r_  r2   c                     grG  rI  r  s      r.   rM  rN  x  r  r2   c                     grG  rI  rw  s         r.   rM  rN  y  rC  r2   c                     grG  rI  r  s      r.   rM  rN  z  r  r2   c                     grG  rI  r  s     r.   rM  rN  {  rO  r2   c                     grG  rI  r  s     r.   rM  rN  |  rQ  r2   c                     grG  rI  )rK  r  r  r  r  r  pad_moder  r  r  align_to_windows              r.   rM  rN  ~  s	      ~@r2   c                     grG  rI  re  s      r.   rM  rN    rg  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r]  r2   c                     grG  rI  r  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r]  r2   c                     grG  rI  )r  r  cs      r.   rM  rN    r  r2   c                     grG  rI  )r$   s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rR  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rR  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rR  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r_  r2   c                     grG  rI  )rK  r6  
compute_uvrL  s       r.   rM  rN    rI  r2   c                     grG  rI  )rK  r  r  Ms       r.   rM  rN    rX  r2   c                     grG  rI  )rK  full_matricesrL  s      r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    r  r2   c                     grG  rI  rK  dim0r  s      r.   rM  rN    rg  r2   c                     grG  rI  )rK  axis0axis1s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    ra  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  r\  s    r.   rM  rN    rc  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    ra  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r\  s    r.   rM  rN    rZ  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  re  s      r.   rM  rN    rX  r2   c                     grG  rI  re  s      r.   rM  rN    r:  r2   c                     grG  rI  r\  s    r.   rM  rN    ra  r2   c                     grG  rI  r  s      r.   rM  rN    rI  r2   c                     grG  rI  r  s      r.   rM  rN    r}  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r}  r2   c                     grG  rI  r  s      r.   rM  rN    r}  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r\  s    r.   rM  rN    r_  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r  s       r.   rM  rN    rC  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r:  s     r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rO  r2   c                     grG  rI  r\  s    r.   rM  rN    rX  r2   c                     grG  rI  r\  s    r.   rM  rN    rX  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rM  r2   c                     grG  rI  r  s      r.   rM  rN    rU  r2   c                     grG  rI  r\  s    r.   rM  rN    r  r2   c                     grG  rI  re  s      r.   rM  rN    rU  r2   c                     grG  rI  re  s      r.   rM  rN    r  r2   c                     grG  rI  )r  rf  rL  s      r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    s    rr2   c                     grG  rI  )rK  r7  s     r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  rL  s       r.   rM  rN    r  r2   c                     grG  rI  rJ  s     r.   rM  rN    rO  r2   c                     grG  rI  rJ  s     r.   rM  rN    rZ  r2   c                     grG  rI  )r  inds     r.   rM  rN    r  r2   c                     grG  rI  )r  r  r!  s      r.   rM  rN    r  r2   c                     grG  rI  )r  r  r!  rL  s       r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s       r.   rM  rN    r}  r2   c                     grG  rI  r   s     r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r  r1  rL  s        r.   rM  rN    r  r2   c                     grG  rI  r\  s    r.   rM  rN    rE  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    rc  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r"  r  unitriangulars        r.   rM  rN    r  r2   c                     grG  rI  )rK  r  r"  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c
                     grG  rI  r  s
             r.   rM  rN    rr  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rQ  r2   c                     grG  rI  rJ  s     r.   rM  rN    ra  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  )rK  r  sizesr  s       r.   rM  rN    r  r2   c                     grG  rI  )rK  sortedreturn_inversereturn_countsr  s        r.   rM  rN    r  r2   c                     grG  rI  )rK  r"  r#  r  s       r.   rM  rN    r$  r2   c                     grG  rI  )r  rg  s     r.   rM  rN    r  r2   c                     grG  rI  r3  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    rt  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s      r.   rM  rN    r  r2   c                     grG  rI  )r  rJ  s     r.   rM  rN    rc  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  r  s     r.   rM  rN    rQ  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r>  s     r.   rM  rN    rQ  r2   c                     grG  rI  )	conditionr  r  s      r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r  s       r.   rM  rN    r  r2   c                     grG  rI  )rK  input_scaleinput_zero_point	prepacked	out_scaleout_zero_pointout_channels          r.   rM  rN    r  r2   c                     grG  rI  r  s        r.   rM  rN    r  r2   c                     grG  rI  )r  levels     r.   rM  rN     r  r2   c                     grG  rI  )primaltangentr;  s      r.   rM  rN    rU  r2   c                     grG  rI  r  s    r.   rM  rN    r_  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rO  r2   c                     grG  rI  )r  r  r  r  s       r.   rM  rN    rn  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r  s       r.   rM  rN    r  r2   )implicitc                    grG  rI  )r  r  rG  s      r.   rM  rN  	  r  r2   c                     grG  rI  )r  r  r<  r  s       r.   rM  rN  
  r  r2   c                     grG  rI  )r  r!  s     r.   rM  rN    rc  r2   c                     grG  rI  r  r  r  s      r.   rM  rN    rU  r2   c                     grG  rI  )r  r  r7  s      r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  r<  r  r  s        r.   rM  rN    r+  r2   c                     grG  rI  )r  
split_sizer  s      r.   rM  rN    r  r2   c                     grG  rI  )r  split_sizesr  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r_  r2   c                     grG  rI  r@  s    r.   rM  rN    rE  r2   c                     grG  rI  )r  r  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rQ  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r_  r2   c                     grG  rI  r@  s    r.   rM  rN    ra  r2   c                     grG  rI  r@  s    r.   rM  rN    r_  r2   c                     grG  rI  r@  s    r.   rM  rN    ra  r2   c                     grG  rI  r  s     r.   rM  rN    rc  r2   c                     grG  rI  r  rb   s     r.   rM  rN    ra  r2   c                     grG  rI  r  	dimensionr  r  s       r.   rM  rN     rX  r2   c                     grG  rI  r@  s    r.   rM  rN  !  r  r2   c                     grG  rI  r  rf  s     r.   rM  rN  "  r  r2   c                     grG  rI  rh  s     r.   rM  rN  #  rg  r2   c                     grG  rI  rh  s     r.   rM  rN  $  rg  r2   c                     grG  rI  rh  s     r.   rM  rN  %  rQ  r2   c                     grG  rI  rh  s     r.   rM  rN  &  r  r2   c                     grG  rI  rh  s     r.   rM  rN  '  r  r2   c                     grG  rI  rh  s     r.   rM  rN  (  rc  r2   c                     grG  rI  rh  s     r.   rM  rN  )  rQ  r2   c                     grG  rI  rh  s     r.   rM  rN  *  rQ  r2   c                     grG  rI  rh  s     r.   rM  rN  +  rc  r2   c                     grG  rI  rh  s     r.   rM  rN  ,  rQ  r2   c                     grG  rI  rh  s     r.   rM  rN  -  rQ  r2   c                     grG  rI  rh  s     r.   rM  rN  .  rZ  r2   c                     grG  rI  rh  s     r.   rM  rN  /  rO  r2   c                     grG  rI  rh  s     r.   rM  rN  0  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  1  r  r2   c                     grG  rI  r@  s    r.   rM  rN  2  r  r2   c                     grG  rI  rb  s     r.   rM  rN  3  r_  r2   c                     grG  rI  r@  s    r.   rM  rN  4  rR  r2   c                     grG  rI  rh  s     r.   rM  rN  5  r  r2   c                     grG  rI  r@  s    r.   rM  rN  6  r]  r2   c                     grG  rI  r@  s    r.   rM  rN  7  r  r2   c                     grG  rI  rh  s     r.   rM  rN  8  rZ  r2   c                     grG  rI  rh  s     r.   rM  rN  9  ra  r2   c                     grG  rI  rh  s     r.   rM  rN  :  ra  r2   c                     grG  rI  )r  arrays     r.   rM  rN  ;  r  r2   c                     grG  rI  )r  idxs     r.   rM  rN  <  r_  r2   c                     grG  rI  )r  memos     r.   rM  rN  =  rQ  r2   c                     grG  rI  r@  s    r.   rM  rN  >  r]  r2   c                     grG  rI  r@  s    r.   rM  rN  ?  rR  r2   c                     grG  rI  r@  s    r.   rM  rN  @  r  r2   c                     grG  rI  r@  s    r.   rM  rN  A  r]  r2   c                     grG  rI  )r  format_specs     r.   rM  rN  B  r  r2   c                     grG  rI  )r  protos     r.   rM  rN  C  rg  r2   c                     grG  rI  r@  s    r.   rM  rN  D  rM  r2   )tensor_contentsc                    grG  rI  )r  r  s     r.   rM  rN  E  rX  r2   c                     grG  rI  )r  r  r  s      r.   rM  rN  F  rc  r2   c                     grG  rI  )r  ds     r.   rM  rN  G  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  H  r  r2   c                     grG  rI  r@  s    r.   rM  rN  I  r  r2   c                     grG  rI  r@  s    r.   rM  rN  J  r  r2   c                     grG  rI  r@  s    r.   rM  rN  K  r  r2   c                     grG  rI  r@  s    r.   rM  rN  L  r  r2   c                     grG  rI  r@  s    r.   rM  rN  M  r  r2   c                     grG  rI  r@  s    r.   rM  rN  N  rO  r2   c                     grG  rI  r@  s    r.   rM  rN  O  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  P  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  Q  rO  r2   c                     grG  rI  r@  s    r.   rM  rN  R  r_  r2   c                     grG  rI  r@  s    r.   rM  rN  S  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  T  rQ  r2   c                     grG  rI  r@  s    r.   rM  rN  U  r_  r2   c                     grG  rI  )r  cuda_enabledcpu_enabled
cuda_dtype	cpu_dtypes        r.   rM  rN  V  s    npr2   c                     grG  rI  )r  r  r  s      r.   rM  rN  W  r  r2   c                     grG  rI  r@  s    r.   rM  rN  X  r  r2   c                     grG  rI  r@  s    r.   rM  rN  Y  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  Z  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  [  rO  r2   c                     grG  rI  r@  s    r.   rM  rN  \  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  ]  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  ^  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  _  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  `  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  a  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  b  rg  r2   c                     grG  rI  r@  s    r.   rM  rN  c  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  d  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  e  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  f  rc  r2   c                     grG  rI  r@  s    r.   rM  rN  g  ra  r2   c                     grG  rI  r@  s    r.   rM  rN  h  rc  r2   c                     grG  rI  r@  s    r.   rM  rN  i  rg  r2   c                     grG  rI  r@  s    r.   rM  rN  j  rc  r2   c                     grG  rI  r@  s    r.   rM  rN  k  r  r2   c                     grG  rI  r@  s    r.   rM  rN  l  rc  r2   c                     grG  rI  r@  s    r.   rM  rN  m  r_  r2   c                     grG  rI  r@  s    r.   rM  rN  n  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  o  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  p  rO  r2   c                     grG  rI  r@  s    r.   rM  rN  q  rZ  r2   c                     grG  rI  r@  s    r.   rM  rN  r  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  s  rc  r2   c                     grG  rI  r@  s    r.   rM  rN  t  r  r2   c                     grG  rI  r@  s    r.   rM  rN  u  rO  r2   c                     grG  rI  r@  s    r.   rM  rN  v  r_  r2   c                     grG  rI  r@  s    r.   rM  rN  w  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  x  rM  r2   c                     grG  rI  r@  s    r.   rM  rN  y  rU  r2   c                     grG  rI  )r  rb   non_blockingr%   s       r.   rM  rN  z  r+  r2   c                     grG  rI  r@  s    r.   rM  rN  {  rE  r2   c                     grG  rI  r@  s    r.   rM  rN  |  rE  r2   c                     grG  rI  r@  s    r.   rM  rN  }  rR  r2   c                     grG  rI  r@  s    r.   rM  rN  ~  rR  r2   c                     grG  rI  r@  s    r.   rM  rN        "r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  s      r.   rM  rN    r[  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                     grG  rI  rh  s     r.   rM  rN    ra  r2   c                     grG  rI  )r  orderellipsis_idxs      r.   rM  rN    r[  r2   c                     grG  rI  )r  callables     r.   rM  rN    r_  r2   c                     grG  rI  rL  s      r.   rM  rN    r  r2   c                     grG  rI  rL  s      r.   rM  rN    r  r2   c                     grG  rI  )r  gradientretain_graphcreate_graphrW  s        r.   rM  rN    s    ikr2   c                     grG  rI  r  ri   s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   )r  c                    grG  rI  )r  mediansigmar  s       r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rR  r2   c                     grG  rI  )r  	coalesceds     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rn  r2   c                     grG  rI  )r  r  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rR  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r  r  s        r.   rM  rN    rC  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  ambiguity_checks     r.   rM  rN    rU  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rt  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r  s     r.   rM  rN    rM  r2   c                     grG  rI  rh  s     r.   rM  rN    r_  r2   c                    grG  rI  )r  re  r  s      r.   rM  rN    r  r2   c                     grG  rI  r  rs  s     r.   rM  rN    rM  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r}  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                    grG  rI  )r  r  r  s      r.   rM  rN    rU  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  r  s     r.   rM  rN    r}  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                     grG  rI  r  s     r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    rO  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r\   s     r.   rM  rN    rc  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                    grG  rI  )r  r  r  r  s       r.   rM  rN    r  r2   c                     grG  rI  r  s     r.   rM  rN    r_  r2   c                     grG  rI  r  s     r.   rM  rN    rI  r2   c                     grG  rI  )r  r\   r  s      r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r  s       r.   rM  rN    r  r2   c                     grG  rI  )r  ry  r   s      r.   rM  rN    r  r2   c                     grG  rI  )r  rf  assigns      r.   rM  rN    rU  r2   c                     grG  rI  )r  re  r<  r  s       r.   rM  rN    rx  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rR  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                     grG  rI  r@  s    r.   rM  rN    rE  r2   c                     grG  rI  r  s     r.   rM  rN    rM  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  r\   r  s       r.   rM  rN    r)  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                    grG  rI  )r  from_tor  s       r.   rM  rN    rt  r2   c                     grG  rI  )r  streams     r.   rM  rN    r  r2   c                     grG  rI  )r  r  s     r.   rM  rN    r  r2   c                     grG  rI  r  hooks     r.   rM  rN    r  r2   c                     grG  rI  r2  s     r.   rM  rN    r  r2   c                     grG  rI  )r  names     r.   rM  rN    rM  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  )r  r  s     r.   rM  rN    r:  r2   c                     grG  rI  rh  s     r.   rM  rN    rc  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  r  s     r.   rM  rN    rO  r2   c                     grG  rI  rh  s     r.   rM  rN    r_  r2   c                     grG  rI  rh  s     r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r}  r  r  r  s        r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r7  s       r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    rO  r2   c                     grG  rI  r  s     r.   rM  rN    r}  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  r  r<  r  r  s         r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  r  s     r.   rM  rN    rc  r2   c                     grG  rI  )r  r  accumulate_matchess      r.   rM  rN    r  r2   c                     grG  rI  r  size1size2	dense_dims       r.   rM  rN    r)  r2   c                     grG  rI  rJ  s       r.   rM  rN    rn  r2   c                     grG  rI  )r  rx  ry  rm  rl  rL  s         r.   rM  rN    rC  r2   c                     grG  rI  r@  s    r.   rM  rN    r]  r2   c                     grG  rI  r@  s    r.   rM  rN    ra  r2   c                     grG  rI  r@  s    r.   rM  rN    rZ  r2   c                     grG  rI  r@  s    r.   rM  rN    rM  r2   c                     grG  rI  r  s     r.   rM  rN    rc  r2   c                     grG  rI  )r  repss     r.   rM  rN    r  r2   c                     grG  rI  )r  rb   r  copyri   s        r.   rM  rN    s    lnr2   )masked_gradc                    grG  rI  r  rb   rY  s      r.   rM  rN    rI  r2   c                     grG  rI  r[  s      r.   rM  rN    r)  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  rh  s     r.   rM  rN    rZ  r2   c                     grG  rI  rd  s       r.   rM  rN    r[  r2   c                     grG  rI  )r  r,  r-  s      r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   c                     grG  rI  )r  rg  s     r.   rM  rN    r  r2   c                     grG  rI  rh  s     r.   rM  rN    rZ  r2   c                     grG  rI  r@  s    r.   rM  rN    rE  r2   c                     grG  rI  )r  r/  max_version	dl_devicerX  s        r.   rM  rN    r_  r2   c                     grG  rI  r@  s    r.   rM  rN    rc  r2   c                     grG  rI  )r  r  r  s      r.   rM  rN    r  r2   c                     grG  rI  )r  r  r   drivers       r.   rM  rN    r  r2   c                     grG  rI  )r  ra   r  r%   s       r.   rM  rN    r  r2   c                     grG  rI  r@  s    r.   rM  rN    r  r2   is______i__rbitwise_c                     grG  rI  )r\   r  rM  async_op	group_srcs        r.   rM  rN  +  r$  r2   c                     grG  rI  )r\   oprM  rw  s       r.   rM  rN  ,  r  r2   c                     grG  rI  )r\   dstrz  rM  rw  	group_dsts         r.   rM  rN  -  rP  r2   c                     grG  rI  )r  rz  rM  rw  s       r.   rM  rN  .  r  r2   c                     grG  rI  )tensor_listr\   rM  rw  s       r.   rM  rN  /  r  r2   c                     grG  rI  )output_tensorinput_tensorrM  rw  s       r.   rM  rN  0  r  r2   c                     grG  rI  )output_tensor_listsinput_tensor_listrM  rw  s       r.   rM  rN  1  r  r2   c                     grG  rI  )r\   gather_listr|  rM  rw  r}  s         r.   rM  rN  2  r  r2   c                     grG  rI  )r\   scatter_listr  rM  rw  rx  s         r.   rM  rN  3  r  r2   c                     grG  rI  )r  
input_listrz  rM  rw  s        r.   rM  rN  4  r$  r2   c                     grG  rI  )r  rK  rz  rM  rw  s        r.   rM  rN  5  rn  r2   c                     grG  rI  )r  rK  output_split_sizesinput_split_sizesrM  rw  s         r.   rM  rN  6  s	      LNr2   c                     grG  rI  )output_tensor_listr  rM  rw  s       r.   rM  rN  7  rP  r2   c                     grG  rI  r\   r|  rM  tagr}  s        r.   rM  rN  8  ru  r2   c                     grG  rI  r\   r  rM  r  rx  s        r.   rM  rN  9  ru  r2   c                     grG  rI  r  s        r.   rM  rN  :  r1  r2   c                     grG  rI  r  s        r.   rM  rN  ;  r1  r2   N)rS  rS  N)rS  N)h㈵>:0yE>F)F)NFN)Nr   FT)NN)NNNr  N)Nr   )FFN)r   N)       @#use_mm_for_euclid_dist_if_necessary)r  F)FN)NNN)rS  NN)   F)NrS  r   rS  rS  )NrS  r   r   rS  rS  )r   NNr  )rS  r  )rH  N)r   r  FrG  )rS  rH  NNN)r   r   rS  )r   rH  )r  )LN)NNr  FF)Nr  Fr  FNN)NNNF)FF)NrH  N)Nr  rH  N)r   rH  )TT)NF)NNrS  )NNr  T)      ?)NFr   N)r  NNr  )d   r   r   N)r  NNNFN)NNF)T)NNNTFNNF)NNr  F)NNNNNNNNNNNNN)TFN)TN)Nr   rS  F)Nr   rS  FF)NFNN)rH  FN)        NNN)NNr  )Nr  )r  NFN)r  FF)Nr   FTN)NNF皙?r  )NNNr  )NNNr  r  )r  TF)	NNr  Fr  FNFN)rS  r   rS  )NNFN)Fư>r  )none)rH  )bilinearr   N)rS  Fg|=rH  )g      r  F)NNNNTr  r  )NNnearestNNF)NNr  N)g{Gz?F)g-C6?g      ?r  )Nr4   N)Nr   N)TNTNFNNNNNNF)rS  r  NNNr  )NNr  )NNr  Nr  )r  rS  g-q=N)r   r   )r  r  F)TFNr  Nr  )Nr  )g      ?gUUUUUU?FF)Nr  )NNr  r  )r  r  N)rS     )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)rI  rQ  rR  F)rI  )r   r   )rS  rS  F)rI  )r   r   r   )rS  rS  rS  F)rS  ro  )r8  NNNr   F)r   NNrS  )rH  F)	NNNTreflectFTNN)TTN)   r  N)r  N)TFF)TFFN)rS  r  )Nr   NN)NNNN)NNNFN)NNr   N(  r6   r7   rP  absoluteadaptive_avg_pool1dadaptive_max_pool1dacosr  arccosacosharccoshr=  addbmmaddcdivaddcmuladdmmaddmvaddraffine_grid_generatorallallclosealpha_dropoutamaxaminaminmaxangleanyargmaxargminargsortasin_assert_asyncarcsinasinharcsinhatanarctanatan2arctan2atanharctanh
atleast_1d
atleast_2d
atleast_3d
avg_pool1dbaddbmm
batch_normbatch_norm_backward_elemtbatch_norm_backward_reducebatch_norm_elemtbatch_norm_gather_stats#batch_norm_gather_stats_with_countsbatch_norm_statsbatch_norm_update_stats	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dotrW  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igammacrL  	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_expr8  maximumfmax
max_pool1d
max_pool2d
max_pool3dmax_pool1d_with_indicesr  nanmeanr  	nanmedianmeshgridr7  minimumfminmiopen_batch_normmiopen_convolutionmiopen_convolution_add_relumiopen_convolution_relumiopen_convolution_transposemiopen_depthwise_convolution
miopen_rnnmmr  movedimmoveaxismsortmulmultiplymultinomialmvmvlgammanarrow
nan_to_numnative_batch_norm_native_batch_norm_legitnative_dropoutnative_layer_norm_fused_rms_normnative_group_normnative_normnative_channel_shufflene	not_equalnegr  	nextafterr   r   adaptive_avg_pool2dadaptive_avg_pool3d adaptive_max_pool1d_with_indicesadaptive_max_pool2d adaptive_max_pool2d_with_indicesadaptive_max_pool3d adaptive_max_pool3d_with_indicesaffine_grid
avg_pool2d
avg_pool3dbinary_cross_entropycross_entropy	dropout1d	dropout2d	dropout3delufoldfractional_max_pool2d"fractional_max_pool2d_with_indicesfractional_max_pool3d"fractional_max_pool3d_with_indicesgaussian_nll_lossgeluglugrid_samplegumbel_softmaxhardtanhinterpolatel1_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_hotrZ  pairwise_distancepoisson_nll_lossprelurelurelu6rms_normrreluselusilumishscaled_dot_product_attentionsmooth_l1_loss
huber_losssoft_margin_losssoftmaxsoftminsoftplus
softshrinksoftsign
tanhshrinkr  triplet_margin_loss!triplet_margin_with_distance_lossunfoldr   uniform_normal_	constant_kaiming_uniform_nonzerononzero_staticargwherer  vector_normmatrix_normnorm_except_dimnuclear_normr-  orgqrormqrpermutepca_lowrankpdistpinversepinvpixel_shufflepixel_unshufflepoisson	polygammar  	ones_liker  prodputq_per_channel_axisq_per_channel_scalesq_per_channel_zero_pointsq_scaleq_zero_pointqrquantilenanquantilequantize_per_channelquantize_per_tensorquantize_per_tensor_dynamicquantized_batch_normquantized_gru_cellquantized_lstm_cellquantized_max_pool1dquantized_max_pool2dquantized_max_pool3dquantized_rnn_relu_cellquantized_rnn_tanh_cellrad2degravelrK  vdotvecdotview_as_realview_as_complex
reciprocal	remainderrenormrepeat_interleavereshapernn_relurnn_relu_cellrnn_tanhrnn_tanh_cellrollrot90round	row_stack_rowwise_prunersqrtrsubsaddmmscatterscatter_addscatter_reducesearchsorted_segment_reduceselectselect_scatterslice_inverseslice_scatterr   signsignbitsgnsinsincsinhslogdetsmmspmmr  solve_exsortsplitsplit_with_sizessqrtsquaresqueezesspaddmmstackr  std_meanstftsubsubtractsum	sym_floatsym_intsym_maxsym_minsym_notsym_itesym_sum	_sym_sqrt_sym_cos	_sym_cosh_sym_sin	_sym_sinh_sym_tan	_sym_tanh	_sym_asin	_sym_acos	_sym_atannansumsvdsvd_lowranksvdvalsswapaxesswapdimsspecialairy_ai	bessel_j0	bessel_j1	bessel_y0	bessel_y1chebyshev_polynomial_tchebyshev_polynomial_uchebyshev_polynomial_vchebyshev_polynomial_wentrerfcxexpitgammainc	gammainccgammalnhermite_polynomial_hhermite_polynomial_hei0ei1i1elaguerre_polynomial_llegendre_polynomial_plog_ndtrmodified_bessel_i0modified_bessel_i1modified_bessel_k0modified_bessel_k1multigammalnndtrndtripsiscaled_modified_bessel_k0scaled_modified_bessel_k1shifted_chebyshev_polynomial_tshifted_chebyshev_polynomial_ushifted_chebyshev_polynomial_vshifted_chebyshev_polynomial_wspherical_bessel_j0xlog1pyzetattaketake_along_dimtanr   	tensorinvtensorsolve	tensordottensor_splittiletopktracer  trapz	trapezoidtriangular_solvesolve_triangulartriltriutrue_dividetruncunbindr  uniqueunique_consecutiveunravel_indexunsafe_chunkunsafe_splitunsafe_split_with_sizes	unsqueezer   r  var_meanvsplitvstackwhere_wrapped_linear_prepack#_wrapped_quantized_linear_prepacked
zeros_like_fw_primal_copy_make_dual_copyview_as_real_copyview_as_complex_copy
_conj_copy_neg_view_copyas_strided_copy_sparse_broadcast_to_copydiagonal_copyexpand_copynarrow_copypermute_copy_reshape_alias_copyselect_copydetach_copy
slice_copy
split_copysplit_with_sizes_copysqueeze_copyt_copytranspose_copyunsqueeze_copy_indices_copy_values_copyindices_copyvalues_copycrow_indices_copycol_indices_copyccol_indices_copyrow_indices_copyunbind_copy	view_copyunfold_copy
alias_copy__floordiv____rfloordiv____ifloordiv____truediv____rtruediv____itruediv__
__lshift____rlshift____ilshift__
__rshift____rrshift____irshift____and____or____xor__	__float____complex__	__array____bool____contains____neg__
__invert____mod____rmod____imod____array_wrap____getitem____deepcopy____int____long__	__index____len__
__format____reduce_ex____reversed____repr____setitem____setstate__Tr.  HmTmH_backward_hooks_post_accumulate_grad_hooksrA  _cdatarB  rC  _grad_fngrad_fn
grad_dtype_version_autocast_to_reduced_precision_autocast_to_full_precision#_clear_non_serializable_cached_datar  ra   rb   is_cudais_cpuis_xlais_xpuis_ipuis_leafretains_gradis_metais_mpsis_mtia	is_nestedis_maia	is_mkldnnis_quantized	is_sparseis_sparse_csr	is_vulkanitemsizerp   r6  r  nbytesndim	output_nrr  rg  volatile__cuda_array_interface__type_dimI_dimV_indices_is_view_nnzcrow_indicescol_indicesccol_indicesrow_indices_update_names_valuesalign_asalign_toapply_rs   as_strided_backwardbfloat16preserve_formatboolbytecharcauchy_coalesce_coalesced_
contiguouscontiguous_formatcopy_cpucudamtiaxpuipudata_ptrrM  r  	dim_orderdoublecdoubleelement_sizeexpand	expand_asexponential_fill_fill_diagonal_floatcfloat
geometric_halfchalf	has_namesr  intis_coalescedis_contiguous	is_pinned	is_set_to	is_shareditemlog_normal_longmap_map2_module_load
ndimensionnelement_nested_tensor_size_nested_tensor_storage_offsets_nested_tensor_stridesnumpy
pin_memoryput_rj   random_record_streamrefine_namesregister_hook"register_post_accumulate_grad_hookrenamerepeatrequires_grad_
reshape_asresizeresize_	resize_asresize_as_sparse_retain_gradset_share_memory_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_asrh  viewview_aszero_
__dlpack____dlpack_device__r7  r  utilsbackend_registration_privateuse1_backend_namehasattrgetattrr   items__name__
startswithlenextendr  updatedistributedis_availabletorch.distributed	broadcast
all_reducer  all_reduce_coalesced
all_gatherall_gather_into_tensorall_gather_coalescedreduce_scatterreduce_scatter_tensorall_to_all_single
all_to_allisendirecvsendrecv)r7   retprivateuse1_backend_nameret2ignoredr  r  r  subnamer6  r   r6  s               r.   r   r     s5Y   6 \\F{%		-{%2{% 	!!#@{% 	!!#A	{%
 	

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

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

.+{%, 	

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Z 

Z  		Y!" 		Y#	
, Jr2   
dispatcherc                 d   ^  S[         [        [        4   S[         [        [        4   4U 4S jjnU$ )a=  Wraps a given function with ``__torch_function__`` -related functionality.

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

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

Examples
--------
>>> def dispatcher(a):  # Must have the same signature as func
...     return (a,)
>>> @torch.overrides.wrap_torch_function(dispatcher)
>>> def func(a):  # This will make func dispatchable by __torch_function__
...     return a + 0
r   r"   c                    >^ ^ [         R                  " T 5      S[        R                  S[        R                  S[
        4UU U4S jj5       m[        [        [        [
        4   T5      $ )Nr$   r%   r"   c                     > T" U 0 UD6n[        U5      (       a+  [        [        [        [        [
        4   T5      U/U Q70 UD6$ T" U 0 UD6$ r  )r   r   r   r   r   r   )r$   r%   relevant_argsr  r   wrappeds      r.   r  3wrap_torch_function.<locals>.inner.<locals>.wrapped]  sa    &77M!-00,"b&)73]EIMS  (((r2   )	functoolsr   r   r$   r%   r   r   r   )r   r  r  s   `@r.   r  "wrap_torch_function.<locals>.inner\  sX    			)277 	)bii 	)B 	) 	) 
	) HRV$g..r2   )r   r   r   )r  r  s   ` r.   r   r   B  s/    4/HRV$ /"b&)9 / Lr2   r  get_type_fnc                    Uc  [         n[        R                  R                  5       (       d  / $ [	        5       n/ nU  H  nU" U5      nXR;  d  M  [        US5      (       d  M%  UR                  [        R                  R                  Ld  MN  U(       a]  UR                  U5        [        U5      n[        U5       H  u  px[        XQ" U5      5      (       d  M  Un  O   UR                  Xd5        M  U1nU/nM     U$ )a  Returns a list of arguments on which to call __torch_function__.

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

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

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

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

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

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

.. _NEP-0018:
   https://numpy.org/neps/nep-0018-array-function-protocol.html
r  )r
  r6   _C_is_torch_function_enabledsetr
  r  _disabled_torch_function_implr=  r
  	enumerate
issubclassinsert)	r  r  overloaded_typesoverloaded_argsargarg_typer7  iold_args	            r.   _get_overloaded_argsr+  l  s    J  88..00	"%%!#Os# ,"677++8899:
   $$X. O,"+O"<JA!(K,@AA ! #=  &&u2$,: #&%; < r2   
public_apir$   r%   c           	         [        U5      n[        [        [        U5      5      n[	        5       (       a0  [        5        nUR                  XX#5      nSSS5        W[        La  U$ U H|  nUR                  n	[        U	S5      (       aF  U	R                  UL a7  U	[        R                  R                  La  [        R                  " S[        SS9  U	" XX#5      nU[        Ld  Mz  Us  $    U R                    SU R"                   3n
SU
 SU Vs/ s H  n[        U5      PM     sn 3n[	        5       (       a  US	[%        5        3-  n['        U5      e! , (       d  f       GN= fs  snf )
a  Implement a function with checks for ``__torch_function__`` overrides.

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

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

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

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

Example
-------
>>> def func(a):
...     if has_torch_function_unary(a):
...         return handle_torch_function(func, (a,), a)
...     return a + 0
N__self__zDefining your `__torch_function__ as a plain method is deprecated and will be an error in future, please define it as a classmethod.r  
stacklevel.zno implementation found for 'z.' on types that implement __torch_function__: z nor in mode )r+  tuplemapr
  r   _pop_mode_temporarilyr  NotImplementedr
  r.  r6   r  r!  r*   warnDeprecationWarning
__module__r
  _get_current_function_mode	TypeError)r,  r  r$   r%   r&  typesr  resultoverloaded_argtorch_func_method	func_namer'  r  s                r.   r   r     s_   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	E

Ea  Check for __torch_function__ implementations in the elements of an iterable
    or if a __torch_function__ mode is enabled.  Considers exact ``Tensor`` s
    and ``Parameter`` s non-dispatchable.  Use this to guard a call to
    :func:`handle_torch_function`; don't use it to test if something
    is Tensor-like, use :func:`is_tensor_like` instead.
    Arguments
    ---------
    relevant_args : iterable
        Iterable or arguments to check for __torch_function__ methods.
    Returns
    -------
    bool
        True if any of the elements of relevant_args have __torch_function__
        implementations, False otherwise.
    See Also
    ________
    torch.is_tensor_like
        Checks if something is a Tensor-like, including an exact ``Tensor``.
    zSpecial case of `has_torch_function` for single inputs.
    Instead of:
      `has_torch_function((t,))`
    call:
      `has_torch_function_unary(t)`
    which skips unnecessary packing and unpacking work.
    a'  Special case of `has_torch_function` that skips tuple creation.

    This uses the METH_FASTCALL protocol introduced in Python 3.7

    Instead of:
      `has_torch_function((a, b))`
    call:
      `has_torch_function_variadic(a, b)`
    which skips unnecessary packing and unpacking work.
    c                     [         R                  " [        5      n 0 nS[        [        R                  4S[        R
                  [        R
                  R                  4S[        R                  R
                  [        [        R                  R
                  5      4S[        R                  R                  [        [        R                  R                  5      4S[        R                  [        [        R                  5      4S[        R                  [        [        R                  5      4S[        R                  [        [        R                  5      4S[        R                  [        [        R                  5      4/nU GH  u  p4nU GHt  nS	nU[        R                  Lan  UR                  S
5      (       a  M1  UR                  S5      (       a  SnOgUR                  S5      (       a  SnONUS   R                  5       (       d  SnO3US:X  a  M  O*[!        XF5      n[!        ["        US 5      U:X  a  M  US:X  a  M  [!        XF5      nU[        R                  L a  [!        ["        US 5      U:X  a  M  [%        U[&        R(                  5      (       a  GM  [%        U[*        R,                  5      (       a  GM*  [/        U5      (       d  [1        US5      (       a  U SU S3XR2                  '   U SU S3XR4                  '   U(       a  GM}  UR2                  [7        5       ;   aA  Sn	UR2                  [9        5       ;   a$  [;        U	R=                  XHR>                  5      5      eGM  X   RA                  UR2                  5        GM  [/        U5      (       d  GM	  U SU 3X'   U(       a  GM  U[7        5       ;   a7  Sn	U[9        5       ;   a$  [;        U	R=                  XHR>                  5      5      eGMa  X   RA                  U5        GMw     GM     X4$ )Nr6   ztorch.functionalztorch.nn.functionalztorch.nn.initztorch.Tensorztorch.linalgz	torch.fftztorch.specialFrr  rq  Tr   
unique_dim__weakref__r.  r1  z.__get__z.__set__zk{}.{} is in the tuple returned by torch._overrides.get_ignored_functions but still has an explicit override)!collectionsdefaultdictlistr6   __all__r   r   dirr   r7   r  r   r	  r
  endswithislowerr
  object
isinstancer;  
ModuleType
__future___Featurer  r
  r.  __set__r   r   AssertionErrorformatr
  r  )
overridable_funcsr7  tested_namespacesnamespace_str	namespacens_funcsr?  r'   r   r  s
             r.   _get_overridable_functionsrW  E  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 ##r2   c                      [        5       S   $ )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   )rW  rI  r2   r.   r   r     s     &'**r2   c                     [        U [        R                  R                  [        R                  R                  45      (       a  [        U 5      $ [        5       S   R                  U 5      $ )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.
rS  )rK  r6   _ops
OpOverloadOpOverloadPacketstrrW  get)fs    r.   r   r     sL      !ejj++UZZ-H-HIJJ1v%'*..q11r2   c                  R    [        5       n [        U [        R                     5      nU$ )z<Returns a set of the overridable methods on ``torch.Tensor``)r   r   r6   r7   )rR  methodss     r.   _get_tensor_methodsrb    s&     23#ELL12GNr2   c                 H    U [        5       ;   =(       d    U R                  S:H  $ )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
r.  )rb  r
  )r   s    r.   r   r     s!    . &((FDMMY,FFr2   c                 ^    [        U 5      [        R                  L =(       d    [        U S5      $ )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
  r6   r7   r
  )inps    r.   r   r     s%    D 9$J5I(JJr2   c                   T    \ rS rSr% SrS \S'   SS jrSS jrS rS	 r	\
S
 5       rSrg)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!)
r  Nc                     g r  rI  r@  s    r.   r   TorchFunctionMode.__init__%  s    r2   rI  c                     [         er  )NotImplementedErrorr  r   r;  r$   r%   s        r.   r  $TorchFunctionMode.__torch_function__(  s    !!r2   c                     [        U 5        U $ r  )
_push_moder@  s    r.   	__enter__TorchFunctionMode.__enter__+  s    4r2   c                     [        5         g r  )	_pop_mode)r  exc_typeexc_valexc_tbs       r.   __exit__TorchFunctionMode.__exit__/  s    r2   c                 @    [         R                  " SSS9  U " U0 UD6nU$ )NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`r  r/  )r*   r6  )clsr$   r%   instances       r.   pushTorchFunctionMode.push2  s*    ^	
 ''r2   )r"   NrI  N)r
  r8  __qualname____firstlineno____doc____annotations__r   r  rp  rw  classmethodr|  __static_attributes__rI  r2   r.   rg  rg    s7    < "  r2   rg  c                  B    [        5       n U S:  a  [        U S-
  5      $ S $ )Nr   rS  )r   r   )	stack_lens    r.   r9  r9  <  s%    )+I4=M!)a-0KtKr2   c                  j    [        5       n [        U 5       Vs/ s H  n[        U5      PM     sn$ s  snf r  )r   r   r   )r  r)  s     r.    _get_current_function_mode_stackr  A  s/    )+I/4Y/?@/?!"1%/?@@@s   0c                     [        U 5        g r  )r   )r  s    r.   ro  ro  F  s
    !$'r2   c                      [        5       n U $ r  )r   olds    r.   rs  rs  J  s    
#
%CJr2   c               #   `   #    [        5       n  U v   [        U 5        g ! [        U 5        f = f7fr  )rs  ro  r  s    r.   r4  r4  O  s$     
+C	3
3s   . .+.c                       \ rS rSrSS jrSrg)BaseTorchFunctionModeiX  rI  Nc                     Uc  0 nU" U0 UD6$ r  rI  rl  s        r.   r  (BaseTorchFunctionMode.__torch_function__Y  s    >FT$V$$r2   r~  )r
  r8  r  r  r  r  rI  r2   r.   r  r  X  s    %r2   r  c               #   Z  #    [         R                  R                  5       n  [         R                  R                  [         R                  R                  R
                  5        S v   [         R                  R                  U 5        g ! [         R                  R                  U 5        f = f7fr  )r6   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               #      #    [         R                  R                  5           S v    S S S 5        g ! f = f! , (       d  f       g = f7fr  )r6   r  _RestorePythonTLSSnapshotrI  r2   r.   r   r   i  s9      
	+	+	-		 
.	- 	 
.	-s%   A61	A36
A A)z.*is deprecated, please use.*r6   r  )Dr  rM  rC  
contextlibr  r;  r;  r*   collections.abcr   r   r   typingr   r   r   typing_extensionsr	   r6   torch._Cr
   r   r   r   r   r   r   r   r   rF  r   r   r]  r1   cacher   r   rD  dictr   r   r
  rE  r+  r$   r%   r   r   r   r   r2  rW  r   r   rb  r
  r   r   rg  r9  r  ro  rs  contextmanagerr4  r  r  r   rI  r2   r.   <module>r     s  ,     
   .  % % ' 
 
 
 t_T]
 1 
2r6
     b"f	 F `s8} `  `F	 c(m  2 XtHh$67 X  Xv$'Xc]*+'xB (2r6"223'X 15LC=L3%+&-L 
#YL^VR VC=V 77V ii	V
 Vr ! . '	  * 	  Q$Ed8n	tHcM22% Q$ Q$h 	+4T(^(;#< 	+ 	+ 2 2( S]   Gx GD G G2"KJ6 6rL
A
(
  %- % 6 6  r2   