Searched refs:kTensorCore (Results 1 – 3 of 3) sorted by relevance
105 int num_local_devices = host.Cores(kTensorCore).size(); in GetDefaultDeviceAssignment()154 topology.cores(TpuCoreTypeEnum::kTensorCore)) { in GetTpuDevices()
34 kTensorCore, enumerator
1410 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseTopologyAttr()1465 int devices_per_chip = tpu_topology.LogicalDevicesPerChip(kTensorCore); in ParseDeviceAssignmentAttr()1485 tpu_topology.Core(kTensorCore, x, y, z, core); in ParseDeviceAssignmentAttr()3962 tpu_topology.LogicalDevicesPerHost(kTensorCore)); in GetDeviceTopology()