Searched refs:update_state_wrapper (Results 1 – 2 of 2) sorted by relevance
58 def update_state_wrapper(update_state_fn): function
155 metrics_utils.update_state_wrapper(update_state_fn), obj)