Searched defs:devices_per_chip (Results 1 – 2 of 2) sorted by relevance
1620 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseTopologyAttr() local1675 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseDeviceAssignmentAttr() local
89 int devices_per_chip = in GetTPUTopology() local