o
    bi                     @   s0   d dl mZ dd Zdd Zdd Zdd	 Zd
S )    )
tensorflowc                 C      t jjj| d d S )Nlogdir)tfprofilerexperimentalstartr    r
   \/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/backend/tensorflow/tensorboard.pystart_trace      r   c                 C   r   )Nsave)r   r   r   stopr   r
   r
   r   
stop_trace   r   r   c                 C   s   t jjjd| d}|  |S )NzProfiled batch)step_num)r   r   r   Trace	__enter__)batchbatch_trace_contextr
   r
   r   start_batch_trace   s
   r   c                 C   s   |  d d d  d S )N)__exit__)r   r
   r
   r   stop_batch_trace   s   r   N)keras.src.utils.module_utilsr   r   r   r   r   r   r
   r
   r
   r   <module>   s
    