/usr/local/lib/python3.6/site-packages/transformers/onnx/__pycache__
Edit: /usr/local/lib/python3.6/site-packages/transformers/onnx/__pycache__/utils.cpython-36.pyc (1524B)
3
øEgÔ ã @ sX d dl mZmZ d dlmZ G dd„ deƒZdeeeedœdd„Zeeedœd d
„ZdS )
é )Úc_floatÚsizeof)ÚEnumc @ s" e Zd ZeZeedœdd„ƒZdS )ÚParameterFormat)Úreturnc C s
t | jƒS )zf
Number of byte required for this data type
Returns:
Integer > 0
)r Úvalue)Úself© r úA/usr/local/lib/python3.6/site-packages/transformers/onnx/utils.pyÚsize s zParameterFormat.sizeN)Ú__name__Ú
__module__Ú__qualname__r ZFloatÚpropertyÚintr r r r r
r s r )Ú dimensionÚfixed_dimensionÚnum_token_to_addr c C s | dkr|} | |8 } | S )ze
Args:
dimension:
fixed_dimension:
num_token_to_add:
Returns:
r r )r r r r r r
Ú compute_effective_axis_dimension! s r )Únum_parametersÚdtyper c C s
| |j S )a7
Compute the size taken by all the parameters in the given the storage format when serializing the model
Args:
num_parameters: Number of parameters to be saved
dtype: The data format each parameter will be saved
Returns:
Size (in byte) taken to save all the parameters
)r )r r r r r
Ú"compute_serialized_parameters_size4 s r N)r ) Úctypesr r Úenumr r r r r r r r r
Ú
s