Searched refs:_op_in_graph_mode (Results 1 – 2 of 2) sorted by relevance
289 def _op_in_graph_mode(tensor): function309 math_ops.is_finite(value), lambda: _op_in_graph_mode(var.assign(value)),431 incr_loss_scale, lambda: _op_in_graph_mode(
191 def _op_in_graph_mode(tensor): function211 math_ops.is_finite(value), lambda: _op_in_graph_mode(var.assign(value)),373 lambda: _op_in_graph_mode(self.counter.assign_add(1)))