Searched defs:logical_device (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 44 def __init__(self, device_map, worker_device_pairs=None, logical_device=0): argument
75 def logical_device(self): member in InputWorkers
|
| D | values.py | 235 def __init__(self, device_map, values, logical_device=None): argument
268 def logical_device(self): member in DistributedValues
444 def __init__(self, strategy, device_map, values, logical_device=None): argument
652 self, strategy, device_map, values, aggregation, logical_device=None): argument
774 self, strategy, device_map, values, aggregation, logical_device=None): argument
821 def logical_device(self): member in TPUMirroredVariable
1232 self, strategy, device_map, values, aggregation, logical_device=None): argument
|
| D | tpu_strategy.py | 71 strategy, device_map, logical_device, real_mirrored_creator, argument
|
| D | mirrored_strategy.py | 198 def _create_mirrored_variable(strategy, device_map, logical_device, # pylint: disable=missing-docs… argument
|