
    BVhv                         d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 d Z
d Zee_        d	 Zee_        d
 Zee_         e	dg      d        Zy)z*Extending CheckpointReader for TensorFlow.    )dtypes)errors_impl)compat)CheckpointReader)	tf_exportc                    t        |       }d|v sd|v rt        j                  dd|      d|v sd|v rt        j                  dd|      d|v rt        j                  dd|      d|v rt        j
                  dd|      d|v sd	|v rt        j                  dd|      t        j                  dd|t        j                        )
z,Translate the tensor_slice_reader.cc errors.znot found in checkpointz%Failed to find any matching files forNz$Sliced checkpoints are not supportedzData type not supportedzFailed to get matching files onzUnable to open table filez&Failed to find the saved tensor slicesznot convertible to numpy dtype)	strr   NotFoundErrorUnimplementedErrorInvalidArgumentErrorDataLossErrorInternalErrorOpErrorUNKNOWN)eerror_messages     _/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/training/py_checkpoint_reader.pyerror_translatorr      s     a&--/,4- 
#
#D$
>>->#C$ 
(
(t]
CC(M9

*
*4}
EE"m3

#
#D$
>>/=@&-7

#
#D$
>>


dD-9L9L
MM    c                     | j                         j                         D ci c]  \  }}|t        j                  |       c}}S c c}}w N)_GetVariableToDataTypeMapitemsr   DType)selfname	type_enums      r   get_variable_to_dtype_mapr   0   sH     ";;=CCE

$	 FLL##
  
s    Ac                 J    | j                  t        j                  |            S r   )
_HasTensorr   as_bytes)r   
tensor_strs     r   
has_tensorr#   9   s    	4	55r   c                     	 t        j                  | t        j                  |            S # t        $ r}t        |       Y d}~yd}~ww xY w)z*Get the tensor from the Checkpoint object.N)r   CheckpointReader_GetTensorr   r!   RuntimeErrorr   )r   r"   r   s      r   
get_tensorr'   ?   sE    66fooj)+ + 
 Qs   (+ 	A	AA	ztrain.NewCheckpointReader)v1c                     	 t        t        j                  |             S # t        $ r}t	        |       Y d}~yd}~ww xY w)zA function that returns a CheckPointReader.

  Args:
    filepattern: The filename.

  Returns:
    A CheckpointReader object.
  N)r   r   r!   r&   r   )filepatternr   s     r   NewCheckpointReaderr+   Q   s8    FOOK899 
 Qs     	>9>N)__doc__tensorflow.python.frameworkr   r   tensorflow.python.utilr   0tensorflow.python.util._pywrap_checkpoint_readerr    tensorflow.python.util.tf_exportr   r   r   r#   r'   r+    r   r   <module>r2      sm    1 . 3 ) M 6N2 .G  *6 )   )   *+, -r   