1VhrdZddlZddlmZeZGddZd dZda ejZ dZ y) z6Provides a lazy wrapper for deferring Tensor creation.N)tf2ceZdZdZdZdZy)LazyTensorCreatoraLazy auto-converting wrapper for a callable that returns a `tf.Tensor`. This class wraps an arbitrary callable that returns a `Tensor` so that it will be automatically converted to a `Tensor` by any logic that calls `tf.convert_to_tensor()`. This also memoizes the callable so that it is called at most once. The intended use of this class is to defer the construction of a `Tensor` (e.g. to avoid unnecessary wasted computation, or ensure any new ops are created in a context only available later on in execution), while remaining compatible with APIs that expect to be given an already materialized value that can be converted to a `Tensor`. This class is thread-safe. ct|std|z||_d|_t j |_ty)zInitializes a LazyTensorCreator object. Args: tensor_callable: A callable that returns a `tf.Tensor`. zNot a callable: %rN)callable ValueError_tensor_callable_tensor threadingRLock _tensor_lock"_register_conversion_function_once)selftensor_callables T/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/util/lazy_tensor_creator.py__init__zLazyTensorCreator.__init__,sA (1OCD D / %OO-*,cX|j|jturj|j5|jtur td|j t|_|j |_ddd|jS|jS#1swY|jSxYw)Nz4Cannot use LazyTensorCreator with reentrant callable)r _CALL_IN_PROGRESS_SENTINELr RuntimeErrorr )rs r__call__zLazyTensorCreator.__call__9s << 4<<3M#M"" ;<<#==&N\\)#=DL#'#8#8#:DL ;||t|| ;||s A BB)N)__name__ __module__ __qualname____doc__rrrrrrs - rrFc~t|tstd|z|r td|}|d|jfvrtd|jd||S)Nz"Expected LazyTensorCreator, got %rz1Cannot use LazyTensorCreator to create ref tensorz1Cannot convert LazyTensorCreator returning dtype z to dtype ) isinstancerrdtype)valuernameas_reftensors r_lazy_tensor_creator_converterr$Fsj e. /?%GHH NOO WF T6<<((||U $   Mrcts8t5ts"datjtt ddddyy#1swYyxYw)aPerforms one-time registration of `_lazy_tensor_creator_converter`. This helper can be invoked multiple times but only registers the conversion function on the first invocation, making it suitable for calling when constructing a LazyTensorCreator. Deferring the registration is necessary because doing it at at module import time would trigger the lazy TensorFlow import to resolve, and that in turn would break the delicate `tf.summary` import cycle avoidance scheme. Tr) base_typeconversion_funcpriorityN)_conversion_registered_conversion_registered_locktf#register_tensor_conversion_functionrr$rrrrr[sI " ( ))-&66/$B   "  s )AA )NNF) rr tensorboard.compatrr+objectrrr$r)Lockr*rrrrr0sF=( $X((V ",inn.r