Home
last modified time | relevance | path

Searched refs:contains_indexed_slices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_utils_test.py82 self.assertTrue(cross_device_utils.contains_indexed_slices(t))
90 self.assertTrue(cross_device_utils.contains_indexed_slices([t0, t1]))
98 self.assertTrue(cross_device_utils.contains_indexed_slices((t0, t1)))
108 self.assertTrue(cross_device_utils.contains_indexed_slices(per_replica))
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py676 contains_indexed_slices = cross_device_utils.contains_indexed_slices(
679 and not contains_indexed_slices):
975 if cross_device_utils.contains_indexed_slices(per_replica_value):
998 if cross_device_utils.contains_indexed_slices(value_destination_pairs):
Dcross_device_utils.py674 def contains_indexed_slices(value): function
679 return any(contains_indexed_slices(v) for v in value)
681 return contains_indexed_slices(value.values)