Home
last modified time | relevance | path

Searched refs:is_distributed_variable (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py62 def is_distributed_variable(v): function
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_utils_test.py159 self.assertTrue(distribute_utils.is_distributed_variable(v))
160 self.assertTrue(distribute_utils.is_distributed_variable(
Ddistribute_utils.py221 def is_distributed_variable(v): function
Dvalues.py1041 def is_distributed_variable(self): member in DistributedVariable
/external/tensorflow/tensorflow/python/training/
Doptimizer_test.py283 self.assertTrue(distribute_utils.is_distributed_variable(v))
290 self.assertTrue(distribute_utils.is_distributed_variable(slot))
Doptimizer.py84 if (distribute_utils.is_distributed_variable(var) and
859 if (distribute_utils.is_distributed_variable(slot) and
860 not distribute_utils.is_distributed_variable(var)):
/external/tensorflow/tensorflow/python/saved_model/
Dload.py119 return x.handle if distribute_utils.is_distributed_variable(x) else x
122 return _unused_handle() if distribute_utils.is_distributed_variable(x) \
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py511 if not distributed_training_utils.is_distributed_variable(variable):
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2599 bool is_distributed_variable = in BuildTPUReplicateOps() local
2603 if (!is_distributed_variable && input_num_replicas != num_replicas) { in BuildTPUReplicateOps()
2610 if (!is_distributed_variable) { in BuildTPUReplicateOps()