Searched refs:core_location (Results 1 – 2 of 2) sorted by relevance
36 explicit TpuCoreLocationExternal(SE_TpuTopology_Core* core_location) in TpuCoreLocationExternal() argument37 : core_location_(core_location) {} in TpuCoreLocationExternal()
1484 tpu::TpuCoreLocationExternal core_location = in ParseDeviceAssignmentAttr() local1493 (*device_assignment)(replica, logical_core) = core_location; in ParseDeviceAssignmentAttr()1537 const tpu::TpuCoreLocationExternal& core_location = in BuildGeneralDeviceAssignment() local1543 topology(core_location.chip_coordinates().x, in BuildGeneralDeviceAssignment()1544 core_location.chip_coordinates().y, in BuildGeneralDeviceAssignment()1545 core_location.chip_coordinates().z, core_location.index()); in BuildGeneralDeviceAssignment()1548 (**xla_device_assignment)(replica, computation) = core_location.Id(); in BuildGeneralDeviceAssignment()