Home
last modified time | relevance | path

Searched refs:_op_in_graph_mode (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py265 def _op_in_graph_mode(tensor): function
285 math_ops.is_finite(value), lambda: _op_in_graph_mode(var.assign(value)),
397 incr_loss_scale, lambda: _op_in_graph_mode(
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py72 def _op_in_graph_mode(tensor): function
92 math_ops.is_finite(value), lambda: _op_in_graph_mode(var.assign(value)),
252 lambda: _op_in_graph_mode(self.counter.assign_add(1)))