Searched refs:update_state_wrapper (Results 1 – 3 of 3) sorted by relevance
62 def update_state_wrapper(update_state_fn): function
896 metric.update_state = types.MethodType(metrics_utils.update_state_wrapper(
187 metrics_utils.update_state_wrapper(update_state_fn), obj)