o
    ?e                     @   s&   d Z ddlmZ G dd dejZdS )z(`Callable` class used for checkpointing.    )saveable_objectc                       s    e Zd ZdZ fddZ  ZS )Callablea  A callable that represents a Tensor that should be saved to checkpoint.

  This can be returned from `_serialize_to_tensor` in place of a Tensor. The
  callable will be executed on the specified device when the checkpoint is
  about to be written.

  Any class can use `Callable` for checkpointing, but for SavedModel export,
  only resource-type variables* are supported.

  * `resource_variable_ops.is_resource_variable(obj)` must return True.
  c                    s   t  |dd|| dS )zInitializes a `Callable` object.

    Args:
      tensor_callable: A callable that takes no arguments and returns a Tensor.
      dtype: Dtype of the tensor returned by the callable.
      device: Device of the tensor returned by the callable.
    N)super__init__)selfZtensor_callableZdtypeZdevice	__class__ m/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/checkpoint/tensor_callable.pyr   !   s   zCallable.__init__)__name__
__module____qualname____doc__r   __classcell__r	   r	   r   r
   r      s    r   N)r   Z!tensorflow.python.training.savingr   ZSaveSpecr   r	   r	   r	   r
   <module>   s   