Home
last modified time | relevance | path

Searched refs:AutoCastDistributedVariable (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dautocast_variable.py170 class AutoCastDistributedVariable(AutoCastVariable, class
178 super(AutoCastDistributedVariable, self).__init__(variable)
Dautocast_variable_test.py61 return autocast_variable.AutoCastDistributedVariable(var)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py383 variable = autocast_variable.AutoCastDistributedVariable(variable)