� J��gl�� �$�dZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZddlmZddlZddlZddlmZmZmZmZmZmZmZmZmZgd�Z d+d e d ed ed e fd �Zejd��ed ee fd�����Z ejd��d ee fd���Z!ejd��ed ee e ffd�����Z"de fd�Z#d,de e de e ge fd e e fd�Z$de de e d e fd�Z%eed��Z&eed��Z'eed��Z(ejd��d eee e e fee efffd���Z)ed ee e e ffd���Z*ed���Z+ejd��d ee fd���Z,ed e d e-fd���Z.d �Z/Gd!�d"��Z0d#�Z1d$�Z2d%�Z3d&�Z4ej5d'���Z6Gd(�d)e0��Z7ej5d*���Z8dS)-aE Python implementation of ``__torch_function__`` While most of the torch API and handling for ``__torch_function__`` happens at the C++ level, some of the torch API is written in Python so we need python-level handling for ``__torch_function__`` overrides as well. The main developer-facing functionality in this file are handle_torch_function and has_torch_function. See torch/functional.py and test/test_overrides.py for usage examples. Note ---- heavily inspired by NumPy's ``__array_function__`` (see: https://github.com/pytorch/pytorch/issues/24015 and https://www.numpy.org/neps/nep-0018-array-function-protocol.html ) If changing this file in a way that can affect ``__torch_function__`` overhead, please report the benchmarks in ``benchmarks/overrides_benchmark``. See the instructions in the ``README.md`` in that directory. �N)�Dict�Set�List�Any�Callable�Iterable�Type�Tuple��wraps) �_has_torch_function�_has_torch_function_unary�_has_torch_function_variadic� _add_docstr�_push_on_torch_function_stack�_pop_torch_function_stack�_get_function_stack_at�_len_torch_function_stack�_is_torch_function_mode_enabled) �get_ignored_functions�get_overridable_functions�get_testing_overrides�handle_torch_function�has_torch_function� resolve_name�is_tensor_like�is_tensor_method_or_property�wrap_torch_function�enable_reentrant_dispatch�.*is deprecated, please use.*�torch�func�regex�module�returnc�D����t������fd���}|S)a  Decorator that temporarily disables ``UserWarning``s for the given ``module`` if the warning message matches the given ``regex`` pattern. Arguments --------- func : function Function to disable the warnings for. regex : str A regex pattern compilable by ``re.compile``. This is used to match the ``UserWarning`` message. module : str The python module to which the filtering should be restricted. Returns ------- function The wrapped function. c���tj��5tjdt������|i|��cddd��S#1swxYwYdS)N�ignore)�category�messager$)�warnings�catch_warnings�filterwarnings� UserWarning)�args�kwargsr"r$r#s ����_/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/torch/overrides.py�wrapperz'_disable_user_warnings.<locals>.wrapperKs���� � $� &� &� )� )� � #�H�{�E�Z`� a� a� a� a��4��(��(�(� )� )� )� )� )� )� )� )� )� )� )� )���� )� )� )� )� )� )s�%A�A �A r )r"r#r$r2s``` r1�_disable_user_warningsr36sA�����* �4�[�[�)�)�)�)�)�)��[�)� �N�c��tj}htj�tj�tj�tj�tj�tj�tj�tj �tj �tj �tj �tj �tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj�tj �tj!�tj"�tj#�tj$�tj%�tj&�tj'�tj(�tj)�tj*�tj+�tj,�tj-�tj.�tj/�tj0�tj1�tj2�tj3�tj4�tj5�tj6�tj7�tj8�tj9�tj:�tj;�tj<�tj=�tj>�tj?�tj@�tjA�tjB�tjC�tjD�tjE�tjF�tjG�tjH�tjI�tjJ�tjK�tjL�tjM�tjN�tjOjP�tjOjO�tjOj�tjOjQ�tjOj �tjR�tjSjT�tjSjU�tjV�tjW�tjX�tjY�tjZ�tj[�tj\�tj]�tj^�tj_�tj`�tja�tjb�tjc�tjd�tje�tjf�tjg�tjh�tji�tjj�tjk�tjl�tjm�tjn�tjo�tjp�tjq�tjr�tjs�tjt�tju�tjv�tjw�tjx�tjy�tjzj{�tj|j}j~�tj|j}j�tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j}j��tj|j�j��tj|j�j��tj|j�jd�tj|j�j��tj|j�jR�tj|j�j��tj|j�j��tj|j�j��tj|j�j��tj|j�j��tj|j�j��tj|j�j��tj�j���t��t �tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj|j}j��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��tj��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j�j��|j�j��|j��|j��|j��|j��|j��|j��|j��|j��|j��|j��S)a% Return public functions that cannot be overridden by ``__torch_function__``. Returns ------- Set[Callable] A tuple of functions that are publicly available in the torch API but cannot be overridden with ``__torch_function__``. Mostly this is because none of the arguments of these functions are tensors or tensor-likes. Examples -------- >>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions() True >>> torch.add in torch.overrides.get_ignored_functions() False (r!�Tensor�typename� is_tensor� is_storage�set_default_tensor_type�set_default_device�get_default_device� set_rng_state� get_rng_state� manual_seed� initial_seed�seed�save�load�set_printoptions�fork�get_default_dtype�get_num_interop_threads�get_num_threads�init_num_threads�import_ir_module�import_ir_module_from_buffer�is_anomaly_enabled�is_anomaly_check_nan_enabled�is_grad_enabled�merge_type_from_type_comment�parse_ir� parse_schema�parse_type_comment�set_anomaly_enabled�set_flush_denormal�set_num_interop_threads�set_num_threads�wait� as_tensor� from_numpy� get_device�tensor�default_generator�has_cuda� has_cudnn� has_lapack�device�dtype�finfo�has_mkl�has_mps� has_mkldnn� has_openmp�iinfo� memory_format�qscheme�set_grad_enabled�no_grad� enable_grad�inference_mode�is_inference_mode_enabled�layout� align_tensors�arange� as_strided�bartlett_window�blackman_window�broadcast_shapes�can_cast�compile�cudnn_affine_grid_generator�cudnn_batch_norm�cudnn_convolution�cudnn_convolution_transpose�cudnn_convolution_relu�cudnn_convolution_add_relu�cudnn_grid_sampler�cudnn_is_acceptable�empty�empty_permuted� empty_strided�empty_quantized�export� dynamic_dim�register_dataclass�eye�fft�fftfreq�rfftfreq� from_file�full�fill�hamming_window� hann_window� kaiser_window�linspace�logspace�mkldnn_adaptive_avg_pool2d�mkldnn_convolution�mkldnn_max_pool2d�mkldnn_max_pool3d�mkldnn_linear_backward_weights�mkldnn_rnn_layer�normal�ones� promote_types�rand�randn�randint�randperm�range� result_type� scalar_tensor�sparse_coo_tensor�sparse_compressed_tensor�sparse_csr_tensor�sparse_csc_tensor�sparse_bsr_tensor�sparse_bsc_tensor�sym_constrain_range�sym_constrain_range_for_size� tril_indices� triu_indices�vander�zeros� _jit_internal�boolean_dispatch�nn� functional�assert_int_or_pair�upsample�upsample_bilinear�upsample_nearestr�has_torch_function_unary�has_torch_function_variadicr�sigmoid� hardsigmoid�tanh�_canonical_mask�_none_or_dtype�init�calculate_gain�uniform�constant�dirac�xavier_uniform� xavier_normal�kaiming_uniform�kaiming_normal� orthogonal�sparse�nested�to_padded_tensor�set_autocast_enabled�is_autocast_enabled�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�vitals_enabled� set_vital� read_vitals�vmap�cond� frombuffer�asarray�_functional_sym_constrain_range�_make_dep_token� __delitem__�__dir__�__getattribute__�__init__�__iter__�__init_subclass__� __delattr__� __setattr__�__torch_function__�__torch_dispatch__�__new__� __class__�__subclasshook__�__hash__� as_subclass�eig�lstsq� reinforce�new� new_tensor� new_empty�new_empty_strided� new_zeros�new_ones�new_full�_make_subclass�solve�symeig�stride� unflatten� to_sparse_coo� to_sparse_csr� to_sparse_csc� to_sparse_bsr� to_sparse_bsc� _to_sparse�_to_sparse_csr�_to_sparse_csc�_to_sparse_bsr�_to_sparse_bsc�_typed_storage�_reduce_ex_internal� _fix_weakref� _view_func�_view_func_unsafe�_rev_view_func_unsafe�_make_wrapper_subclass�_python_dispatch�__get__�_has_symbolic_sizes_strides�_conj�_conj_physical� _lazy_clone� _neg_view�_is_zerotensor� _is_all_true� _is_any_true�_addmm_activation� _use_count�r6s r1rrSs��(�\�F�A � ��A � ��A � ��A � �%� A � � � A � � � A � ��A � ��A � ��A � ��A � � �A � � �A � � �A � ��A � � �A � ��!A �" �%�#A �$ ��%A �& ��'A �( ��)A �* �*�+A �, � �-A �. �*�/A �0 ��1A �2 �*�3A �4 ��5A �6 ��7A �8 � �9A �: �!�;A �< � �=A �> �%�?A �@ ��AA �B � �CA �D ��EA �F ��GA �H ��IA �J � �KA �L ��MA �N ��OA �P ��QA �R ��SA �T � �UA �V � �WA �X � �YA �Z � �[A �\ � �]A �^ ��_A �` ��aA �b � �cA �d ��eA �f � �gA �h ��iA �j � �kA �l ��mA �n ��oA �p �'�qA �r � �sA �t ��uA �v � �wA �x ��yA �z ��{A �| ��}A �~ ��A �@ ��AA �B � �CA �D �)�EA �F ��GA �H ��IA �J �)�KA �L �$�MA �N �(�OA �P � �QA �R �!�SA �T � �UA �V ��WA �X ��YA �Z ��[A �\ � � �]A �^ � ��_A �` � ��aA �b � �'�cA �d � ��eA �f � �gA �h � ��iA �j � ��kA �l ��mA �n � �oA �p � �qA �r ��sA �t ��uA �v ��wA �x ��yA �z ��{A �| �(�}A �~ � �A �@ ��AA �B ��CA �D �,�EA �F ��GA �H � �IA �J � �KA �L ��MA �N � �OA �P � �QA �R � �SA �T ��UA �V � �WA �X ��YA �Z ��[A �\ ��]A �^ �&�_A �` ��aA �b ��cA �d ��eA �f ��gA �h �!�iA �j �*�kA �l ��mA �n ��oA �p � �qA �r � �sA �t ��,�uA �v ���.�wA �x ���$�yA �z ���-�{A �| ���,�}A �~ ���.�A �@ ���4�AA �B ���7�CA �D ���1�EA �F ���#�GA �H ���'�IA �J ��� �KA �L ���+�MA �N ���*�OA �R �� �$�SA �V �� ��WA �X �� ��YA �Z �� ��[A �\ �� ��]A �^ �� ��_A �` �� �$�aA �b �� �#�cA �d �� �%�eA �f �� �$�gA �h �� � �iA �j �� ��kA �l � �%�mA �n �oA �p �qA �r �"�sA �t �!�uA �v � �wA �x � �yA �z �"�{A �| �&�}A �~ �%�A �@ �&�AA �B �%�CA �D �&�EA �F �%�GA �H �$�IA �J �$�KA �L �$�MA �N �$�OA �P �$�QA �R �$�SA �T �$�UA �V �$�WA �X �(�YA �Z �(�[A �\ �'�]A �^ �(�_A �` ���%�aA �b �!�cA �d �2�eA �f �*�gA �h �;�iA �j �*�kA �l ��mA �n �*�oA �p �*�qA �r �*�sA �t ��uA �v �$�wA �x ��yA �z ��{A �| ��}A �~ ��A �@ � �AA �B � �CA �D ��EA �F � �GA �H �-�IA �J ��KA �L ��MA �N ��OA �P ��QA �R ��SA �T ��UA �V � �WA �X ��YA �Z ��[A �\ �!�]A �^ �!�_A �` ��aA �b ��cA �d ��eA �f ��gA �h ��iA �j � �kA �l � �mA �n ��oA �p � �qA �r ��sA �t ��uA �v � �wA �x ��yA �z ��{A �| ��}A �~ ��A �@ � �AA �B � �CA �D � �EA �F ��GA �H ��IA �J ��KA �L ��MA �N ��OA �P ��QA �R ��SA �T ��UA �V ��WA �X ��YA �Z ��[A �\ ��]A �^ �"�_A �` ��aA �b ��cA �d � �eA �f �$�gA �h �%�iA �j ��'�kA �l �*�2�mA �n � �oA �p ��qA �r ��sA �t ��uA �v ��wA �x ��yA �z ��{A �| � �}A �~ ��A �@ ��AA �Ar4c�`�tj}|jj|jj|jjhS)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). )r!r6�_baser(�grad�_gradr3s r1�get_default_nowrap_functionsr8ls-��$�\�F�� ��� ��� �� �r4c����tj}itj�d�d��tj�d�d��tjd��tjd��tj�d�d��tjd��tj�d�d��tj �d�d ��tj �d�d ��tj �d�d ��tj �dd ��tj �dd��tj�dd��tj�dd��tj�dd��tj�dd��tjd��itj�d�d��tj�dd��tj�dd��tj�d�d��tj�d�d��tj�dd��tj�d�d��tj�dd��tjd��tjd ��tj�d�d!��tj�d�d"��tjd#��tj �d�d$��tj!�d�d%��tj"�d�d&��tj#�d�d'���itj$�d�d(��tj%�d�d)��tj&�d�d*��tj'�d�d+��tj(�d�d,��tj)d-��tj*d.��tj+d/��tj,�dd2��tj-�dd3��tj.d4��tj/d5��tj0d6��tj1d7��tj2d8��tj3d9��tj4d:���itj5d;��tj6�dd<��tj7d=��tj8 �dd?��tj9�dd@��tj:�d�dA��tj;�d�dB��tj<�d�dC��tj=�d�dD��tj>�d�dE��tj?�d�dF��tj@�d�dG��tjAdH��tjB�d�dI��tjCdJ��tjDdK��tjE�d dL���itjFdM��tjG�d dN��tjH�d dO��tjI�d dP��tjJ�d dS��tjK�d�dT��tjL�d dV��tjMddW�dX��tjNdY��tjO�d dZ��tjPjO�d�d[��tjPjQ�d d\��tjR�d d]��tjS�d d^��tjTd_��tjU�d1d`��tjV�dda���itjW�ddb��tjX�d�dc��tjY�d�dd��tjZ�d�de��tj[�ddf��tj\dg��tj]�ddi��tj^dj��tj_�d�dk��tj`dl��tjPja�d�dm��tjb�d�dn��tjc�d�do��tjd�d�dp��tje�d�dq��tjf�d1dr��tjg�dds���itjh�ddt��tji�ddu��tjjdv��tjk�d1dw��tjl�ddx��tjm�ddy��tjn�ddz��tjod{��tjp�d�d|��tjq�dd}��tjr�d�d~��tjs�dd��tjtd���tju�dd���tjPju�dd���tjv �dd���tjw�d�d����itjx�d�d���tjy�dd���tjz�dd���tj{�dd���tj|�d�d���tj}�d�d���tj~d���tjd���tjPjd���tj�d���tj��d d���tj��d d���tj��d1d���tj��dd���tj��dd���tjPj��dd���tj��dd����itj��d�d���tj��d�d���tj��dd���tj��dd���tj��dd���tj��d�d���tj��dd���tj�d���tj�d���tj�d���tj��d�d���tjPj��d�d���tjPj��d�d���tjPj��dd���tjPj��dd���tj�d���tj� �dd����itj� �dd���tj��dd���tj��d�d���tj�d���tj��d�d���tj��d�d���tj��d�d���tj��d�d���tj��d�d���tj��d�d���tj�d���tj�d���tj� �d d���tj�d���tj�d���tj�d���tj�d����itj�d���tj�d���tj�d���tj�d���tj�d���tj�j��d!d���tj�j��d!d���tj�j��d!d„�tj�j��d!dÄ�tj�j��d!dĄ�tj�j��d"dƄ�tj�j��d"dDŽ�tj�j��d!dȄ�tj�j��d!dɄ�tj�j��ddʄ�tj�j��dd˄�tj�j��dd̄��itj�j��dd̈́�tj�j��d"d΄�tj�j��d"dτ�tj�j��d"dЄ�tj�j��d"dф�tj�j��d�d҄�tj�j��d�dӄ�tj�j��d!dԄ�tj��d�dՄ�tj��d#dք�tj�dׄ�tj�d؄�tj�dل�tj��ddڄ�tj��d�dۄ�tj�d܄�tj��d�d݄��itj��d�dބ�tj��d�d߄�tj��d�d���tj�ddtj�ddfd��tj�d��tj��d$d��tj��d%d��tj��d�d��tj��d�d��tj��d�d��tj��d�d��tj��d�d��tj��d�d��tj��d�d��tj��d�d��tj��d&d��tj�d���itj�d��tj�d���tj��d'd��tj�d��tj��dd��tj��d�d��tj��d�d���tj��d(d���tj��d�d���tj��d)d���tj��d*d���tj��d+d���tj��d,d���tjPj�d���tj��d�d���tj��d��tj��d��d���itj��d��d��tj��d��d��tj��d��d��tj��d��d��tj��d��tj��d��tj��d�d��tj��d��d ��tj��d ��tj��d-�d ��tj��d ��tj��d �d ��tj��d��tj��d��tj��d��d��tj��d��d��tj��d���itj��d��t�j�d��d��tjP�j�d��d��tjP�j�d �d��t�j�d��t�j�d��t�j�d��t�j�d��t�j�d��t�j�d��t�j �d��t�j �d��t�j �d��t�j �d�d ��t�j �d!��t�j �d.�d"��t�j�d/�d#���it�j�d$��t�j�d�d%��tjP�j�d �d&��tjP�j�d �d'��tjP�j�d �d(��t�j�d'�d)��t�j�d��d*��t�j�d��d+��t�j�d��d,��t�j�d��d-��t�j�d��d.��t�j�d��d/��t�j �d0�d0��t�j�d��d1��t�j�d��d2��t�j�d��d3��t�j �d��d4���it�j!�d��d5��t�j"�d��d6��t�j#�d��d7��t�j$�d8��t�j%�d��d9��t�j&�d��d:��t�j'�d��d;��t�j(�d��d<��t�j)�d��d=��t�j*�d��d>��t�j+�d �d?��t�j,�d@��t�j-�d�dA��t�j.�d��dB��t�j/�d��dC��t�j0�d1�dD��t�j1�d��dE���it�j2�d�dF��t�j3�dG��t�j4�dH��t�j5�d��dI��t�j6�d��dJ��tjP�j0�d2�dK��tjP�j7�d2�dL��tjP�j8�d1�dM��tjP�j1�d1�dN��tjP�j6�d��dO��t�j9�dP��tjP�j9�d��dQ��tjP�j:�d%�dR��tjP�j;�d��dS��t�j<�dT��tjP�j<�dU��t�j=�d��dV���it�j>�d��dW��t�j?�d��dX��t�j@�d3�dY��t�jA�d3�dZ��t�jB�d3�d[��t�jC �d4�d\��t�jD�d��d]��t�jE�d5�d^��t�jF�d��d_��t�jG�d��d`��t�jH�da��t�jI�d��db��t�jJ�d��dc��t�jK�d��dd��t�jL�de��t�jM�df��t�jN�dg���it�jO�dh��t�jP�di��t�jQ�dj��t�jR�dk��t�jS�d��dl��t�jT�d6�dm��t�jU�dn��t�jV�do��t�jW�d �dp��t�jX�d��dq��t�jY�d��dr��t�jZ�d �ds��t�j[�d��dt��t�j\�du��t�j]�dv��t�j^�d7�dx��t�j_�dy���it�j`�dz��t�ja�d{��t�jb�d8�d|��t�jc�d}��t�jd�d9�d~��t�je�d��t�jf�d��d���t�jg�d��d���t�jh�d��d���t�ji�d��d���t�jj�d��d���t�jk�jl�jm�d���t�jk�jl�jn�d���t�jk�jlj�d�d���t�jk�jl�jo�d�d���t�jk�jl�jp�d�d���t�jk�jl�jq�d�d����it�jk�jl�jr�d�d���t�jk�jl�js�d�d���t�jk�jl�jt�d��d���t�jk�jlj�d:�d���t�jk�jl�ju �d;�d���t�jk�jl�jv �d;�d���t�jk�jlj. �d<�d���t�jk�jlj7�d��d���t�jk�jl�jw �d=�d���t�jk�jlj8 �d�d���t�jk�jljL�d �d���t�jk�jljq �d�d���t�jk�jl�jx �d>�d���t�jk�jljv �d�d���t�jk�jlj��d?�d���t�jk�jl�jy�d?�d���t�jk�jl�jz�d?�d����it�jk�jl�j{�d?�d���t�jk�jl�j|�d �d���t�jk�jlj� �d�d���t�jk�jlj� �d@�d���t�jk�jlj��d:�d���t�jk�jl�j}�dA�d���t�jk�jl�j~ �dB�d���t�jk�jl�j �dB�d���t�jk�jl�j� �dB�d���t�jk�jl�j� �dB�d���t�jk�jl�j��dC�d���t�jk�jl�j��dD�d���t�jk�jl�j��dE�d���t�jk�jl�j��dF�d���t�jk�jlj��d8�d���t�jk�jl�j��dG�d���t�jk�jlj��d(�d����it�jk�jl�j��dH�d���t�jk�jlj� �d)�d���t�jk�jlj� �dI�d���t�jk�jl�j� �dJ�d���t�jk�jl�j�d/�d���t�jk�jl�j��dK�d���t�jk�jl�j�d8�d���t�jk�jl�j��dL�d���t�jk�jl�j��d��d���t�jk�jl�j��dM�d„�t�jk�jl�j�dN�dĄ�t�jk�jl�j��dń�t�jk�jl�j��d%�dƄ�t�jk�jl�j��d%�dDŽ�t�jk�jl�j��d%�dȄ�t�jk�jl�j2 �d�dɄ�t�jk�jl�j@ �d4�dʄ��it�jk�jl�jC �d4�d˄�t�jk�jl�jA �d4�d̄�t�jk�jl�j� �d4�d̈́�t�jk�jl�jB �d4�d΄�t�jk�jl�j� �d4�dτ�t�jk�jl�j��dO�dЄ�t�jk�jl�j��dO�dф�t�jk�jl�j��dO�d҄�t�jk�jl�j��dK�dӄ�t�jk�jl�j� �dP�dԄ�t�jk�jl�j� �dQ�dՄ�t�jk�jl�j� �dK�dք�t�jk�jl�j� �d=�dׄ�t�jk�jl�j� �dR�d؄�t�jk�jl�j��dS�dڄ�t�jk�jl�j��dE�dۄ�t�jk�jl�j��dT�d݄��it�jk�jl�j��dU�dބ�t�jk�jl�j� �dV�d߄�t�jk�jl�j��d���t�jk�jl�j��d�d��t�jk�jl�j��d�d��t�jk�jl�j��dW�d��t�jk�jl�j��dX�d��t�jk�jl�j��d�d��t�jk�jl�j��d�d��t�jk�jl�j��d�d��t�jk�jl�j��dY�d��t�jk�jl�j��dZ�d��t�jk�jl�j��d[�d��t�jk�jl�j��dK�d��t�jk�jl�j��dN�d��t�jk�jl�j��dN�d��t�jk�jl�j��d\�d���it�jk�jl�j��d(�d��t�jk�jl�j��d��t�jk�jl�j��d��t�jk�jl�j��d�d���t�jk�jl�j� �d]�d���t�jk�jl�j�ddUdd>�d���d���t�jk�jl�j��dA�d���t�jk�j��j��d^�d���t�jk�j��j��d^�d���t�jk�j��j��d���t�jk�j��j��d_�d���t�j��d�d��t�j�d��d��d��t�j��d��t�j��d`�d��tjP�j��da�d��tjP�j��db�d���itjP�j��dc�d��t�j��dd�d ��t�j��d`�d ��t�j��d ��t�j��d ��t�j��de�d ��t�j��dU�d��t�j��d��t�j��df�d��t�j��d�d��t�j��dg�d��tjP�j��dh�d��t�j��d��t�j��d��t�j��d��d��t�j��d��t�j��d��d���it�j��d��d��t�j��d��t�j��d�d��t�j��d��d��t�j��d��d��t�j��d�d��t�j��d ��t�j��d!��t�j��d"��t�j��d#��t�j��d$��t�j��d2�d%��tjP�j��di�d'��t�j��dj�d)��t�j��dj�d*��t�j��d+��t�j��d,���it�j��d-��t�j��d.��t�j��d/��t�j��d0��t�j��t����d1�d2df�d3��t�j��t����d4�d5df�d6��t�j��t����d7�d8df�d9��t�j��d:��t�j��d;��t�j��d��d<��t�j��d�d=��t�j�dtj�ddf�d>��t�j��d�d?��t�j��d@��t�j��d��dA��t�j��d��dB��tjP�j��d�dC���it�j��dD��t�j��dE��t�j��d��dF��t�j��d�dG��t�j��d��dH��t�j��d��dI��t�j��d��dJ��t�j��dK��t�j��dW�dL��t�j��dM��t�j��d�dN��t�j��dO��t�j��d�dP��t�j��d��dQ��t�j��dk�dS��t�j��d��dT��t�j��d��dU���it�j�dV��t�j��dX�dW��t�j�d��dX��t�j�d2�dY��t�j�d�dZ��t�j�d[��t�j�d\��t�j�d-�d]��t�j�d �d^��t�j�dl�d`��t�j �da��t�j �db��t�j �dm�dc��t�j �dm�dd��t�j��d�de��t�j �d��df��t�j�d��dg���it�j�d��dh��t�j�d��di��t�j�d��dj��t�j�d��dk��t�j�d��dl��t�j�dm��tjP�j�dn��t�j�do��t�j�dp��t�j��d��dq��tjP�j�d2�dr��tjP�j�d1�ds��t�j�dndd�dt��du��t�j�d1�dv��t�j�d1�dw��t�j�d��dx��t�j�d��dy���it�j�d�dz��t�j�d�d{��t�j �d �d|��t�j!�d��d}��t�j"�d��d~��t�j# �do�d���t�j$�d��d���t�j%�d��d���t�j&�d��d���t�j'�d���t�j(�d���t�j)�d���t�j*�d���t�j+�d���t�j,�d���t�j-�d���t�j.�d����it�j/�d���t�j0�d���t�j1�d���t�j2�d���t�j3�d���t�j4�d���t�j5�d���t�j6�d���t�j7�d��d���t�j8�dp�d���t�j9�dq�d���tjP�j8�d2�d���tjP�j:�d��d���t�j;�d���t�j<�d���t�j=�j>�d���t�j=�j?�d����it�j=�j@�d���t�j=�jA�d���t�j=�jB�d���t�j=�jC�d��d���t�j=�jD�d��d���t�j=�jE�d��d���t�j=�jF�d��d���t�j=j��d���t�j=�jG�d���t�j=j��d���t�j=j��d���t�j=�jH�d���t�j=j��d���t�j=j��d���t�j=�jI�d���t�j=j��d���t�j=�jJ�d��d����it�j=�jK�d��d���t�j=�jL�d���t�j=�jM�d��d���t�j=�jN�d��d���t�j=j��d���t�j=�jO�d���t�j=�jP�d���t�j=�jQ�d���t�j=�jR�d��d���t�j=�jS�d��d���t�j=�j �d���t�j=�jT�d���t�j=�j�d��d���t�j=�j*�d���t�j=�j+�d �d���t�j=�jU�d���t�j=�jV�d����it�j=�jW�d���t�j=�jX�d���t�j=�jY�d„�t�j=�jZ�dÄ�t�j=�j[�dĄ�t�j=�j��d��dń�t�j=�j\�dƄ�t�j=�j��dDŽ�t�j=�j]�dȄ�t�j=�j^�dɄ�t�j=�j_�d��dʄ�t�j=�j`�d��d˄�t�j=�ja�d��d̄�t�j=�jb�d��d̈́�t�j=�j�d΄�t�j=�j��d��dτ�t�j=�jc�dЄ��it�j=�jd�d��dф�t�j=�j%�d��d҄�t�j=�je�d��dӄ�t�jf�dԄ�t�jg�dՄ�t�jh�d�dք�t�ji�d��dׄ�t�jj�d��d؄�tjP�jk�d�dل�tjP�jl�d��dڄ�t�jm�dr�dۄ�t�jn�d1�d܄�t�j��d�d݄�t�jo�dބ�t�jp�d6�d߄�t�jq�d���t�jr�d���it�js�d�d��t�jt�d�d��t�ju�ds�d��tjP�jv�de�d��t�jw�d �d��t�j� �d]�d��t�jx�d �d��t�jy�d��t�jz�d��d��t�j{�d1�d��t�j|�d��t�j}�dt�d��t�j~�d �d��t�j�d��t�j��d1�d���t�j��d1�d��t�j��d1�d���it�j��d��d��tjP�j��d��d��t�j��d��d���t�j��d��d���t�j��d���t�j��d��d���t�j��d�d���t�j��d�d���t�j��d���t�j��d���t�j��d���t�j��d���t�j��d���t�j��d��t�j��d��d��t�j��d��t�j��d�d���it�j�d�d��d��t�j��d��t�j��d��t�j��d��t�j��d ��t�j��d ��t�j��dm�d ��t�j��d1�d ��t�j��d1�d ��t�j��d��t�j��d��t�j��d��t�j��d��t�j��d��t�j��d��t�j��d��t�j��d���it�j��d��t�j��d��t�j��d��t�j��d��t�j��d1�d��t�j��d��t�j��d��t�j��d��|�j��d��|�j��d��|�j��d ��|�j��d!��|�j��d"��|�j��d#��|�j��d$��|�j��d%��|�j��d&���i|�j��d'��|�j��d(��|�j��d)��|�j��d*��|�j��d+��|�j��d,��|�j��d-��|�j��d.��|�j��d/��|�j��d0��|�j��d1��|�j��d2��|�j��d3��|�j��d4��|�j��d5��|�j��d6��|�j��d7���i|�j��d8��|�j��d9��|�j��d:��|�j��d;��|�j��d<��|�j��d=��|�j��d>��|�j��d?��|�j��d@��|�j�d�dA��dB��|�j��dC��|�j��dD��|�j��j��dE��|�j��j��dF��|�j��j��dG��|�j��j��dH��|�j��j��dI���i|�j��j��dJ��|�j��j��dK��|�j��j��dL��|�j��j��dM��|�j��j��dN��|�j��j��dO��|�j��j��dP��|�j��j��dQ��|�j��dR��|�j��dS��|�j��j��dT��|�j��j��dU��|�j��j��dV��|�j��j��dW��|�j��j��dX��|�j��j��dY��|�j��j��dZ���i|�j��j��d[��|�j��j��d\��|�j��j��d]��|�j��j��d^��|�j��j��d_��|�j��j��d`��|�j��j��da��|�j��j��db��|�j��j��dc��|�j��j��dd��|�j��j��de��|�j��j��df��|�j��j��dg��|�j��j��dh��|�j��j��di��|�j��j��dj��|�j��j��dk���i|�j��j��dl��|�j��j��dm��|�j��j��dn��|�j��j��do��|�j��j��dp��|�j�j��dq��|�j��j��dr��|j��j��ds��|�j�j��dt��|�j�d%�du��|�j�dv��|�j�dw��|�j�dx��|�j�dy��|�j�dz��|�j�d{��|�j �d|���i|�j �d}��|�j �d~��|�j �d��|�j �d���|j�d���|�j�d���|�j�d���|�j�d���|�j�d���|�j�d���|�j�dB�d���|�jt�jf�d���|�jt�jf�d���|�jt�jf�d���|�jt�jf�d���|�j�dRd�d���d���|�j�d����i|�j�d���|�jt�jf�d���|�j�d�d���|�jt�jf�d���|�j t�jf�d���|�j!t�jf�d���|�j"t�jf�d���|�j#�d���|�j$�d���|j��d�d���|�j%�d���|�j&�d���|�j't�jf�d���|�j(t�jf�d���|�j)�d���|�j*�d���|�j+�d����i|�j,�d2d�d���d���|�j-�d���|�j.�d���|�j/t�jf�d���|�j0t�jf�d���|�j1d�d���d���|�j2�d���|�j3t�jf�d���|�j4t�jf�d���|�j5�d���|�j6�d���|�j7t�jf�d���|�j8�d���|�j9�d���|�j�d���|�j:�d���|�j;�d����i|�j<�d���|�j=�d���|�j>�dud�d���d���|�j�d���|�j?t�jf�d���|�j@�d���|�jA�d���|�jS�d���|�jB�d�d���|�j��d���|�jC�d���|�jD�d���|�jE�d���|�jF�d���|�jG�d���|�j��d���|�jH�d����i|�j��d„�|�jI�dÄ�|�jJ�d�dĄ�|�jK�dń�|�jL�d d�d���dƄ�|�jM�dDŽ�|�jN�dȄ�|�jO�dɄ�|�jP�dʄ�|�jQ�d˄�|�jR�d̄�|�jS�d-�d̈́�|�jT�d΄�|�jU�dτ�|�jV�dЄ�|�jW�dф�|�jX�d҄��i|�jY�dӄ�|�jZ�dv�dԄ�|�j �dՄ�|�j[�dք�|�j\t�jf�dׄ�|�j]�d؄�|�j �dm�dل�|�j^�dڄ�|�j_�dۄ�|�j`�d�d܄�|�ja�d݄�|�jb�dބ�|�j�d�d߄�|�jc�d���|�jd�d��|�je�d��|�jf�d���i|�jg�d��|�jo�d��|�jhddt�jf�d��|�ji�d�d�d��d��|�jj�d�d��|�jk�d��|�jl�d��|�jm�d��|�jn�d��|�j��d��|�j��dR�d��|�jo�d���|�jp�d��|�jq�d��|�jr�d��|�js�d��d��|�jt�d����tjP�ju�d�d��i�}t�jv�jw�jx}�t�||��r8�d%�d��|�t�||��<�d��|�t�|�d�|�����j�<i}�t���}|��|��D]�\}}|�j}|�j}�d�z�d�|�j}z�d�z�d�|�j}z�d�z�d�|�j}z�d�zg}|�j}��~�d���rP|�j}�t��d���d�}|����d�|z�d�z�d�|z�d�z�d�|z�d�zg��|D]1} �t�|| d��} �t| ��r | |vr | |vr||| <�2��|���|��|S(warReturn 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��dS�N��������input�outs r1�<lambda>z'get_testing_overrides.<locals>.<lambda>����2�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=�r?� output_sizes r1rAz'get_testing_overrides.<locals>.<lambda>����b�r4c��dSr;r=)�inputsrGs r1rAz'get_testing_overrides.<locals>.<lambda>����r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=�r?s r1rAz'get_testing_overrides.<locals>.<lambda>����R�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>����b�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>����R�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>����r�r4c��dSr;r=�r?�otherr@s r1rAz'get_testing_overrides.<locals>.<lambda>����"�r4�c��dSr;r=�r?�batch1�batch2�alpha�betar@s r1rAz'get_testing_overrides.<locals>.<lambda>����r�r4c��dSr;r=�r?�tensor1�tensor2�valuer@s r1rAz'get_testing_overrides.<locals>.<lambda>����"�r4c��dSr;r=rds r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=�r?�mat1�mat2rar`r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=)r?�mat�vecrar`r@s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=)r?�vec1�vec2rar`r@s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=��theta�size� align_cornerss r1rAz'get_testing_overrides.<locals>.<lambda>�rqr4c��dSr;r=�r?�dims r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4��h㈵��>�:�0�yE>Fc��dSr;r=)r?rY�trol�atol� equal_nans r1rAz'get_testing_overrides.<locals>.<lambda>����VX�r4c��dSr;r=�r?�p�train�inplaces r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=�r?r}�keepdimr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rqr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=)r?�msgs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>����b�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=��tensorss r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4rTc��dSr;r=)r?� kernel_sizer�padding� ceil_mode�count_include_pads r1rAz'get_testing_overrides.<locals>.<lambda>����vx�r4c��dSr;r=r]s r1rAz'get_testing_overrides.<locals>.<lambda>����PR�r4c ��dSr;r=) r?�weight�bias� running_mean� running_var�training�momentum�eps� cudnn_enableds r1rAz'get_testing_overrides.<locals>.<lambda>����y{�r4c��dSr;r=)�grad_outr?�mean�invstdr��sum_dy� sum_dy_xmu� count_tensors r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r?r�r�r��input_g�weight_g�bias_gs r1rAz'get_testing_overrides.<locals>.<lambda>����su�r4c��dSr;r=)r?r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c��dSr;r=�r?r�r�r�r�r�r��counts r1rAz'get_testing_overrides.<locals>.<lambda>����tv�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�� ��AC�r4c��dSr;r=�r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>����2�r4c��dSr;r=)r?r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����Z\�r4c��dSr;r=)r?� generatorr@s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=��input1�input2r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����R�r4r�c��dSr;r=�r?�targetr�� size_average�reduce� reduction� pos_weights r1rAz'get_testing_overrides.<locals>.<lambda>�s ��TV�r4c��dSr;r=)r?�weights� minlengths r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r��probr�s r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>����"�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>����"�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=�r?rmr@s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=��selfrys r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r?� boundaries� out_int32�rightr@s r1rAz'get_testing_overrides.<locals>.<lambda>����[]�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=�r�r}r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>����r�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4�@�#use_mm_for_euclid_dist_if_necessaryc��dSr;r=)�x1�x2r�� compute_modes r1rAz'get_testing_overrides.<locals>.<lambda>����_a�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4��?c��dSr;r=�r?r`r�s r1rAz'get_testing_overrides.<locals>.<lambda>����2�r4)r@c��dSr;r=)r@�matricess r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=�r?�groupss r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?�upperr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=�r?� check_errorsr@s r1rAz'get_testing_overrides.<locals>.<lambda>����b�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>����R�r4c��dSr;r=)r�r�rr@s r1rAz'get_testing_overrides.<locals>.<lambda>����B�r4c��dSr;r=)r?�numel�n_bins�ratio� bit_widths r1rAz'get_testing_overrides.<locals>.<lambda>����WY�r4c��dSr;r=�r?�chunksr}s r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=�r?�min�maxr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r,s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=)r?r-r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r.r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?� correction�fweights�aweightss r1rAz'get_testing_overrides.<locals>.<lambda>����R�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>����2�r4�c��dSr;r=)r?�r�with_replacements r1rAz'get_testing_overrides.<locals>.<lambda>����r�r4c��dSr;r=)�real�imags r1rAz'get_testing_overrides.<locals>.<lambda>���"�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>rr4c��dSr;r=)�abs�angs r1rAz'get_testing_overrides.<locals>.<lambda>���b�r4c��dSr;r=)r?�ords r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>���R�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r?�padrgs r1rAz'get_testing_overrides.<locals>.<lambda>rr4c��dSr;r=�r?r�r�rr��dilationrs r1rAz'get_testing_overrides.<locals>.<lambda> ���bd�r4c��dSr;r=rTs r1rAz'get_testing_overrides.<locals>.<lambda> rVr4c��dSr;r=rTs r1rAz'get_testing_overrides.<locals>.<lambda> rVr4c ��dSr;r=) r?r�r�rr�rU� transposed� output_addingrs r1rAz'get_testing_overrides.<locals>.<lambda> s��uw�r4c��dSr;r=)r?r�r�rRs r1rAz'get_testing_overrides.<locals>.<lambda> rr4c��dSr;r=�r?r�r�rr��output_paddingrrUs r1rAz'get_testing_overrides.<locals>.<lambda>� ��A�r4c��dSr;r=r^s r1rAz'get_testing_overrides.<locals>.<lambda>r`r4c��dSr;r=r^s r1rAz'get_testing_overrides.<locals>.<lambda>r`r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>rIr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=�r�r�r��marginr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=)rrr}r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=�r?rYr}r@s r1rAz'get_testing_overrides.<locals>.<lambda>rHr4r<c��dSr;r=rls r1rAz'get_testing_overrides.<locals>.<lambda>���2�r4c��dSr;r=�� log_probs�targets� input_lengths�target_lengths�blankr�� zero_infinitys r1rAz'get_testing_overrides.<locals>.<lambda>s ��.0�r4c��dSr;r=�r?r}r@s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rxs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=�r?r}r@ras r1rAz'get_testing_overrides.<locals>.<lambda>r0r4c��dSr;r=r{s r1rAz'get_testing_overrides.<locals>.<lambda>rKr4c��dSr;r=��y�xr}s r1rAz'get_testing_overrides.<locals>.<lambda>rHr4c��dSr;r=rxs r1rAz'get_testing_overrides.<locals>.<lambda> r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>!rVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>"����r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>#����r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>$r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>%r�r4c��dSr;r=�r?�diagonalr@s r1rAz'get_testing_overrides.<locals>.<lambda>&rr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>'rHr4c��dSr;r=)r?�offsets r1rAz'get_testing_overrides.<locals>.<lambda>(rDr4c��dSr;r=)r?�nr}�prepend�appendr@s r1rAz'get_testing_overrides.<locals>.<lambda>)���TV�r4c��dSr;r=�r?r��dim1�dim2s r1rAz'get_testing_overrides.<locals>.<lambda>*r0r4�����c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>+rur4c��dSr;r=)r?�srcr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>,r9r4c��dSr;r=)r�r�ryr�storage_offsets r1rAz'get_testing_overrides.<locals>.<lambda>-r'r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>.rVr4c��dSr;r=)r?rYr�s r1rAz'get_testing_overrides.<locals>.<lambda>/rRr4c��dSr;r=�r?rY� rounding_moder@s r1rAz'get_testing_overrides.<locals>.<lambda>0���b�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>1rur4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>2rZr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>3rHr4c��dSr;r=�r?rms r1rAz'get_testing_overrides.<locals>.<lambda>4r�r4c��dSr;r=)rlrms r1rAz'get_testing_overrides.<locals>.<lambda>5���r�r4c��dSr;r=�r?�indices_or_sectionss r1rAz'get_testing_overrides.<locals>.<lambda>6r�r4c��dSr;r=r4s r1rAz'get_testing_overrides.<locals>.<lambda>7rDr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>8rZr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>9r�r4�Lc��dSr;r=�r?�UPLOr@s r1rAz'get_testing_overrides.<locals>.<lambda>:r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>;r�r4c��dSr;r=)�equation�operandss r1rAz'get_testing_overrides.<locals>.<lambda><rZr4c��dSr;r=�r?r�� padding_idx�max_norm� norm_type�scale_grad_by_freqr�s r1rAz'get_testing_overrides.<locals>.<lambda>=s ��(*�r4c ��dSr;r=) r?r��offsetsr�r�r��moder��per_sample_weightsr�s r1rAz'get_testing_overrides.<locals>.<lambda>?s ��df�r4c��dSr;r=�r?raror`� requires_grads r1rAz'get_testing_overrides.<locals>.<lambda>A���ce�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>Br�r4c��dSr;r=�r?rYs r1rAz'get_testing_overrides.<locals>.<lambda>CrDr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>DrBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>ErMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>FrRr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>GrBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>HrMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>IrTr4c��dSr;r=)r?�scale� zero_point�axis� quant_min� quant_maxs r1rAz'get_testing_overrides.<locals>.<lambda>J���mo�r4c��dSr;r=)r?r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>K���fh�r4c ��dSr;r=) r�� observer_on� fake_quant_on�averaging_const� running_min� running_maxr�r�r�r��ch_axis�per_row_fake_quant�symmetric_quants r1rAz'get_testing_overrides.<locals>.<lambda>Ls ��ac�r4c��dSr;r=�r?� packed_weightr�s r1rAz'get_testing_overrides.<locals>.<lambda>Or!r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>P���\^�r4c��dSr;r=�r?r��packed� col_offsets� weight_scale�weight_zero_pointr�s r1rAz'get_testing_overrides.<locals>.<lambda>Q���{}�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>R� ��SU�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>Trr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>Ur�r4c��dSr;r=)r?�a�bs r1rAz'get_testing_overrides.<locals>.<lambda>Vr0r4c��dSr;r=�r?r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>Wr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Xrr4c��dSr;r=�r?r�r}�norms r1rAz'get_testing_overrides.<locals>.<lambda>Yr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Zr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>[r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>\r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>]r�r4�r�r<c��dSr;r=�r?�sr}r�s r1rAz'get_testing_overrides.<locals>.<lambda>^rqr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>_rur4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>`r�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>arnr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>brnr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>cr�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>dr�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>err4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>fr@r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>grqr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>hrqr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>irur4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>jr�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>krNr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>lr0r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>mrBr4c��dSr;r=)r?� start_dim�end_dims r1rAz'get_testing_overrides.<locals>.<lambda>nrHr4c��dSr;r=�r?�dimss r1rAz'get_testing_overrides.<locals>.<lambda>or�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>pr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>qr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>rrbr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>srTr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>tr�r4c��dSr;r=�r?�exponentr@s r1rAz'get_testing_overrides.<locals>.<lambda>ur�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>vr�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>wrMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>xrTr4c��dSr;r=)r?� fill_valuer@raror`r�s r1rAz'get_testing_overrides.<locals>.<lambda>ys ��BD�r4c��dSr;r=)r?r�� dep_tokens r1rAz'get_testing_overrides.<locals>.<lambda>zr�r4c��dSr;r=)�LU_data� LU_pivots� unpack_data� unpack_pivotss r1rAz'get_testing_overrides.<locals>.<lambda>{r�r4c��dSr;r=)r?r}�indexr@� sparse_grads r1rAz'get_testing_overrides.<locals>.<lambda>|r9r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>}rZr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>~r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?rtr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r6s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�spacingr}� edge_orders r1rAz'get_testing_overrides.<locals>.<lambda>�r!r4c��dSr;r=�r?�grid�interpolation_mode� padding_moderzs r1rAz'get_testing_overrides.<locals>.<lambda>����ac�r4c��dSr;r=r<s r1rAz'get_testing_overrides.<locals>.<lambda>����df�r4c��dSr;r=r<s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=)r?� num_groupsr�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����km�r4c ��dSr;r=� r?�hx�params� has_biases� num_layers�dropoutr�� bidirectional� batch_firsts r1rAz'get_testing_overrides.<locals>.<lambda>����qs�r4c��dSr;r=�r?rI�w_ih�w_hh�b_ih�b_hhs r1rAz'get_testing_overrides.<locals>.<lambda>�r!r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4��?c��dSr;r=�r?�lambds r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�valuesr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?r�rgr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����xz�r4�dc��dSr;r=)r?�binsr-r.r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?rdr-r.r��densityr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rFr4c��dSr;r=)r?rdr�r�rfs r1rAz'get_testing_overrides.<locals>.<lambda>�r'r4c��dSr;r=�r?�taus r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)rlrmr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r4s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=�r?r}r-�sources r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rss r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�indicesr^� accumulates r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=)r?r}r-r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r0r4c��dSr;r=)r?r}r-rgs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?r}r-rtr�� include_inputs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r[s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)�e�te� assume_unique�inverts r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=r~s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r~s r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c ��dSr;r=) r?r�r�r�r��use_input_statsr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�s ��-/�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rIr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rur4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?rY�rtolr�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����UW�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c ��dSr;r=) r?�n_fft� hop_length� win_length�window�center� normalized�onesided�length�return_complexs r1rAz'get_testing_overrides.<locals>.<lambda>�s ��Z\�r4c��dSr;r=�r?r�r�r�r�� log_targets r1rAz'get_testing_overrides.<locals>.<lambda>�s��pr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�����r4c��dSr;r=)r?�kr}r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r!r4c��dSr;r=)r?� hermitianrr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r@r4c��dSr;r=)r?r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=)�LD�pivots�Br�r@s r1rAz'get_testing_overrides.<locals>.<lambda>����QS�r4c��dSr;r=)r?�normalized_shaper�r��espr�s r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�endr�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=)r?r�r��Xr��iK�niter�tol�largest�method�tracker� ortho_iparams� ortho_fparams� ortho_bparamss r1rAz'get_testing_overrides.<locals>.<lambda>�s ��RT�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=�r?r}ras r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�rr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=)r?�namesr�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r@r4c ��dSr;r=) �data� batch_sizesrIrJrKrLrMr�rNs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rRs r1rAz'get_testing_overrides.<locals>.<lambda>�r9r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)�A�pivot� get_infosr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rnr4c��dSr;r=)r�r(r)r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r0r4c��dSr;r=rfs r1rAz'get_testing_overrides.<locals>.<lambda>�r`r4c��dSr;r=)r?�maskrgs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�rts r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=�r?r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�rr@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)�LUr�r��left�adjointr@s r1rAz'get_testing_overrides.<locals>.<lambda>����Y[�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=�r?r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rHr4c��dSr;r=)r?r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>����2�r4c��dSr;r=r4s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?r�rr�rUr�s r1rAz'get_testing_overrides.<locals>.<lambda>����jl�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=�r?r�rr�rU�return_indicesr�s r1rAz'get_testing_overrides.<locals>.<lambda>�� ��OQ�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=)r?r}r�rar@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r0s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�r�r�r�r��exponential_average_factor�epsilons r1rAz'get_testing_overrides.<locals>.<lambda>�� ��GI�r4c ��dSr;r=� r?r�r�r�rrUr� benchmark� deterministics r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c ��dSr;r=) r?r��zr`r�rr�rUrs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rTs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c ��dSr;r=) r?r�r�r�r_rrUrrrs r1rAz'get_testing_overrides.<locals>.<lambda>rr4c ��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>s ��<>�r4c��dSr;r=)r?r��weight_stride0rI�cxr�� hidden_sizerLrOrMr�rNr�� dropout_states r1rAz'get_testing_overrides.<locals>.<lambda>s ��VX�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda> rDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda> rnr4c��dSr;r=�r?rt� destinations r1rAz'get_testing_overrides.<locals>.<lambda> r�r4c��dSr;r=r&s r1rAz'get_testing_overrides.<locals>.<lambda> rr4c��dSr;r=)r?� descendingr@s r1rAz'get_testing_overrides.<locals>.<lambda>rKr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>rZr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>rr4c��dSr;r=)r?� num_samples� replacementr@s r1rAz'get_testing_overrides.<locals>.<lambda>���SU�r4c��dSr;r=)r?rpr@s r1rAz'get_testing_overrides.<locals>.<lambda>rVr4c��dSr;r=�r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r?r}�startr�s r1rAz'get_testing_overrides.<locals>.<lambda>rr4�c��dSr;r=)r?�nan�posinf�neginfr@s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r?r�r�r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>rPr4c��dSr;r=)r?r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>���]_�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=�r?r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=)r?r�r��N�C�HxW�groupr�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r?r�r}r�ras r1rAz'get_testing_overrides.<locals>.<lambda>r0r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>rr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda> rDr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>!rr4c��dSr;r=rFs r1rAz'get_testing_overrides.<locals>.<lambda>"r!r4c��dSr;r=rFs r1rAz'get_testing_overrides.<locals>.<lambda>#r!r4c��dSr;r=�r?rGrs r1rAz'get_testing_overrides.<locals>.<lambda>$rVr4c��dSr;r=rPs r1rAz'get_testing_overrides.<locals>.<lambda>%���oq�r4c��dSr;r=rPs r1rAz'get_testing_overrides.<locals>.<lambda>&rVr4c��dSr;r=rPs r1rAz'get_testing_overrides.<locals>.<lambda>'rRr4c��dSr;r=rPs r1rAz'get_testing_overrides.<locals>.<lambda>(rVr4c��dSr;r=rPs r1rAz'get_testing_overrides.<locals>.<lambda>)rRr4c��dSr;r=rws r1rAz'get_testing_overrides.<locals>.<lambda>*r�r4c��dSr;r=�r?r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>+r r4c��dSr;r=�r?r�rr�r�r��divisor_overrides r1rAz'get_testing_overrides.<locals>.<lambda>,� ��XZ�r4c��dSr;r=r[s r1rAz'get_testing_overrides.<locals>.<lambda>.r]r4皙�����?c��dSr;r=)r?r�r�r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>0s ��BD�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>2r�r4c��dSr;r=�r?r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>3� ��EG�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>5s ��oq�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>7rhr4c��dSr;r=rfs r1rAz'get_testing_overrides.<locals>.<lambda>8r�r4����c��dSr;r=)r?r�r�r�� ignore_indexr�r��label_smoothings r1rAz'get_testing_overrides.<locals>.<lambda>:� ��`b�r4c��dSr;r=rps r1rAz'get_testing_overrides.<locals>.<lambda><s ��NP�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>>���XZ�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>?r�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>@r�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>Ar�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>Brur4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>C� ��PR�r4c ��dSr;r=) r?r�r�r�r�r�r�r�r��include_last_offsetr�s r1rAz'get_testing_overrides.<locals>.<lambda>Es ��Y[�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>H���gi�r4c��dSr;r=)r?rGr�rUr�rs r1rAz'get_testing_overrides.<locals>.<lambda>IrFr4c��dSr;r=�r?r�rG� output_ratior�_random_sampless r1rAz'get_testing_overrides.<locals>.<lambda>Jrlr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>M� ��"$�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>Orlr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>Rr�r4�����ư>c��dSr;r=)r?r��varr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>Tr�r4�nonec��dSr;r=)r?� approximates r1rAz'get_testing_overrides.<locals>.<lambda>Ur�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>Vrr4�bilinearr�c��dSr;r=)r?r=r�r?rzs r1rAz'get_testing_overrides.<locals>.<lambda>Wrar4c��dSr;r=)r?rEr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>X���eg�r4绽���|�=c��dSr;r=)�logitsrj�hardr�r}s r1rAz'get_testing_overrides.<locals>.<lambda>Yr@r4c��dSr;r=r[s r1rAz'get_testing_overrides.<locals>.<lambda>Zr�r4��c��dSr;r=)r?�min_val�max_valr�s r1rAz'get_testing_overrides.<locals>.<lambda>[r�r4c��dSr;r=r`s r1rAz'get_testing_overrides.<locals>.<lambda>\rdr4c��dSr;r=)r?r�r�r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>^s ��[]�r4�nearestc��dSr;r=)r?ry� scale_factorr�rz�recompute_scale_factor� antialiass r1rAz'get_testing_overrides.<locals>.<lambda>`r]r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>br`r4c��dSr;r=�r?r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>cr�r4c��dSr;r=r@s r1rAz'get_testing_overrides.<locals>.<lambda>drFr4�{�G�z�?c��dSr;r=)r?�negative_sloper�s r1rAz'get_testing_overrides.<locals>.<lambda>er�r4c��dSr;r=)r?r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>frr4�-C��6?��?c��dSr;r=)r?ryr`rar�s r1rAz'get_testing_overrides.<locals>.<lambda>gr�r4�c��dSr;r=�r?r}� _stacklevelras r1rAz'get_testing_overrides.<locals>.<lambda>hr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>ir�r4c��dSr;r=�r?r�r�rr�s r1rAz'get_testing_overrides.<locals>.<lambda>jrFr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>krFr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>lrFr4c��dSr;r=rfs r1rAz'get_testing_overrides.<locals>.<lambda>ms ��QS�r4c��dSr;r=�r?r�rr�rUr�rs r1rAz'get_testing_overrides.<locals>.<lambda>orur4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>q� ��]_�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>srur4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>ur�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>wrur4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>yr�r4c��dSr;r=�r?rwr�rr�rGs r1rAz'get_testing_overrides.<locals>.<lambda>{rar4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>|rar4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>}rar4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>~���np�r4c��dSr;r=)�query�keyrg�embed_dim_to_check� num_heads�in_proj_weight� in_proj_bias�bias_k�bias_v� add_zero_attn� dropout_p�out_proj_weight� out_proj_biasr��key_padding_mask� need_weights� attn_mask�use_separate_proj_weight� q_proj_weight� k_proj_weight� v_proj_weight�static_k�static_v�average_attn_weights� is_causals r1rAz'get_testing_overrides.<locals>.<lambda>�s��km�r4c��dSr;r=)r?r�r�rgr�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rcs r1rAz'get_testing_overrides.<locals>.<lambda>�� ��Y[�r4c��dSr;r=)r?r�r�r�rjr�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�s ��FH�r4��-���q=c��dSr;r=)r?r�r}r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r[� num_classess r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4r�c��dSr;r=)r?rRr�rgs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=�rrr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r r4c��dSr;r=)r?r�� log_inputr�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r0r4c��dSr;r=�r?r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r=r4��?�UUUUUU�?c��dSr;r=�r?�lowerrr�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�s��wy�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=)r�r�rgr�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�r�r�r�ras r1rAz'get_testing_overrides.<locals>.<lambda>�s��}�r4c��dSr;r=)r?r�r��deltas r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�ror4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�ror4�c��dSr;r=)r?ra� thresholds r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=r[s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?r�rgr�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c ��dSr;r=� �anchor�positive�negativergr�r��swapr�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�s ��pr�r4)�distance_functionrgr r�c��dSr;r=)rrr r rgr r�s r1rAz'get_testing_overrides.<locals>.<lambda>�s ��^`�r4c��dSr;r=)r?r�rUr�rs r1rAz'get_testing_overrides.<locals>.<lambda>�s��`b�r4c��dSr;r=)r[r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r[r��stdr�s r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c��dSr;r=)r[�vals r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4�fan_in� leaky_reluc��dSr;r=)r[r�r�� nonlinearityr�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�as_tuples r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4)r$c��dSr;r=)r?ryr$s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rIr4�froc��dSr;r=�r?r�r}r�r@ras r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r?rKr}r�r@ras r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)�v�powr}s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c��dSr;r=ris r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r��input3r�� transposes r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r�r}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�qr�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r@r4c��dSr;r=r3s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4�V瞯�<c��dSr;r=)r?�rconds r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r3r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r@r4c��dSr;r=)r?�upscale_factors r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?�downscale_factors r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=)r?r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=)r?r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r'r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=)r?ras r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=)r?r-rtrxs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=)r?�somer@s r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4�reducedc��dSr;r=)r?r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4�linearc��dSr;r=�r?r/r}r�� interpolationr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�ryr4c��dSr;r=)r?�scales� zero_pointsr�ras r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�r�ras r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?ra� reduce_ranges r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?r�r�r�r�r�� output_scale�output_zero_points r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=�r?rIrSrTrUrV� packed_ih� packed_hh�col_offsets_ih�col_offsets_hh�scale_ih�scale_hh� zero_point_ih� zero_point_hhs r1rAz'get_testing_overrides.<locals>.<lambda>�s ��eg�r4c��dSr;r=rZs r1rAz'get_testing_overrides.<locals>.<lambda>�s ��fh�r4)r)r[c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rdr4)rr)r[r[c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4)rrr)r[r[r[c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�s ��JL�r4c��dSr;r=rZs r1rAz'get_testing_overrides.<locals>.<lambda>�� ��jl�r4c��dSr;r=rZs r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r?�highraror`r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rls r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?r�r}�maxnormr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�shapes r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c ��dSr;r=rHs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rRs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c ��dSr;r=rHs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rRs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�shiftsrs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4�rr[c��dSr;r=)r?r�rs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r4s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r�r��compressed_indices_dtypes r1rAz'get_testing_overrides.<locals>.<lambda>rr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>s��^`�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rTr4c��dSr;r=)r?rYr`s r1rAz'get_testing_overrides.<locals>.<lambda>rZr4c��dSr;r=rks r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=�r?r}r-r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r?r}r-r�r�� include_selfs r1rAz'get_testing_overrides.<locals>.<lambda>ror4c��dSr;r=)�sorted_sequencer?r�r�r@s r1rAz'get_testing_overrides.<locals>.<lambda> r�r4r.c��dSr;r=)r�r��lengthsrwr�r��unsafes r1rAz'get_testing_overrides.<locals>.<lambda> s��z|�r4c��dSr;r=)r?r}r-s r1rAz'get_testing_overrides.<locals>.<lambda> rDr4c��dSr;r=)r?r�r}r-s r1rAz'get_testing_overrides.<locals>.<lambda> r�r4c��dSr;r=�r?r�r}r5r��steps r1rAz'get_testing_overrides.<locals>.<lambda> r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rVr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>rPr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r�r�r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>rHr4c��dSr;r=)r�r�r�rr@s r1rAz'get_testing_overrides.<locals>.<lambda>r�r4)�stabler@c��dSr;r=)r?r}r*r�r@s r1rAz'get_testing_overrides.<locals>.<lambda>r'r4c��dSr;r=�r[�split_size_or_sectionsr}s r1rAz'get_testing_overrides.<locals>.<lambda>rnr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda> rr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>!rMr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>"rRr4c��dSr;r=rxs r1rAz'get_testing_overrides.<locals>.<lambda>#r�r4c��dSr;r=rks r1rAz'get_testing_overrides.<locals>.<lambda>$r9r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>%r�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>&rBr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>'rDr4�reflectc ��dSr;r=) r?r�r�r�r�r��pad_moder�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>(s ��_a�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>*rZr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>+rr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>,rBr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>-r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>.rPr4c��dSr;r=�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>/r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>0r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>1rPr4c��dSr;r=)r�r��cs r1rAz'get_testing_overrides.<locals>.<lambda>2r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>3r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>4rIr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>5r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>6rIr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>7r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>8rIr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>9r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>:r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>;r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda><r�r4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>=rRr4c��dSr;r=)r?rH� compute_uvr@s r1rAz'get_testing_overrides.<locals>.<lambda>>r@r4�c��dSr;r=)r?r/r��Ms r1rAz'get_testing_overrides.<locals>.<lambda>?rKr4c��dSr;r=)r?� full_matricesr@s r1rAz'get_testing_overrides.<locals>.<lambda>@r�r4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>Ar�r4c��dSr;r=�r?�dim0r�s r1rAz'get_testing_overrides.<locals>.<lambda>BrZr4c��dSr;r=)r?�axis0�axis1s r1rAz'get_testing_overrides.<locals>.<lambda>Cr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>DrTr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>ErVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>FrVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>GrVr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>HrVr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Irur4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Jrur4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Krur4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Lrur4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>MrTr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>NrDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>Or�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>PrDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>QrBr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>RrMr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>SrDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>TrBr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>UrBr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>VrKr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>Wr0r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>XrTr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Yr@r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Zrqr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>[r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>\r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>]r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>^r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>_rqr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>`rqr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>arBr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>brRr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>cr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>drBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>er9r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>frr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>grr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>hrr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>irr4c��dSr;r=r3s r1rAz'get_testing_overrides.<locals>.<lambda>jrNr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>krDr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>lrBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>mr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>nr�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>orBr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>prKr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>qrKr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>rr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>sr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>tr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>ur�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>vrDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>wrHr4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>xr�r4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>yrHr4c��dSr;r=rXs r1rAz'get_testing_overrides.<locals>.<lambda>zr�r4c��dSr;r=)r�rYr@s r1rAz'get_testing_overrides.<locals>.<lambda>{r�r4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>|s��r�r4c��dSr;r=)r?r-s r1rAz'get_testing_overrides.<locals>.<lambda>}r�r4c��dSr;r=)r?rwr}r@s r1rAz'get_testing_overrides.<locals>.<lambda>~rur4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=)r��inds r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r�rs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r�rr@s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r?r�r}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�rqr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�r}r*r@s r1rAz'get_testing_overrides.<locals>.<lambda>�rur4c��dSr;r=rOs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r~s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r~s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�rr*� unitriangulars r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r�rr�r+s r1rAz'get_testing_overrides.<locals>.<lambda>�r r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c ��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>�s ��VX�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r>s r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=)r?r}�sizesr�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?�sorted�return_inverse� return_countsr}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r?r7r8r}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)rwr|s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r)s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rhr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rxs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�rBs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=r|s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r4s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=)� conditionr�rs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r��levels r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)�primal�tangentrHs r1rAz'get_testing_overrides.<locals>.<lambda>�rHr4c��dSr;r=�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=)r�ryrr�s r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4)�implicitc��dSr;r=)r�ryrTs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r}r5r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�rs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=�r�ryrs r1rAz'get_testing_overrides.<locals>.<lambda>�rHr4c��dSr;r=)r�r}r-s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r}r5r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r!r4c��dSr;r=)r�� split_sizer}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�� split_sizesr}s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=r-s r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c��dSr;r=)r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=r-s r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=r-s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=�r�ras r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=�r�� dimensionryr�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r�rYs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=ros r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rIr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=)r��arrays r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r��idxs r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=)r��memos r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rIr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=)r�� format_specs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=)r��protos r1rAz'get_testing_overrides.<locals>.<lambda>�rZr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4)�tensor_contentsc��dSr;r=)r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rKr4c��dSr;r=)r�r�r#s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r��ds r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rRr4c��dSr;r=)r�� cuda_enabled� cpu_enabled� cuda_dtype� cpu_dtypes r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rZr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rZr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rRr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>rHr4c��dSr;r=)r�ra� non_blockingr0s r1rAz'get_testing_overrides.<locals>.<lambda>r!r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>r;r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>r;r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda> rIr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>!rIr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>"���"�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>#rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>$r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>%rDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>&r�r4c��dSr;r=)r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>'rr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>(rPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>)rPr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>*rTr4c��dSr;r=)r��order� ellipsis_idxs r1rAz'get_testing_overrides.<locals>.<lambda>+rr4c��dSr;r=)r��callables r1rAz'get_testing_overrides.<locals>.<lambda>,rRr4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>-r�r4c��dSr;r=rYs r1rAz'get_testing_overrides.<locals>.<lambda>.rr4c��dSr;r=)r��gradient� retain_graph� create_graphrJs r1rAz'get_testing_overrides.<locals>.<lambda>/s��ik�r4c��dSr;r=�r�rhs r1rAz'get_testing_overrides.<locals>.<lambda>0r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>1r@r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>2r@r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>3r@r4)r�c��dSr;r=)r��median�sigmar�s r1rAz'get_testing_overrides.<locals>.<lambda>4r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>5rIr4c��dSr;r=)r�� coalesceds r1rAz'get_testing_overrides.<locals>.<lambda>6r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>7rbr4c��dSr;r=)r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>8r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>9r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>:r@r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>;r�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda><r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>=rIr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>>r�r4c��dSr;r=)r�r�r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>?r9r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>@r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Ar�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Brur4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Crhr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>DrDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>ErDr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>FrRr4c��dSr;r=)r�r\r�s r1rAz'get_testing_overrides.<locals>.<lambda>Gr�r4c��dSr;r=�r�rgs r1rAz'get_testing_overrides.<locals>.<lambda>HrDr4c��dSr;r=rs r1rAz'get_testing_overrides.<locals>.<lambda>Ir�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Jrqr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Krur4c��dSr;r=)r�r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>LrHr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Mr�r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Nr@r4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Orqr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Pr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>QrPr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>Rr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>SrDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>TrBr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>UrDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Vr�r4c��dSr;r=)r�r[s r1rAz'get_testing_overrides.<locals>.<lambda>WrVr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Xr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>Yr�r4c��dSr;r=)r�r�rr�s r1rAz'get_testing_overrides.<locals>.<lambda>Zr�r4c��dSr;r=r-s r1rAz'get_testing_overrides.<locals>.<lambda>[rRr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>\r@r4c��dSr;r=)r�r[r�s r1rAz'get_testing_overrides.<locals>.<lambda>]r�r4c��dSr;r=)r�r�rr�s r1rAz'get_testing_overrides.<locals>.<lambda>^r�r4c��dSr;r=)r�rms r1rAz'get_testing_overrides.<locals>.<lambda>_rIr4c��dSr;r=)r�rY�assigns r1rAz'get_testing_overrides.<locals>.<lambda>`rHr4c��dSr;r=)r�rrr5r�s r1rAz'get_testing_overrides.<locals>.<lambda>arnr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>br�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>crIr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>dr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>er�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>fr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>grPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>hr;r4c��dSr;r=r-s r1rAz'get_testing_overrides.<locals>.<lambda>irDr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>jr�r4c��dSr;r=)r�rwr[rxs r1rAz'get_testing_overrides.<locals>.<lambda>krr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>lrPr4c��dSr;r=)r��from_�tor�s r1rAz'get_testing_overrides.<locals>.<lambda>mrhr4c��dSr;r=�r��streams r1rAz'get_testing_overrides.<locals>.<lambda>nr�r4c��dSr;r=)r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>or�r4c��dSr;r=�r��hooks r1rAz'get_testing_overrides.<locals>.<lambda>pr�r4c��dSr;r=r<s r1rAz'get_testing_overrides.<locals>.<lambda>qr�r4c��dSr;r=)r��names r1rAz'get_testing_overrides.<locals>.<lambda>rrDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>srBr4c��dSr;r=)r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>tr0r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>urVr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>vrBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>wrBr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>xrRr4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>yr�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>zr�r4c��dSr;r=)r�rtr�ryrs r1rAz'get_testing_overrides.<locals>.<lambda>{r�r4c��dSr;r=)r�r�r}r-s r1rAz'get_testing_overrides.<locals>.<lambda>|r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>}rBr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>~rqr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>r�r4c��dSr;r=)r�r�r}r5r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r�r��accumulate_matchess r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=�r��size1�size2� dense_dims r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rTs r1rAz'get_testing_overrides.<locals>.<lambda>�rbr4c��dSr;r=)r�rlrmrar`r@s r1rAz'get_testing_overrides.<locals>.<lambda>�r9r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rTr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rDr4c��dSr;r=r�s r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r��repss r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�rar��copyrhs r1rAz'get_testing_overrides.<locals>.<lambda>�s��ln�r4)� masked_gradc��dSr;r=�r�rarcs r1rAz'get_testing_overrides.<locals>.<lambda>�r@r4c��dSr;r=res r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rqs r1rAz'get_testing_overrides.<locals>.<lambda>�rr4c��dSr;r=)r�r5r6s r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r|s r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=rus r1rAz'get_testing_overrides.<locals>.<lambda>�rMr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r;r4c��dSr;r=r8s r1rAz'get_testing_overrides.<locals>.<lambda>�rNr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�rVr4c��dSr;r=)r�r�r��drivers r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4c��dSr;r=)r�r`r�r0s r1rAz'get_testing_overrides.<locals>.<lambda>�rPr4c��dSr;r=rMs r1rAz'get_testing_overrides.<locals>.<lambda>�r�r4�is_�_�__�__i�__r�bitwise_�N)r[r[N)r[N)r~rF)F)NFN)NrFT)NN)NNNr�N)Nr)FFN)rN)rr)r F)FN)NNN)r[NN)r<F)Nr[rr[r[)Nr[rrr[r[)rNNr�)r[r)r<N)rr�Fr;)r[r<NNN)rrr[)rr�r<)r<)r�N)NNrFF)Nr<Fr�FNN)NNNF)FF)Nr<N)Nr�N)rr<)TT)NF)NNr[)NNr~T)rY)r NNr�)rbrrN)rbNNNFN)NNF)T)NNNTFNNF)NNr�F) NNNNNNNNNNNNN)TFN)TN)Nrr[F)Nrr[FF)NFNN)r<FN)r6NNN)NNr~)r<NFN)rYFF)NrFTN)NNFr_r~)NNNr�)NNrhNr�r6)rYTF) NNr<Fr�FNFN)r[rr[)NNFN)Fr�r�)r�)r<)r�r�N)r[Fr�r<)r�r F)NNNNTr_r~)NNr�NNF)NNr�)r�F)r�r�r )Nr�N)NrN) TNTNFNNNNNNF)r[r NNNr�)NNrhNr�)r<r[r�N)r�r)rr�F)TFNrNr�)Nr�)r�r�FF)Nr6)NNr�r )r�r )r[r�)r r<r�FNNr�)r6r N)rrrN)rNFNN)NNFNN)r<NFNN)rr�FNN)r<r)TF)NTr<)r1)r1F)rIN)NFrKN)r[r�)r.NNNrF)rNNr[)r<F)NNNTr�FTN)TTN)r�r<N)r<N)TFF)TFFN)r[r<)NrNN(�r!r6rG�absolute�adaptive_avg_pool1d�adaptive_max_pool1d�acosr��arccos�acosh�arccosh�add�addbmm�addcdiv�addcmul�addmm�addmv�addr�affine_grid_generator�all�allclose� alpha_dropout�amax�amin�aminmax�angle�any�argmax�argmin�argsort�asin� _assert_async�arcsin�asinh�arcsinh�atan�arctan�atan2�arctan2�atanh�arctanh� atleast_1d� atleast_2d� atleast_3d� avg_pool1d�baddbmm� batch_norm�batch_norm_backward_elemt�batch_norm_backward_reduce�batch_norm_elemt�batch_norm_gather_stats�#batch_norm_gather_stats_with_counts�batch_norm_stats�batch_norm_update_stats� bernoullir�� binary_cross_entropy_with_logits�bincount�binomial� bitwise_and� bitwise_not� bitwise_or� bitwise_xor�bitwise_left_shift�bitwise_right_shift� block_diag�bmm�broadcast_tensors� broadcast_to� bucketize�cartesian_prod�cat�concat� concatenate�cdist�ceil�celu� chain_matmul�channel_shuffle�cholesky�linalg� cholesky_ex�cholesky_inverse�cholesky_solve�choose_qparams_optimized�chunk�clamp�clip� clamp_min� clamp_max� column_stack�cov�clone� combinations�complex�copysign�polarr��conj� conj_physical� resolve_conj� resolve_neg�constant_pad_nd�conv1d�conv2d�conv3d� convolution�conv_tbc�conv_transpose1d�conv_transpose2d�conv_transpose3d�corrcoef�cos�cosine_embedding_loss�cosh�cosine_similarity� count_nonzero�cross�ctc_loss�cummax�cummin�cumprod�cumsum�cumulative_trapezoid� logcumsumexp�deg2rad� dequantize�det�detach�diag� diag_embed�diagflat�diffr��diagonal_scatter�as_strided_scatter�digamma�dist�div�divide�dotrM�dsmm�hsmm�dsplit�dstackr�eigvals�eigh�eigvalsh�einsum� embedding� embedding_bag� empty_like�eq�equal�erf�erfc�erfinv�exp�exp2�expm1� fake_quantize_per_channel_affine�fake_quantize_per_tensor_affine�fused_moving_avg_obs_fake_quant�fbgemm_linear_fp16_weight�)fbgemm_linear_fp16_weight_fp32_activation�fbgemm_linear_int8_weight�)fbgemm_linear_int8_weight_fp32_activation�fbgemm_linear_quantize_weight�fbgemm_pack_gemm_matrix_fp16�fbgemm_pack_quantized_matrix�feature_alpha_dropout�feature_dropoutr��ifft�rfft�irfft�hfft�ihfft�hfft2�ihfft2�hfftn�ihfftn�fftn�ifftn�rfftn�irfftn�fft2�ifft2�rfft2�irfft2�fftshift� ifftshift�fix�flatten�flip�fliplr�flipud�frobenius_norm�floor� floor_divide� float_power�fmod�frac�frexp� full_like�strided�_functional_assert_async� lu_unpack�gather�gcd�ge� greater_equal�geqrf�i0�inner�outer�gerr�� grid_sampler�grid_sampler_2d�grid_sampler_3d� group_norm�gru�gru_cell�gt�greater� hardshrink� heaviside�hinge_embedding_loss�histc� histogram� histogramdd�householder_product�hspmm�hsplit�hstack�hypot�igamma�igammacrC� index_add� index_copy� index_put� index_select� index_fill� index_reduce�isfinite�isin�isinf�isreal�isposinf�isneginf� instance_norm�int_repr�inverse�inv�inv_ex� is_complex�is_conj�is_neg�is_distributed� is_inference�is_floating_point� is_nonzero� is_same_size� is_signed�isclose�isnan�istft�kl_div�kron�kthvalue� ldl_factor_ex� ldl_factor� ldl_solve� layer_norm�lcm�ldexp�le� less_equal�lerp�lgamma�lobpcg�log� log_softmax�log10�log1p�log2� logaddexp� logaddexp2�logdet�xlogy� logical_and� logical_not� logical_or� logical_xor�logit� logsumexp�lstm� lstm_cell�lt�less�lu�lu_solve�margin_ranking_loss� masked_fill�masked_scatter� masked_select�matmul� lu_factor� lu_factor_ex� matrix_power� matrix_rank� multi_dot� matrix_expr.�maximum�fmax� max_pool1d� max_pool2d� max_pool3d�max_pool1d_with_indicesr��nanmeanr�� nanmedian�meshgridr-�minimum�fmin�miopen_batch_norm�miopen_convolution�miopen_convolution_add_relu�miopen_convolution_relu�miopen_convolution_transpose�miopen_depthwise_convolution� miopen_rnn�mmr��movedim�moveaxis�msort�mul�multiply� multinomial�mv�mvlgamma�narrow� nan_to_num�native_batch_norm�_native_batch_norm_legit�native_dropout�native_layer_norm�native_group_norm� native_norm�native_channel_shuffle�ne� not_equal�negr � nextafterr�r��adaptive_avg_pool2d�adaptive_avg_pool3d� adaptive_max_pool1d_with_indices�adaptive_max_pool2d� adaptive_max_pool2d_with_indices�adaptive_max_pool3d� adaptive_max_pool3d_with_indices� affine_grid� avg_pool2d� avg_pool3d�binary_cross_entropy� cross_entropy� dropout1d� dropout2d� dropout3d�elu�fold�fractional_max_pool2d�"fractional_max_pool2d_with_indices�fractional_max_pool3d�"fractional_max_pool3d_with_indices�gaussian_nll_loss�gelu�glu� grid_sample�gumbel_softmax�hardtanh� interpolate�l1_lossrrK�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_hotrR�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�tuple�quantized_max_pool2d�quantized_max_pool3d�quantized_rnn_relu_cell�quantized_rnn_tanh_cell�rad2deg� rand_like� randint_like� randn_like�ravelrB�vdot�vecdot� view_as_real�view_as_complex� reciprocal� remainder�renorm�repeat_interleave�reshape�rnn_relu� rnn_relu_cell�rnn_tanh� rnn_tanh_cell�roll�rot90�round� row_stack�_rowwise_prune�rsqrt�rsub�saddmm�scatter� scatter_add�scatter_reduce� searchsorted�_segment_reduce�select�select_scatter� slice_inverse� slice_scatterr��sign�signbit�sgn�sin�sinc�sinh�slogdet�smm�spmmr�solve_ex�sort�split�split_with_sizes�sqrt�square�squeeze�sspaddmm�stackr�std_mean�stft�sub�subtract�sum� sym_float�sym_int�sym_max�sym_min�sym_not�sym_ite� _sym_sqrt�_sym_cos� _sym_cosh�_sym_sin� _sym_sinh�_sym_tan� _sym_tanh� _sym_asin� _sym_acos� _sym_atan�nansum�svd� svd_lowrank�svdvals�swapaxes�swapdims�special�airy_ai� bessel_j0� bessel_j1� bessel_y0� bessel_y1�chebyshev_polynomial_t�chebyshev_polynomial_u�chebyshev_polynomial_v�chebyshev_polynomial_w�entr�erfcx�expit�gammainc� gammaincc�gammaln�hermite_polynomial_h�hermite_polynomial_he�i0e�i1�i1e�laguerre_polynomial_l�legendre_polynomial_p�log_ndtr�modified_bessel_i0�modified_bessel_i1�modified_bessel_k0�modified_bessel_k1� multigammaln�ndtr�ndtri�psi�scaled_modified_bessel_k0�scaled_modified_bessel_k1�shifted_chebyshev_polynomial_t�shifted_chebyshev_polynomial_u�shifted_chebyshev_polynomial_v�shifted_chebyshev_polynomial_w�spherical_bessel_j0�xlog1py�zeta�t�take�take_along_dim�tanr�� tensorinv� tensorsolve� tensordot� tensor_split�tile�topk�tracer*�trapz� trapezoid�triangular_solve�solve_triangular�tril�triu� true_divide�trunc�unbindr�unique�unique_consecutive� unravel_index� unsafe_chunk� unsafe_split�unsafe_split_with_sizes� unsqueezer�r��var_mean�vsplit�vstack�where� zeros_like�_fw_primal_copy�_make_dual_copy�view_as_real_copy�view_as_complex_copy� _conj_copy�_neg_view_copy�as_strided_copy�_sparse_broadcast_to_copy� diagonal_copy� expand_copy� narrow_copy� permute_copy�_reshape_alias_copy� select_copy� detach_copy� slice_copy� split_copy�split_with_sizes_copy� squeeze_copy�t_copy�transpose_copy�unsqueeze_copy� _indices_copy� _values_copy� indices_copy� values_copy�crow_indices_copy�col_indices_copy�ccol_indices_copy�row_indices_copy� unbind_copy� view_copy� unfold_copy� alias_copy� __floordiv__� __rfloordiv__� __ifloordiv__� __truediv__� __rtruediv__� __itruediv__� __lshift__� __rlshift__� __ilshift__� __rshift__� __rrshift__� __irshift__�__and__�__or__�__xor__� __float__� __complex__� __array__�__bool__� __contains__�__neg__� __invert__�__mod__�__rmod__�__imod__�__array_wrap__� __getitem__� __deepcopy__�__int__�__long__� __index__�__len__� __format__� __reduce_ex__� __reversed__�__repr__� __setitem__� __setstate__�Tr(�H�mT�mH�_backward_hooks�_post_accumulate_grad_hooksr5�_cdatar6r7�_grad_fn�grad_fn�_version�_autocast_to_reduced_precision�_autocast_to_full_precisionr�r`ra�is_cuda�is_cpu�is_xla�is_xpu�is_ipu�is_leaf� retains_grad�is_meta�is_mps�is_mtia� is_nested�is_maia� is_mkldnn� is_quantized� is_sparse� is_sparse_csr� is_vulkan�itemsizeror@r��nbytes�ndim� output_nrr�r|�volatile�__cuda_array_interface__�type�_dimI�_dimV�_indices�_is_view�_nnz� crow_indices� col_indices� ccol_indices� row_indices� _update_names�_values�align_as�align_to�apply_rr� as_strided_�backward�bfloat16�preserve_format�bool�byte�char�cauchy_�coalesce� _coalesced_� contiguous�contiguous_format�copy_�cpu�cuda�xpu�ipu�data_ptrrWr}� dim_order�double�cdouble� element_size�expand� expand_as� exponential_�fill_�fill_diagonal_�float�cfloat� geometric_rZ�half�chalf� has_namesrw�int� is_coalesced� is_contiguous� is_pinned� is_set_to� is_shared�item� log_normal_�long�map_�map2_� module_load� ndimension�nelement�_nested_tensor_size�_nested_tensor_storage_offsets�_nested_tensor_strides�numpy� pin_memory�put_ri�random_� record_stream� refine_names� register_hook�"register_post_accumulate_grad_hook�rename�repeat�requires_grad_� reshape_as�resize�resize_� resize_as�resize_as_sparse_� retain_grad�set_� share_memory_�shortry� sparse_dim� sparse_mask�_sparse_mask_projection�sparse_resize_�sparse_resize_and_clear_�storage�untyped_storager�� storage_type� sum_to_sizer6�to_dense� _to_dense� to_sparse�tolist� to_mkldnn�type_asr^�view�view_as�zero_� __dlpack__�__dlpack_device__r�utils�backend_registration�_privateuse1_backend_name�hasattr�getattrr�items�__name__� startswith�len�extendr��update) r6�ret�privateuse1_backend_name�ret2�ignoredr�r#r��subnamer@r"s r1rr�s�`��6�\�F�{%� � �-�-�-�{%� ��2�2�2�{%� �!�#@�#@�{%� �!�#A�#A� {%� � �.�.�.� {%� � �'�'� {%� � �0�0�0�{%� � �/�/�/�{%� � �1�1�1�{%� � �4�4�4�{%� � �Q�Q�Q�{%� � �L�L�L�{%� � �L�L�L�{%� � �L�L�L�{%� � �J�J�J�{%� � �K�K�K�!{%�" �#�%J�%J�#{%�{%�$ � �-�-�-�%{%�& ��X�X�X�'{%�( ��F�F�F�){%�* � �.�.�.�+{%�, � �.�.�.�-{%�. � �J�J�J�/{%�0 � �/�/�/�1{%�2 � �F�F�F�3{%�4 � �&�&�5{%�6 � �&�&�7{%�8 � �1�1�1�9{%�: � �.�.�.�;{%�< ��2�2�={%�> � �0�0�0�?{%�@ � �/�/�/�A{%�B � �1�1�1�C{%�D � �.�.�.�E{%�{%�{%�F � �0�0�0�G{%�H � �6�6�6�I{%�J � �8�8�8�K{%�L � �/�/�/�M{%�N � �1�1�1�O{%�P ��-�-�Q{%�R ��-�-�S{%�T ��-�-�U{%�V ��x�x�x�W{%�X � �R�R�R�Y{%�Z ��{�{�[{%�\ �'�){�){�]{%�^ �(�*u�*u�_{%�` �� Q� Q�a{%�b �%�'v�'v�c{%�d �1�4C�4C�e{%�f �� 5� 5�g{%�{%�{%�h �%�'\�'\�i{%�j ��C�C�C�k{%�l ��?�?�m{%�n �.�NR�2W�2W�2W�o{%�r ��C�C�C�s{%�t ��>�>�>�u{%�v ��<�<�<�w{%�x ��5�5�5�y{%�z ��;�;�;�{{%�| ��<�<�<�}{%�~ � �"C�"C�"C�{%�@ �!�#D�#D�#D�A{%�B ��-�-�C{%�D � �3�3�3�E{%�F ��!4�!4�G{%�H ��1�1�I{%�J ��]�]�]�K{%�{%�{%�L ��1�1�M{%�N � �6�6�6�O{%�P � �9�9�9�Q{%�R ��>�>�>�S{%�T � �a�a�a�U{%�V � �.�.�.�W{%�X � �=�=�=�Y{%�Z ��$�:�:�:�:�[{%�\ ��8�8�]{%�^ ��?�?�?�_{%�` � ��9�9�9�a{%�b � � �"P�"P�"P�c{%�d �� G� G� G�e{%�f ��N�N�N�g{%�h �&�(Y�(Y�i{%�j � �4�4�4�k{%�l � �C�C�C�m{%�{%�{%�n � �B�B�B�o{%�p ��8�8�8�q{%�r ��8�8�8�s{%�t ��8�8�8�u{%�v � �O�O�O�w{%�x � �%�%�y{%�z ��I�I�I�{{%�| � �,�,�}{%�~ ��9�9�9�{%�@ � �(�(�A{%�B � ��5�5�5�C{%�D � �.�.�.�E{%�F ��7�7�7�G{%�H ��6�6�6�I{%�J ��5�5�5�K{%�L ��=�=�=�M{%�N � �d�d�d�O{%�{%�{%�P � �d�d�d�Q{%�R � �d�d�d�S{%�T ��w�w�U{%�V ��=�=�=�W{%�X ��!A�!A�!A�Y{%�Z ��!A�!A�!A�[{%�\ ��!A�!A�!A�]{%�^ ��(�(�_{%�` � �-�-�-�a{%�b �#�&C�&C�&C�c{%�d � �.�.�.�e{%�f ��!C�!C�!C�g{%�h ��-�-�i{%�j � �@�@�@�k{%�l � ��E�E�E�m{%�n ��',�1�1�1�o{%�r � �5�5�5�s{%�{%�{%�t � �5�5�5�u{%�v � �B�B�B�w{%�x � �A�A�A�y{%�z �"�$@�$@�$@�{{%�| ��;�;�;�}{%�~ � �1�1�1�{%�@ ��*�*�A{%�B � �#�#�C{%�D � ��*�*�E{%�F � �&�&�G{%�H � �:�:�:�I{%�J ��@�@�@�K{%�L ��2�2�2�M{%�N � �V�V�V�O{%�P ��B�B�B�Q{%�R � ��K�K�K�S{%�T �� O� O� O�U{%�{%�{%�V � �"Y�"Y�"Y�W{%�X � �1�1�1�Y{%�Z � �0�0�0�[{%�\ � �H�H�H�]{%�^ � �K�K�K�_{%�` � �4�4�4�a{%�b � �@�@�@�c{%�d � �*�*�e{%�f � �)�)�g{%�h � �;�;�i{%�j � �2�2�2�k{%�l � ��4�4�4�m{%�n � ��8�8�8�o{%�p � ��?�?�?�q{%�r � ��C�C�C�s{%�t � �4�4�u{%�v ��!&�+�+�+�w{%�{%�{%�z ��^b�g�g�g�{{%�~ ��e�e�e�{%�@ ��3�3�3�A{%�B � �,�,�C{%�D � �-�-�-�E{%�F � �.�.�.�G{%�H � �0�0�0�I{%�J � �-�-�-�K{%�L � �.�.�.�M{%�N � �/�/�/�O{%�P �.�0o�0o�Q{%�R �-�/h�/h�S{%�T �-�Z_�1d�1d�1d�U{%�Z �'�)N�)N�[{%�\ �7�9^�9^�]{%�^ �'�)}�)}�_{%�` �7�;V�;V�a{%�{%�{%�d �+�-=�-=�e{%�f �*�,<�,<�g{%�h �*�,B�,B�i{%�j �#�%?�%?�k{%�l ��9�9�m{%�n � ��C�C�C�o{%�p � ��C�C�C�q{%�r � ��D�D�D�s{%�t � ��C�C�C�u{%�v � ��D�D�D�w{%�x � ��J�J�J�y{%�z � ��K�K�K�{{%�| � ��D�D�D�}{%�~ � ��E�E�E�{%�@ � ��E�E�E�A{%�B � ��F�F�F�C{%�D � ��F�F�F�E{%�{%�{%�F � ��G�G�G�G{%�H � ��I�I�I�I{%�J � ��J�J�J�K{%�L � ��J�J�J�M{%�N � ��K�K�K�O{%�P � ��6�6�6�Q{%�R � ��7�7�7�S{%�T � � �B�B�B�U{%�V � �-�-�-�W{%�X � �@�@�@�Y{%�Z � �*�*�[{%�\ � �&�&�]{%�^ � �&�&�_{%�` ��Q�Q�Q�a{%�b � �/�/�/�c{%�d ��3�3�e{%�f ��?�?�?�g{%�{%�{%�h � �5�5�5�i{%�j � �.�.�.�k{%�l � �/�/�/�m{%�n ��t�4�PU�P]�fj�z�D�D�D�o{%�p �&�(H�(H�q{%�r ��\�\�\�s{%�t � �O�O�O�u{%�v � �4�4�4�w{%�x ��3�3�3�y{%�z ��>�>�>�{{%�| � �/�/�/�}{%�~ ��,�,�,�{%�@ � �6�6�6�A{%�B � �5�5�5�C{%�D � �3�3�3�E{%�F ��N�N�N�G{%�H ��c�c�I{%�{%�{%�J ��f�f�K{%�L ��f�f�M{%�N ��m�m�m�O{%�P � �s�s�Q{%�R ��N�N�N�S{%�T ��3�3�3�U{%�V � �8�8�8�W{%�X ��5�5�5�Y{%�Z ��;�;�;�[{%�\ �"�$z�$z�$z�]{%�^ � �G�G�G�_{%�` ��m�m�m�a{%�b ��Y�Y�Y�c{%�d � �(�*?�*?�e{%�f � �4�4�4�g{%�h � �;�;�i{%�j � �2�2�2�k{%�{%�{%�l � �6�6�6�m{%�n � �7�7�7�o{%�p � �8�8�8�q{%�r � �.�.�.�s{%�t ��=�=�u{%�v ��>�>�w{%�x ��L�L�L�y{%�z ��B�B�B�{{%�| ��=�=�}{%�~ ��\�\�\�{%�@ ��)�)�A{%�B � �G�G�G�C{%�D � �&�&�E{%�F � �'�'�G{%�H ��2�2�2�I{%�J ��2�2�2�K{%�L ��0�0�M{%�{%�{%�P ��(�(�Q{%�R � �1�1�1�S{%�T � ��4�4�4�U{%�V � ��K�K�K�W{%�X ��*�*�Y{%�Z � �'�'�[{%�\ � �&�&�]{%�^ ��.�.�_{%�` ��,�,�a{%�b ��!1�!1�c{%�d ��*�*�e{%�f ��3�3�g{%�h ��)�)�i{%�j � �W�W�W�k{%�l � �%�%�m{%�n � �SX�]�]�]�o{%�r � �r�r�r�s{%�{%�{%�t � �+�+�u{%�v ��N�N�N�w{%�x � �"�$c�$c�$c�y{%�z � ��!L�!L�!L�{{%�| � �� S� S� S�}{%�~ ��s�s�s�{%�@ � �4�4�4�A {%�B � �6�6�6�C {%�D ��3�3�3�E {%�F ��;�;�;�G {%�H � �;�;�;�I {%�J � �0�0�0�K {%�L � �LP�U�U�U�M {%�P � �-�-�-�Q {%�R ��<�<�<�S {%�T � �/�/�/�U {%�V � �/�/�/�W {%�{%�{%�X � �.�.�.�Y {%�Z ��:�:�:�[ {%�\ ��;�;�;�] {%�^ � �&�&�_ {%�` � �.�.�.�a {%�b ��<�<�<�c {%�d ��5�5�5�e {%�f ��;�;�;�g {%�h ��<�<�<�i {%�j � �/�/�/�k {%�l ��I�I�I�m {%�n � �s�s�o {%�p ��O�O�O�q {%�r ��3�3�3�s {%�t � �5�5�5�u {%�v ��E�E�E�w {%�x ��B�B�B�y {%�{%�{%�z �!�$A�$A�$A�{ {%�| ��8�8�} {%�~ ��<�<� {%�@ ��=�=�=�A {%�B � �7�7�7�C {%�D � ��?�?�?�E {%�F � �� F� F� F�G {%�H � �!�#]�#]�#]�I {%�J � ��[�[�[�K {%�L � ��>�>�>�M {%�N ��/�/�O {%�P � �!�#@�#@�#@�Q {%�R � � �"M�"M�"M�S {%�T � �� <� <� <�U {%�V ��*�*�W {%�X � ��!1�!1�Y {%�Z � �-�-�-�[ {%�{%�{%�\ � �8�8�8�] {%�^ � �5�5�5�_ {%�` ��l�l�l�a {%�b ��l�l�l�c {%�d ��l�l�l�e {%�f �%�HM�)R�)R�)R�g {%�j � �.�.�.�k {%�l � �V�V�V�m {%�n � �0�0�0�o {%�p ��3�3�3�q {%�r ��5�5�s {%�t � �-�-�-�u {%�v � �8�8�8�w {%�x � �5�5�5�y {%�z ��#J�#J�{ {%�~ � �"}�"}� {%�@ �)�+v�+v�A {%�{%�{%�B �%�'h�'h�C {%�D �*�.R�.R�E {%�H �*�.?�.?�I {%�L ��Y�Y�M {%�P ��2�2�2�Q {%�R � �E�E�E�S {%�T � �<�<�U {%�V ��=�=�W {%�X � �A�A�A�Y {%�Z � �4�4�4�[ {%�\ ��9�9�9�] {%�^ ��U�U�U�_ {%�` ��1�1�1�a {%�b ��+�+�c {%�d � �:�:�e {%�f ��W�W�W�g {%�h ��!s�!s�i {%�{%�{%�j �&�(_�(_�k {%�l ��8�8�m {%�n ��!f�!f�!f�o {%�p ��!V�!V�q {%�r ��U�U�U�s {%�t �$�&?�&?�u {%�v ��3�3�3�w {%�x ��:�:�:�y {%�z � �-�-�-�{ {%�| ��2�2�2�} {%�~ ��:�:�:� {%�@ ���/�1N�1N�A {%�B ���/�1N�1N�C {%�D ���/�1d�1d�1d�E {%�F ���<�>q�>q�>q�G {%�H ���/�1d�1d�1d�I {%�J ���<�>q�>q�>q�K {%�{%�{%�L ���/�1d�1d�1d�M {%�N ���<�>q�>q�>q�O {%�P ���'�)S�)S�)S�Q {%�R ���)�+a�+a�+a�S {%�T ���&�RV�*[�*[�*[�U {%�X ���&�RV�*[�*[�*[�Y {%�\ ���&�;@�*E�*E�*E�] {%�` ���$�&R�&R�&R�a {%�b ���0�=C�4H�4H�4H�c {%�f ���<�im�@r�@r�@r�g {%�j ��� �"L�"L�"L�k {%�l ���1�KQ�5V�5V�5V�m {%�p ���)�[^�-c�-c�-c�q {%�t ���$�GL�(Q�(Q�(Q�u {%�x ���#�%Z�%Z�%Z�y {%�z ���%�'\�'\�'\�{ {%�| ���%�'\�'\�'\�} {%�{%�{%�~ ���%�'\�'\�'\� {%�@ ����!K�!K�!K�A {%�B ���%�IN�)S�)S�)S�C {%�F ���)�tx�SW�-\�-\�-\�G {%�L ���1�3i�3i�3i�M {%�N ��� �"m�"m�"m�O {%�P ���1�Z^�5c�5c�5c�Q {%�T ���>�[`� � %� %� %�W {%�Z ���1�Z^�5c�5c�5c�[ {%�^ ���>�[`� � %� %� %�a {%�d ���-�/u�/u�/u�e {%�f ��� �"F�"F�"F�g {%�h ����!9�!9�!9�i {%�j ���'�)z�)z�)z�k {%�l ���&�(g�(g�(g�m {%�n ���*�,c�,c�,c�o {%�p ���&�(C�(C�(C�q {%�{%�{%�r ���$�&^�&^�&^�s {%�t ���0�=C�4H�4H�4H�u {%�x ���)�TY�-^�-^�-^�y {%�| ���'�QV�+[�+[�+[�} {%�@ ���"�%A�%A�%A�A{%�B ���#�%o�%o�%o�C{%�D ���&�(m�(m�(m�E{%�F ���&�(\�(\�(\�G{%�H ���"�$G�$G�$G�I{%�J ���/�1g�1g�1g�K{%�L ���'�)^�)^�)^�M{%�N ���&�(8�(8�O{%�P ���%�'m�'m�'m�Q{%�R ���%�'m�'m�'m�S{%�T ���%�'m�'m�'m�U{%�V ���/�IO�3T�3T�3T�W{%�Z ���&�IN�*S�*S�*S�[{%�{%�{%�^ ���3�V[�7`�7`�7`�_{%�b ���&�IN�*S�*S�*S�c{%�f ���3�V[�7`�7`�7`�g{%�j ���&�IN�*S�*S�*S�k{%�n ���3�V[�7`�7`�7`�o{%�r ���(�*z�*z�*z�s{%�t ���(�*z�*z�*z�u{%�v ���(�*z�*z�*z�w{%�x ���$�&p�&p�&p�y{%�z ���8�fj�qu�ch� m� m� m�}{%�D ���-�GM�1R�1R�1R�E{%�H ���2�?E�6J�6J�6J�I{%�L ���7�QW�;\�;\�;\�M{%�P ���$�>D�(I�(I�(I�Q{%�T ���%�'X�'X�'X�U{%�V ���#�%F�%F�%F�W{%�X ����!P�!P�!P�Y{%�{%�{%�Z ���-�/a�/a�/a�[{%�\ ���,�QW�0\�0\�0\�]{%�` ���!�#;�#;�a{%�b ��� �"A�"A�"A�c{%�d ���!�#B�#B�#B�e{%�f ���$�&_�&_�&_�g{%�h ���!�#y�#y�#y�i{%�j ��� �"A�"A�"A�k{%�l ��� �"A�"A�"A�m{%�n ��� �"A�"A�"A�o{%�p ���8�:u�:u�:u�q{%�r ���*�,�,�,�s{%�t ���&�(\�(\�(\�u{%�v ���,�.x�.x�.x�w{%�x ���#�%Z�%Z�%Z�y{%�z ���#�%Z�%Z�%Z�{{%�| ���$�&L�&L�&L�}{%�{%�{%�~ ���&�(C�(C�(C�{%�@ ���$�&6�&6�A{%�B ���&�(8�(8�C{%�D ���%�'X�'X�'X�E{%�F ���/�hn�3s�3s�3s�G{%�J ���=�RV�_b�EJ�V\�Aa�Aa�Aa�Aa�K{%�P ���"�$b�$b�$b�Q{%�R �� �� M� M� M�S{%�T �� ��Q�Q�Q�U{%�V �� ��!7�!7�W{%�X �� �&�(x�(x�(x�Y{%�Z � �7�7�7�[{%�\ ���F�F�F�F�]{%�^ ��(�(�_{%�` � �\�\�\�a{%�b � ��d�d�d�c{%�d � � �"h�"h�"h�e{%�{%�{%�f � � �"p�"p�"p�g{%�h ��9�9�9�i{%�j ��d�d�d�k{%�l � �%�%�m{%�n � �*�*�o{%�p � �Q�Q�Q�q{%�r ��!S�!S�!S�s{%�t � �+�+�u{%�v ��I�I�I�w{%�x � �*�*�*�y{%�z ��5�5�5�{{%�| � ��I�I�I�}{%�~ ��=�=�{%�@ ��A�A�A{%�B � �7�7�7�C{%�D �� Y� Y�E{%�F ��6�6�6�G{%�{%�{%�H ��2�2�2�I{%�J � �-�-�K{%�L ��d�d�d�M{%�N � �7�7�7�O{%�P � �0�0�0�Q{%�R � �D�D�D�S{%�T � �"2�"2�U{%�V �"�$4�$4�W{%�X �'�)9�)9�Y{%�Z � �'�'�[{%�\ ��,�,�]{%�^ ��7�7�7�_{%�` � ��C�C�C�a{%�b ��f�f�f�c{%�d ��i�i�i�e{%�f �"�$V�$V�g{%�h �!�#M�#M�i{%�{%�{%�j �)�+P�+P�k{%�l �"�$s�$s�m{%�n � �$h�$h�o{%�t �!�%i�%i�u{%�x �"�u�w�w�X\�.2�e�&H�&H�&H�y{%�| �"�u�w�w�X^�.4��&J�&J�&J�}{%�@ �"�u�w�w�Xa�.7�5�&M�&M�&M�A{%�D �%�)m�)m�E{%�H �%�)m�)m�I{%�L � �1�1�1�M{%�N ��d�d�d�O{%�P ��d�5�=�Y]�mr�v�v�v�Q{%�R ��e�e�e�S{%�T � �%�%�U{%�V � �.�.�.�W{%�X � �5�5�5�Y{%�Z � ��F�F�F�[{%�{%�{%�\ ��,�,�]{%�^ ��/�/�_{%�` ��4�4�4�a{%�b � �3�3�3�c{%�d ��:�:�:�e{%�f � �A�A�A�g{%�h ��!;�!;�!;�i{%�j � �.�.�k{%�l ��Q�Q�Q�m{%�n ��x�x�o{%�p ��S�S�S�q{%�r ��x�x�s{%�t ��S�S�S�u{%�v � �7�7�7�w{%�x � �7�7�7�y{%�z � �/�/�/�{{%�| ��5�5�5�}{%�{%�{%�~ ��P�P�{%�@ � �`�`�`�A{%�B � �/�/�/�C{%�D � �4�4�4�E{%�F � �M�M�M�G{%�H � �8�8�I{%�J ��<�<�K{%�L ��Z�Z�Z�M{%�N ��e�e�e�O{%�P ��|�|�|�Q{%�R � �2�2�S{%�T ��?�?�U{%�V ��W�W�W�W{%�X ��W�W�W�Y{%�Z � �3�3�3�[{%�\ � �1�1�1�]{%�^ � �.�.�.�_{%�{%�{%�` � �1�1�1�a{%�b � �-�-�-�c{%�d � �-�-�-�e{%�f � �.�.�.�g{%�h � �.�.�.�i{%�j � �'�'�k{%�l � ��.�.�m{%�n � �)�)�o{%�p � �*�*�q{%�r � �8�8�8�s{%�t � ��@�@�@�u{%�v � ��W�W�W�w{%�x � �Y�e�QU�Y�Y�Y�Y�y{%�z � �E�E�E�{{%�| �� P� P� P�}{%�~ � �.�.�.�{%�@ � �0�0�0�A{%�{%�{%�B � �;�;�;�C{%�D ��O�O�O�E{%�F � �8�8�8�G{%�H � �-�-�-�I{%�J ��2�2�2�K{%�L � �Y]�b�b�b�M{%�P � �4�4�4�Q{%�R ��9�9�9�S{%�T � �-�-�-�U{%�V ��)�)�W{%�X � �'�'�Y{%�Z � ���[{%�\ � ���]{%�^ � �'�'�_{%�` � �)�)�a{%�b ��)�)�c{%�d ��(�(�e{%�{%�{%�f ��)�)�g{%�h ��(�(�i{%�j ��)�)�k{%�l ��(�(�m{%�n ��)�)�o{%�p ��)�)�q{%�r ��)�)�s{%�t ��)�)�u{%�v � �0�0�0�w{%�x � �I�I�I�y{%�z ��A�A�A�{{%�| � ��H�H�H�}{%�~ � ��8�8�8�{%�@ ��4�4�A{%�B ��6�6�C{%�D � ��/�/�E{%�F � ��!1�!1�G{%�{%�{%�H � ��!1�!1�I{%�J � ��!1�!1�K{%�L � ��!1�!1�M{%�N � �,�.K�.K�.K�O{%�P � �,�.K�.K�.K�Q{%�R � �,�.K�.K�.K�S{%�T � �,�.K�.K�.K�U{%�V � ��/�/�W{%�X � ��,�,�Y{%�Z � ��+�+�[{%�\ � ��,�,�]{%�^ � ��-�-�_{%�` � ��.�.�a{%�b � ��,�,�c{%�d � ��-�-�e{%�f � ��-�-�g{%�h � �� A� A� A�i{%�{%�{%�j � ��!B�!B�!B�k{%�l � ��/�/�m{%�n � �*�,I�,I�,I�o{%�p � �+�-J�-J�-J�q{%�r � ��*�*�s{%�t � ��+�+�u{%�v � ��*�*�w{%�x � ��+�+�y{%�z � �+�-J�-J�-J�{{%�| � �+�-J�-J�-J�}{%�~ � ��-�-�{%�@ � �� 0� 0�A{%�B � �!�#D�#D�#D�C{%�D � ��-�-�E{%�F � ��!O�!O�!O�G{%�H � �(�*:�*:�I{%�J � �(�*:�*:�K{%�{%�{%�L � �(�*:�*:�M{%�N � �(�*:�*:�O{%�P � �"�$7�$7�Q{%�R � ��,�,�S{%�T � ��-�-�U{%�V � ��!>�!>�!>�W{%�X � ��+�+�Y{%�Z � ��-�-�[{%�\ � �/�1A�1A�]{%�^ � �/�1A�1A�_{%�` � �4�6S�6S�6S�a{%�b � �4�6S�6S�6S�c{%�d � �4�6S�6S�6S�e{%�f � �4�6S�6S�6S�g{%�h � ��,�,�i{%�j � ��@�@�@�k{%�l � �)�+;�+;�m{%�{%�{%�n � ��@�@�@�o{%�p � ��>�>�>�q{%�r � ��<�<�<�s{%�t ��!�!�u{%�v � �+�+�w{%�x ��K�K�K�y{%�z � �-�-�-�{{%�| � �.�.�.�}{%�~ � �� 3� 3� 3�{%�@ � � �"<�"<�"<�A{%�B ��:�:�:�C{%�D ��H�H�H�E{%�F ��J�J�J�G{%�H � �*�*�I{%�J � �K�K�K�K{%�L � �%�%�M{%�N ��5�5�O{%�{%�{%�P � �1�1�1�Q{%�R ��5�5�5�S{%�T �� e� e� e�U{%�V � �%�'a�'a�'a�W{%�X � �:�:�:�Y{%�Z �!�NT�%Y�%Y�%Y�[{%�` � �:�:�:�a{%�b ��2�2�c{%�d � �/�/�/�e{%�f � �-�-�-�g{%�h ��<�<�i{%�j � �h�h�h�k{%�l � �"g�"g�"g�m{%�n ��6�6�o{%�p ��;�;�;�q{%�r ��L�L�L�s{%�t �%�'W�'W�'W�u{%�{%�{%�v ��8�8�8�w{%�x � ��1�1�1�y{%�z � �-�-�-�{{%�| ��2�2�2�}{%�~ � �;�;�{%�@ � �2�2�2�A{%�B � �9�9�9�C{%�D ��e�e�e�E{%�F ��5�5�G{%�H ��@�@�I{%�J ����K{%�L �"�O�O�M{%�N ��/�/�O{%�P ��o�o�Q{%�R ��Q�Q�Q�S{%�T �'�)>�)>�U{%�V ��F�F�F�W{%�{%�{%�X ��%�C�C�C�C�Y{%�Z ��>�>�[{%�\ ��1�1�]{%�^ �!�#@�#@�_{%�` ��6�6�a{%�b ��?�?�c{%�d ��N�N�N�e{%�f ��<�<�<�g{%�h �#�%H�%H�%H�i{%�j ��0�0�k{%�l � �o�o�m{%�n ��9�9�o{%�p ��2�2�q{%�r ��_�_�s{%�t ��O�O�u{%�v ��O�O�w{%�x ��?�?�y{%�{%�{%�z ����{{%�| ����}{%�~ ����{%�@ ����A{%�B ��1�1�1�C{%�D ��/�/�E{%�F ��A�A�G{%�H ��/�/�I{%�J ��3�3�K{%�L ��4�4�M{%�N ��4�4�O{%�P ��2�2�Q{%�R ��3�3�S{%�T ��3�3�U{%�V ��1�1�W{%�X ��2�2�Y{%�Z ��2�2�[{%�{%�{%�\ ��1�1�]{%�^ ��2�2�_{%�` ��2�2�a{%�b ��.�.�c{%�d � �-�-�e{%�f ��.�.�g{%�h ��/�/�i{%�j ��O�O�k{%�l ��0�0�m{%�n ����o{%�p ��3�3�q{%�r ����s{%�t ��?�?�u{%�v ��.�.�w{%�x ��/�/�y{%�z ��/�/�{{%�| ��5�5�}{%�{%�{%�~ ��0�0�{%�@ ��2�2�A{%�B ����C{%�D ����E{%�F ��/�/�G{%�H ����I{%�J ��7�7�K{%�L ��4�4�M{%�N ��_�_�O{%�P ���A�A�A�A�Q{%�R ��1�1�S{%�T ��/�/�U{%�V ���/�/�W{%�X ���/�/�Y{%�Z � ��?�?�[{%�\ � ��?�?�]{%�^ ��&���_{%�{%�{%�` �*�2�O�O�a{%�b � ��o�o�c{%�d � ����e{%�f � ��_�_�g{%�h � ��o�o�i{%�j �����k{%�l �����m{%�n �����o{%�p �-�/p�/p�q{%�r �*�,V�,V�s{%�t � ��_�_�u{%�v � ����w{%�x � ��o�o�y{%�z �����{{%�| � ����}{%�~ � ����{%�@ � ����A{%�{%�{%�B � ����C{%�D �����E{%�F ��#�_�_�G{%�H �����I{%�J � ����K{%�L �����M{%�N �� �/�/�O{%�P �����Q{%�R �� �/�/�S{%�T ��#�_�_�U{%�V �� �/�/�W{%�X ��$�o�o�Y{%�Z �� �/�/�[{%�\ �����]{%�^ � ����_{%�` � ��_�_�a{%�b � ��o�o�c{%�{%�{%�d � ����e{%�f � ��_�_�g{%�h �� �/�/�i{%�j ��$�o�o�k{%�l � ��o�o�m{%�n �����o{%�p � ��_�_�q{%�r � ��_�_�s{%�t �'�/���u{%�v � �N�N�N�w{%�x � �o�o�y{%�z � �o�o�{{%�| ����}{%�~ ����{%�@ � �_�_�A{%�B ��_�_�C{%�D ��O�O�E{%�{%�{%�F ��_�_�G{%�H ��O�O�I{%�J ��=�=�K{%�L ����M{%�N ����O{%�P ��/�/�Q{%�R ��=�=�S{%�T � �0�0�U{%�V ��8�8�W{%�X ��9�9�Y{%�Z ��k�k�k�[{%�\ ��E�4I�M�M�M�]{%�^ � ��0E�I�I�I�_{%�` � ��0E�I�I�I�a{%�b � ��0E�I�I�I�c{%�d ��M�T�M�M�M�M�e{%�f ����g{%�{%�{%�h ��6�6�i{%�j ��e�6M�Q�Q�Q�k{%�l � �>�>�>�m{%�n � �u�/D�H�H�H�o{%�p � ��0E�I�I�I�q{%�r � �u�/D�H�H�H�s{%�t � �u�/D�H�H�H�u{%�v ����w{%�x ��/�/�y{%�z ��!O�!O�!O�{{%�| � �O�O�}{%�~ ��/�/�{%�@ � �%�2G�K�K�K�A{%�B ��5�3H�L�L�L�C{%�D ��_�_�E{%�F � �,�,�G{%�H ��0�0�I{%�{%�{%�J ��H��H�H�H�H�K{%�L � �,�,�M{%�N ��5�5�O{%�P � ��1F�J�J�J�Q{%�R � �%�2G�K�K�K�S{%�T ���@�@�@�@�U{%�V ��?�?�W{%�X � ��0E�I�I�I�Y{%�Z � ��1F�J�J�J�[{%�\ ��/�/�]{%�^ ����_{%�` � �u�/D�H�H�H�a{%�b ��_�_�c{%�d ��o�o�e{%�f ��_�_�g{%�h ��/�/�i{%�j ��1�1�k{%�{%�{%�l ��/�/�m{%�n � �_�_�o{%�p ��M�T�M�M�M�M�q{%�r ��0�0�s{%�t � ��0E�I�I�I�u{%�v � �6�6�w{%�x � �5�5�y{%�z � �(�(�{{%�| ��@�@�@�}{%�~ ��E�E�{%�@ ��?�?�A{%�B ����C{%�D �"�O�O�E{%�F �-���G{%�H �%���I{%�J ����K{%�L � �o�o�M{%�{%�{%�N ��,�,�O{%�P ��?�?�Q{%�R � �G�G�G�S{%�T ����U{%�V ��L�D�L�L�L�L�W{%�X ��5�5�Y{%�Z ��3�3�[{%�\ ��3�3�]{%�^ �1�3H�3H�_{%�` � �,�,�a{%�b � �-�-�c{%�d ��B�B�B�e{%�f ��1�1�g{%�h � �-�-�i{%�j ��-�-�k{%�l ��0�0�m{%�n � �"8�"8�o{%�{%�{%�p ��O�O�q{%�r � �[�[�[�s{%�t ��?�?�u{%�v ��o�o�w{%�x � ��1F�J�J�J�y{%�z � �_�_�{{%�| ��W�W�W�}{%�~ ��?�?�{%�@ ��1�1�A{%�B �&�(W�(W�(W�C{%�D ��G�G�E{%�F �'�)Q�)Q�G{%�H ��O�O�O�I{%�J ����K{%�L ����M{%�N ����O{%�P ��_�_�Q{%�{%�{%�R ��1�1�S{%�T � �+�+�U{%�V � �E��UZ�Uj�n�n�n�W{%�X ��I��I�I�I�I�Y{%�Z ��G�G�G�[{%�\ ��/�/�]{%�^ � ���_{%�` ��/�/�a{%�b ��.�.�c{%�d � �=�=�e{%�f ��7�7�7�g{%�h � ���i{%�j � �+�+�k{%�l ��.�.�m{%�n � �o�o�o{%�p ��7�7�7�q{%�r � �/�/�s{%�{%�t � ��F�F�F�u{%�{%�C�z %�{�?�Y���v�/�0�0�Y�9s�9s�9s��G�F�4� 5� 5�6�IX���G�F�<�":�<�<� =� =� E�F� �D�#�%�%�G�� � � � �����1� �J� �J�� � �1�:� �� $� �A�J� �� %� �A�J� �� %�  �� �:� � �� ,� ,� ��j��Z���!1�!1�2�G� �L�L��w���%����$�&����$�&�� � � � � � �D��6�4��.�.�D���~�~� �$�c�/�/�d�'�6I�6I���T� �� � �J�J�t���� �Jr4� dispatcherc����fd�}|S)a�Wraps a given function with ``__torch_function__`` -related functionality. Parameters ---------- dispatcher: Callable A callable that returns an iterable of Tensor-likes passed into the function. Note ---- This decorator may reduce the performance of your code. Generally, it's enough to express your code as a series of functions that, themselves, support __torch_function__. If you find yourself in the rare situation where this is not the case, e.g. if you're wrapping a low-level library and you also need it to work for Tensor-likes, then this function is available. Examples -------- >>> def dispatcher(a): # Must have the same signature as func ... return (a,) >>> @torch.overrides.wrap_torch_function(dispatcher) >>> def func(a): # This will make func dispatchable by __torch_function__ ... return a + 0 c�N����tj������fd�����S)Nc�d���|i|��}t|��rt�|g|�Ri|��S�|i|��Sr})rr)r/r0� relevant_argsr� r"�wrappeds ���r1r� z3wrap_torch_function.<locals>.inner.<locals>.wrapped�s]���&�J��7��7�7�M�!�-�0�0� V�,�W�m�U�d�U�U�U�f�U�U�U��4��(��(�(� (r4)� functoolsr )r"r� r� s`@�r1rLz"wrap_torch_function.<locals>.inner�sD����� ��� � � )� )� )� )� )� )� � � )��r4r=)r� rLs` r1rr�s#���. � � � � � �Lr4r� � get_type_fnc���|�t}tj���sgSt ��}g}|D]�}||��}||vr�t |d��r�|jtjjkru|rm|�|��t|��}t|��D]"\}}t|||����r|}n�#|� ||����|h}|g}��|S)a Returns a list of arguments on which to call __torch_function__. Checks arguments in relevant_args for __torch_function__ implementations, storing references to the arguments and their types in overloaded_args and overloaded_types in order of calling precedence. Only distinct types are considered. If a type is a subclass of another type it will have higher precedence, otherwise the precedence order is the same as the order of arguments in relevant_args, that is, from left-to-right in the argument list. The precedence-determining algorithm implemented in this function is described in `NEP-0018`_. See torch::append_overloaded_arg for the equivalent function in the C++ implementation. Parameters ---------- relevant_args : iterable of array-like Iterable of array-like arguments to check for __torch_function__ methods. get_type_fn : callable, optional Function to call on each argument in relevant_args to get its type. Returns ------- overloaded_args : list Arguments from relevant_args on which to call __torch_function__ methods, in the order in which they should be called. .. _NEP-0018: https://numpy.org/neps/nep-0018-array-function-protocol.html Nr) rN r!�_C�_is_torch_function_enabled�setr� r�_disabled_torch_function_implr�r� � enumerate� issubclass�insert) r� r� �overloaded_types�overloaded_args�arg�arg_typer-�i�old_args r1�_get_overloaded_argsr� �s/��D��� � �8� .� .� 0� 0��� �"%�%�%��!#�O��(�(���;�s�#�#�� �,� ,� ,���CW�1X�1X� ,��+�u�x�/U�U�U� � (� �$�$�X�.�.�.��O�,�,��"+�O�"<�"<���J�A�w�!�(�K�K��,@�,@�A�A�� !����� �&�&�u�c�2�2�2�2�$,�:� �#&�%��� �r4� public_apic��t|��}ttt|����}t ��rIt ��5}|�||||��}ddd��n #1swxYwY|tur|S|D]j}|j} t| d��r6| j |ur-| tj j urtjdt��| ||||��}|tur|cS�k|j�d|j��} d| �dd�|D����} t ��r| dt%����z } t'| ���) 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__z�Defining your `__torch_function__ as a plain method is deprecated and will be an error in future, please define it as a classmethod.�.zno implementation found for 'z.' on types that implement __torch_function__: c�,�g|]}t|����Sr=)rN )�.0r� s r1� <listcomp>z)handle_torch_function.<locals>.<listcomp>ws��E�E�E�c��S� � �E�E�Er4z nor in mode )r� rE �maprN r�_pop_mode_temporarilyr�NotImplementedr� r� r!r� r� r+�warn�DeprecationWarning� __module__r� �_get_current_function_mode� TypeError) r� r� r/r0r� �typesr��result�overloaded_arg�torch_func_method� func_namer�s r1rr/s���L+�=�9�9�O� �#�d�O�,�,� -� -�E�'�(�(��#� $� $� N���,�,�Z���f�M�M�F� N� N� N� N� N� N� N� N� N� N� N���� N� N� N� N� �� '� '��M�*����+�=�� �$�j� 1� 1� .�6G�6P�Tb�6b�6b�!���)O�O�O� �M�[�,� .� .� .� #�"�:�u�d�F�C�C�� �� '� '��M�M�M� (��(�@�@�:�+>�@�@�I� H� � H� H�E�E�_�E�E�E� H� H��'�(�(�>� �=�9�;�;�=�=�=�� �C�.�.�s�A3�3A7�:A7a�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``. z�Special case of `has_torch_function` for single inputs. Instead of: `has_torch_function((t,))` call: `has_torch_function_unary(t)` which skips unnecessary packing and unpacking work. a'Special case of `has_torch_function` that skips tuple creation. This uses the METH_FASTCALL protocol introduced in Python 3.7 Instead of: `has_torch_function((a, b))` call: `has_torch_function_variadic(a, b)` which skips unnecessary packing and unpacking work. c �V�tjt��}i}dttjfdtjtjjfdtjjttjj��fdtjjttjj��fdtj ttj ��fdtj ttj ��fdtj ttj ��fdtj ttj ��fg}|D�]c\}}}|D�]X}d }|tj urk|� d ��r�)|� d ��rd }no|�d ��rd }nW|d ���sd }n:|dkr�}n2t!||��}t!t"|d��|kr��|dkr��t!||��}|tj urt!t"|d��|kr��t%|t&j��r��t%|t*j��r��!t/|��s�t1|d��r�|�d|�d�||j<|�d|�d�||j<|r��d|jt7��vr<d} |jt9��vs#J| �||j��������||�|j�����t/|��s���|�d|��||<|r���|t7��vr7d} |t9��vs#J| �||j�������=||�|����Z��e||fS)Nr!ztorch.functionalztorch.nn.functionalz torch.nn.initz torch.Tensorz torch.linalgz torch.fftz torch.specialFryrxTr� unique_dim� __weakref__r(r� z.__get__z.__set__zk{}.{} is in the tuple returned by torch._overrides.get_ignored_functions but still has an explicit override) � collections� defaultdict�listr!�__all__r�r��dirr�r6r�r�r� r� �endswith�islowerr� �object� isinstancer� � ModuleType� __future__�_Featurer�r� r(�__set__rr�formatr� r�) �overridable_funcsr-�tested_namespaces� namespace_str� namespace�ns_funcsr� r(r"r�s r1�_get_overridable_functionsr �s���#�/��5�5�� �E� �%���'� �U�-�u�/?�/G�H� ��� 3�S���9L�5M�5M�N� �%�(�-��U�X�]�);�);�<� ���s�5�<�'8�'8�9� ���s�5�<�'8�'8�9� �e�i��U�Y���0� �%�-��U�]�);�);�<� ��/@�;6�;6�*� �y�(�!�: 6�: 6�I��F��� �,�,��'�'��-�-� ���)�)�#�.�.��!�F�F��'�'��,�,��!�F�F�"�1��-�-�/�/��!�F�F��,�.�.��/��y�)�4�4���6�9�d�3�3�t�;�;��� �-�-���9�i�0�0�D��E�L�(�(�W�V�Y��-M�-M�QU�-U�-U���$�� 0�1�1� ���$� � 3�4�4� ���D�>�>� �g�d�I�&>�&>� �)6�&L�&L��&L�&L�&L��d�l�#�)6�&L�&L��&L�&L�&L��d�l�#�����<�#8�#:�#:�:�:�@�C��<�/D�/F�/F�F�F�F�� � �S\�^b�^k�Hl�Hl�F�F�F��%�d�+�2�2�4�<�@�@�@���D�>�>� ��*�8�8�Y�8�8�E�$�K�� ���,�.�.�.�.�<���#8�#:�#:�:�:�:�C�J�J�y�RV�R_�<`�<`�:�:�:�� �i� (� /� /�� 5� 5� 5� 5�u: 6�v �e� #�#r4c�*�t��dS)aList functions that are overridable via __torch_function__ Returns ------- Dict[Any, List[Callable]] A dictionary that maps namespaces that contain overridable functions to functions in that namespace that can be overridden. r)r r=r4r1rr�s�� &� '� '�� *�*r4c���t|tjjtjjf��rt |��St ��d�|��S)a&Get a human readable string name for a function passed to __torch_function__ Arguments --------- f : Callable Function to resolve the name of. Returns ------- str Name of the function; if eval'ed it should give back the input function. r[)r� r!�_ops� OpOverload�OpOverloadPacket�strr �get)�fs r1rrsO�� �!�e�j�+�U�Z�-H�I�J�J���1�v�v� � %� '� '�� *� .� .�q� 1� 1�1r4c�`�t��}t|tj��}|S)z> Returns a set of the overridable methods on ``torch.Tensor`` )rr� r!r6)r �methodss r1�_get_tensor_methodsr s*��2�3�3���#�E�L�1�2�2�G� �Nr4c�8�|t��vp |jdkS)aw Returns True if the function passed in is a handler for a method or property belonging to ``torch.Tensor``, as passed into ``__torch_function__``. .. note:: For properties, their ``__get__`` method must be passed in. This may be needed, in particular, for the following reasons: 1. Methods/properties sometimes don't contain a `__module__` slot. 2. They require that the first passed-in argument is an instance of ``torch.Tensor``. Examples -------- >>> is_tensor_method_or_property(torch.Tensor.add) True >>> is_tensor_method_or_property(torch.add) False r()r r� )r"s r1rr s"��. �&�(�(� (� F�D�M�Y�,F�Fr4c�X�t|��tjupt|d��S)a9 Returns ``True`` if the passed-in input is a Tensor-like. Currently, this occurs whenever there's a ``__torch_function__`` attribute on the type of the input. Examples -------- A subclass of tensor is generally a Tensor-like. >>> class SubTensor(torch.Tensor): ... >>> is_tensor_like(SubTensor([0])) True Built-in or user types aren't usually Tensor-like. >>> is_tensor_like(6) False >>> is_tensor_like(None) False >>> class NotATensor: ... >>> is_tensor_like(NotATensor()) False But, they can be made Tensor-like by implementing __torch_function__. >>> class TensorLike: ... @classmethod ... def __torch_function__(cls, func, types, args, kwargs): ... return -1 >>> is_tensor_like(TensorLike()) True r)rN r!r6r� )�inps r1rr9s(��D ��9�9�� � $� J���5I�(J�(J�Jr4c�N�eZdZUdZded<d�Zd d�Zd�Zd�Ze d ���Z dS) �TorchFunctionModea A ``TorchFunctionMode`` allows you to override the meaning of all ``__torch_function__`` overrideable functions within a dynamic scope, without having to actually create a tensor subclass or manually monkey-patch functions in the PyTorch API. Some common situations where you should use a mode: * You want to override the meaning of factory functions, or other functions that do not otherwise take a tensor as an argument (these cannot be overridden with tensor subclasses). * You want to override the behavior of all functions without needing to wrap your inputs in tensor subclasses; e.g., if you are just interested in logging intermediate computations. * You want to control the order of execution of various tensor subclasses explicitly, rather than implicitly via the return of ``NotImplemented``. Independent subclasses of :class:`TorchFunctionMode` are compositional: modes can be pushed onto a stack using ``with MyMode():``. When you call functions in the PyTorch API inside your ``__torch_function__`` implementation, by default, they will forward on to the next mode on the mode stack. If you want recursively call back into your current ``__torch_function__`` implementation, either explicitly invoke ``self.__torch_function__(...)``, or use the context manager ``enable_torch_function_mode(self, replace=self.inner)`` to make PyTorch API self-referential (beware of infinite loops, in this case!) rLc��dSr}r=rMs r1r�zTorchFunctionMode.__init__~s�� �r4r=Nc��t�r})�NotImplementedError�r�r"r� r/r0s r1rz$TorchFunctionMode.__torch_function__�s��!�!r4c�$�t|��|Sr})� _push_moderMs r1� __enter__zTorchFunctionMode.__enter__�s���4����� r4c�"�t��dSr})� _pop_mode)r��exc_type�exc_val�exc_tbs r1�__exit__zTorchFunctionMode.__exit__�s��� � � � � r4c�>�tjd��||i|��}|S)NzP`Mode.push()` is no longer necessary and can be replaced with just `with Mode()`)r+r� )�clsr/r0�instances r1�pushzTorchFunctionMode.push�s,��� �h�i�i�i��3��'��'�'���r4�r=N) r� r� � __qualname__�__doc__�__annotations__r�rr r% � classmethodr) r=r4r1r r ]s����������: ���� � � �"�"�"�"�����������[���r4r c�R�t��}|dkrt|dz ��ndS)Nrr[)rr�� stack_lens r1r� r� �s-��)�+�+�I�4=��M�M� !�)�a�-� 0� 0� 0�t�Kr4c�P�t��}d�t|��D��S)Nc�,�g|]}t|����Sr=)r)r� r� s r1r� z4_get_current_function_mode_stack.<locals>.<listcomp>�s!�� @� @� @�!� "�1� %� %� @� @� @r4)rr�r0 s r1� _get_current_function_mode_stackr4 �s)��)�+�+�I� @� @�u�Y�/?�/?� @� @� @�@r4c�$�t|��dSr})r)r�s r1r r �s��!�$�'�'�'�'�'r4c�"�t��}|Sr})r��olds r1r! r! �s�� #� %� %�C� �Jr4c#�vK�t��} |V�t|��dS#t|��wxYwr})r! r r7 s r1r� r� �s<���� �+�+�C��� � � ��3������� �3�������s�'�8c��eZdZdd�ZdS)�BaseTorchFunctionModer=Nc��|�i}||i|��Sr}r=r s r1rz(BaseTorchFunctionMode.__torch_function__�s!�� �>��F��t�T�$�V�$�$�$r4r* )r� r� r+ rr=r4r1r; r; �s(������%�%�%�%�%�%r4r; c#�K�tj���5 dV�n#wxYw ddd��dS#1swxYwYdSr})r!r� �_RestorePythonTLSSnapshotr=r4r1rr�s����� �� +� +� -� -��� � �E�E�E� ��D�D�D�D�D� ��������������������s�;�(�;�*�;�?�?)r r!r})9r, r� r� r� r� r+�typingrrrrrrr r r � contextlibr!�torch._Cr rrrrrrrrr� r r3� lru_cacherr8rrr� rrr�r�r rrr ra rrr r� r4 r r! �contextmanagerr� r; rr=r4r1�<module>rD s@����,������������ � � � �����H�H�H�H�H�H�H�H�H�H�H�H�H�H�H�H�H�H�H�H����������� � � � �%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%� � � ��U\�����"��NQ��`h�����:���T����T�s�8�}�T�T�T�����T�n���T����c�(�m�������2���T����y�t�H�h�$6�7�y�y�y�����y�v!"�H�"�"�"�"�HE�E��� �E�H�c�U�TX�[�DY�E�ei�jm�en�E�E�E�E�PL��L�-5�c�]�L�PS�L�L�L�L�\!�[������.'�;��� � ��*�k� � � � �����T���I$�E�$�s�D��N�/B�*C�T�(�TW�-�EX�*X�$Y�I$�I$�I$���I$�V� +�4��T�(�^�(;�#<� +� +� +��� +��2�2���2�&���T����S��]������� �G�x�G�D�G�G�G���G�0"K�"K�"K�H2�2�2�2�2�2�2�2�jL�L�L� A�A�A�(�(�(����  �������%�%�%�%�%�-�%�%�%� �� � ��� � � r4
Memory