
    AVh                     2    d Z ddlmZ ddlmZ ddlmZ d Zy)zHContains utility functions used by summary ops in distribution strategy.    )distribute_lib)tensor)tensor_utilc                      t        j                         } | sy| j                  }t        |t        j
                        rt        j                  |      }|xr |dkD  S )zDetermines if summary should be skipped.

  If using multiple replicas in distributed strategy, skip summaries on all
  replicas except the first one (replica_id=0).

  Returns:
    True if the summary is skipped; False otherwise.
  Fr   )r   get_replica_contextreplica_id_in_sync_group
isinstancer   Tensorr   constant_value)replica_context
replica_ids     \/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/distribute/summary_op_util.pyskip_summaryr      sR     #668/	 77*
FMM*++J7J		&
Q&    N)__doc__tensorflow.python.distributer   tensorflow.python.frameworkr   r   r    r   r   <module>r      s    O 8 . 3'r   