Searched refs:assigned_logical_device (Results 1 – 1 of 1) sorted by relevance
271 const int assigned_logical_device = in ExtractInputsForLogicalDevices() local273 (*input_list)[assigned_logical_device].emplace_back( in ExtractInputsForLogicalDevices()291 const int assigned_logical_device = sharding.tile_assignment_devices(i); in ExtractInputsForLogicalDevices() local292 (*input_list)[assigned_logical_device].emplace_back(tiled_inputs[i]); in ExtractInputsForLogicalDevices()