Searched refs:assigned_logical_device (Results 1 – 1 of 1) sorted by relevance
277 const int assigned_logical_device = in ExtractInputsForLogicalDevices() local279 (*input_list)[assigned_logical_device].emplace_back( in ExtractInputsForLogicalDevices()297 const int assigned_logical_device = sharding.tile_assignment_devices(i); in ExtractInputsForLogicalDevices() local298 (*input_list)[assigned_logical_device].emplace_back(tiled_inputs[i]); in ExtractInputsForLogicalDevices()