Searched refs:devices_per_chip (Results 1 – 1 of 1) sorted by relevance
1410 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseTopologyAttr() local1413 tpu_topology.chip_bounds().z, devices_per_chip, {-1, -1}); in ParseTopologyAttr()1423 core >= devices_per_chip) { in ParseTopologyAttr()1465 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseDeviceAssignmentAttr() local1468 tpu_topology.chip_bounds().z, devices_per_chip, -1); in ParseDeviceAssignmentAttr()1479 core >= devices_per_chip) { in ParseDeviceAssignmentAttr()