Searched refs:is_indexed_slices (Results 1 – 3 of 3) sorted by relevance
218 def is_indexed_slices(grad): function222 grad.values if is_indexed_slices(grad) else grad
83 self.assertTrue(cross_device_utils.is_indexed_slices(t))
603 def is_indexed_slices(value): function626 if is_indexed_slices(v):