Searched defs:assigned_logical_device (Results 1 – 1 of 1) sorted by relevance
275 const int assigned_logical_device = in ExtractInputsForLogicalDevices() local295 const int assigned_logical_device = sharding.tile_assignment_devices(i); in ExtractInputsForLogicalDevices() local