Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_utils_test.py161 self.assertTrue(distribute_utils.is_distributed_variable(v))
162 self.assertTrue(distribute_utils.is_distributed_variable(
Ddistribute_utils.py219 def is_distributed_variable(v): function
/external/tensorflow/tensorflow/python/saved_model/
Dload.py96 return x.handle if distribute_utils.is_distributed_variable(x) else x
99 return _unused_handle() if distribute_utils.is_distributed_variable(x) \
353 if distribute_utils.is_distributed_variable(bound_input):
385 if distribute_utils.is_distributed_variable(obj):
/external/tensorflow/tensorflow/python/training/
Doptimizer_test.py287 self.assertTrue(distribute_utils.is_distributed_variable(v))
294 self.assertTrue(distribute_utils.is_distributed_variable(slot))
Doptimizer.py89 if (distribute_utils.is_distributed_variable(var) and
766 if (distribute_utils.is_distributed_variable(slot) and
767 not distribute_utils.is_distributed_variable(var)):
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py516 if (not distribute_utils.is_distributed_variable(variable) and
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2647 bool is_distributed_variable = in BuildTPUReplicateOps() local
2651 if (!is_distributed_variable && input_num_replicas != num_replicas) { in BuildTPUReplicateOps()
2658 if (!is_distributed_variable) { in BuildTPUReplicateOps()
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py2700 distribute_utils.is_distributed_variable(model.optimizer.weights[0]))
2708 distribute_utils.is_distributed_variable(model.optimizer.weights[0]))