o ?ev@sdZddlmZddlmZddlmZddlmZddlm Z ddZ d d Z e e_ d d Z e e_ d dZ e e_ e dgdddZdS)z*Extending CheckpointReader for TensorFlow.)dtypes) errors_impl)compat)CheckpointReader) tf_exportcCst|}d|vs d|vrtdd|d|vsd|vr"tdd|d|vr-tdd|d|vr8tdd|d|vs@d |vrGtdd|tdd|tj) z,Translate the tensor_slice_reader.cc errors.znot found in checkpointz%Failed to find any matching files forNz$Sliced checkpoints are not supportedzData type not supportedzFailed to get matching files onzUnable to open table filez&Failed to find the saved tensor slicesznot convertible to numpy dtype) strrZ NotFoundErrorZUnimplementedErrorZInvalidArgumentErrorZ DataLossErrorZ InternalErrorZOpErrorUNKNOWN)e error_messager p/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/training/py_checkpoint_reader.pyerror_translators(r cCsdd|DS)NcSsi|] \}}|t|qSr )rZDType).0nameZ type_enumr r r 1s z-get_variable_to_dtype_map..)Z_GetVariableToDataTypeMapitems)selfr r r get_variable_to_dtype_map0s rcCs|t|S)N)Z _HasTensorras_bytes)r tensor_strr r r has_tensor9src CsDz t|t|WSty!}z t|WYd}~dSd}~ww)z*Get the tensor from the Checkpoint object.N)rZCheckpointReader_GetTensorrr RuntimeErrorr )rrr r r r get_tensor?s rztrain.NewCheckpointReader)v1c Cs@ztt|WSty}z t|WYd}~dSd}~ww)zA function that returns a CheckPointReader. Args: filepattern: The filename. Returns: A CheckpointReader object. N)rrrrr )Z filepatternr r r r NewCheckpointReaderQs  rN)__doc__Ztensorflow.python.frameworkrrZtensorflow.python.utilrZ0tensorflow.python.util._pywrap_checkpoint_readerrZ tensorflow.python.util.tf_exportrr rrrrr r r r s