Home
last modified time | relevance | path

Searched refs:_devices_match (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py171 def _devices_match(left, right): function
177 if not all(_devices_match(v, d) for v, d in value_destination_pairs):
179 if not all(_devices_match(v, value_destination_pairs[0][0])
292 per_replica_value.values) == 1 and _devices_match(
341 per_replica_value.values) == 1 and _devices_match(
841 if (_devices_match(per_replica_value, destinations) and
1137 if _devices_match(per_replica_value, destinations):
1309 if _devices_match(per_replica_value, destinations):
Dmirrored_strategy.py732 …(not cross_device_ops_lib._devices_match(value, destinations) or # pylint: disable=protected-acce…