Searched refs:assigned_core (Results 1 – 1 of 1) sorted by relevance
1989 absl::optional<int64> assigned_core; in AssignArgsAndRetvalsToCores() local2058 assigned_core = args_device_selector.RetrieveAssignment(i); in AssignArgsAndRetvalsToCores()2060 assigned_core = 0; in AssignArgsAndRetvalsToCores()2064 xla::sharding_builder::AssignDevice(*assigned_core)); in AssignArgsAndRetvalsToCores()2069 assigned_core = node_and_sharding->sharding.tile_assignment_devices(0); in AssignArgsAndRetvalsToCores()2077 if (assigned_core.has_value()) { in AssignArgsAndRetvalsToCores()2078 args_device_selector.ReportDeviceAssigned(*assigned_core, i); in AssignArgsAndRetvalsToCores()2080 << ") to core " << *assigned_core in AssignArgsAndRetvalsToCores()2082 args[i]->set_assigned_device_name(CoreDeviceLabel(*assigned_core)); in AssignArgsAndRetvalsToCores()2144 absl::optional<int64> assigned_core; in AssignArgsAndRetvalsToCores() local[all …]