
    g
                     j    d dl Z d dlmZ  e       Z G d d      Zd Zd Zd Zd Zd	 Z	d
 Z
d Zd Zy)    N)import_numpyc                   N    e Zd ZdgZed	d       Zed	d       Zd Zd Zd Z	d Z
y)
TrtTable_tabc                     t         j                  j                  t         j                  j                  ||      }t               }|j                  |||z          |S N)flatbuffersencodeGetpackeruoffsetr   Init)clsbufoffsetnxs        j/var/www/openai/venv/lib/python3.12/site-packages/onnxruntime/quantization/CalTableFlatBuffers/TrtTable.py	GetRootAszTrtTable.GetRootAs   sF    "";#5#5#=#=sFKJ	sAJ    c                 &    | j                  ||      S )z6This method is deprecated. Please switch to GetRootAs.)r   )r   r   r   s      r   GetRootAsTrtTablezTrtTable.GetRootAsTrtTable   s     }}S&))r   c                 N    t         j                  j                  ||      | _        y r   )r	   tableTabler   )selfr   poss      r   r   zTrtTable.Init   s    %%++C5	r   c                    t         j                  j                  j                  | j                  j                  d            }|dk7  r| j                  j                  |      }|t         j                  j                  j                  |      dz  z  }| j                  j                  |      }ddlm	}  |       }|j                  | j                  j                  |       |S y )N   r   )KeyValue)r	   number_typesUOffsetTFlagspy_typer   OffsetVectorIndirect5onnxruntime.quantization.CalTableFlatBuffers.KeyValuer    r   Bytes)r   jor   r    objs         r   DictzTrtTable.Dict   s    $$22::499;K;KA;NO6		  #A))77??BQFFA		""1%AV*CHHTYY__a(Jr   c                     t         j                  j                  j                  | j                  j                  d            }|dk7  r| j                  j                  |      S yNr   r   )r	   r!   r"   r#   r   r$   	VectorLenr   r*   s     r   
DictLengthzTrtTable.DictLength-   sL    $$22::499;K;KA;NO699&&q))r   c                     t         j                  j                  j                  | j                  j                  d            }|dk(  S r.   )r	   r!   r"   r#   r   r$   r0   s     r   
DictIsNonezTrtTable.DictIsNone4   s6    $$22::499;K;KA;NOAvr   N)r   )__name__
__module____qualname__	__slots__classmethodr   r   r   r,   r1   r3    r   r   r   r      sC    I  * *
6r   r   c                 &    | j                  d       y )N   )StartObjectbuilders    r   Startr?   9   s    r   c                     t        |       S z2This method is deprecated. Please switch to Start.)r?   r=   s    r   TrtTableStartrB   =   s    >r   c                 x    | j                  dt        j                  j                  j	                  |      d       y )Nr   )PrependUOffsetTRelativeSlotr	   r!   r"   r#   r>   dicts     r   AddDictrG   B   s-    '';+C+C+Q+Q+Y+YZ^+_abcr   c                     t        | |      S )z4This method is deprecated. Please switch to AddDict.)rG   rE   s     r   TrtTableAddDictrI   F   s    7D!!r   c                 (    | j                  d|d      S )Nr   )StartVectorr>   numElemss     r   StartDictVectorrN   K   s    q(A..r   c                     t        | |      S rA   )rN   rL   s     r   TrtTableStartDictVectorrP   O   s    7H--r   c                 "    | j                         S r   )	EndObjectr=   s    r   EndrS   T   s    r   c                     t        |       S )z0This method is deprecated. Please switch to End.)rS   r=   s    r   TrtTableEndrU   X   s    w<r   )r	   flatbuffers.compatr   npr   r?   rB   rG   rI   rN   rP   rS   rU   r9   r   r   <module>rX      sH   
  +^+ +\
d"
/.
r   