Searched refs:update_state_fn (Results 1 – 2 of 2) sorted by relevance
58 def update_state_wrapper(update_state_fn): argument71 update_op = update_state_fn(*args, **kwargs)76 return tf_decorator.make_decorator(update_state_fn, decorated)
150 update_state_fn = obj.update_state152 update_state_fn = def_function.function(obj.update_state)155 metrics_utils.update_state_wrapper(update_state_fn), obj)