Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_utils_test.py159 self.assertTrue(distribute_utils.is_distributed_variable(v))
160 self.assertTrue(distribute_utils.is_distributed_variable(
H A Ddistribute_utils.py221 def is_distributed_variable(v): function
H A Dvalues.py1041 def is_distributed_variable(self): member in DistributedVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A 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)):
H A Doptimizer_test.py283 self.assertTrue(distribute_utils.is_distributed_variable(v))
290 self.assertTrue(distribute_utils.is_distributed_variable(slot))
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/
H A Ddistributed_training_utils.py62 def is_distributed_variable(v): function
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A 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) \
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py511 if not distributed_training_utils.is_distributed_variable(variable):
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A 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()