Searched refs:num_cores_per_host (Results 1 – 6 of 6) sorted by relevance
208 const int32_t* num_cores_per_host; member228 size_t num_cores_per_host; member
167 params.num_cores_per_host = num_devices_per_host.data(); in ConstructTpuPodState()
215 params.num_cores_per_host = num_devices_per_host; in Compute()
47 int32 num_cores_per_host = 7; field
268 report.config.num_cores_per_host = tt_config.num_replicas_per_host
1389 def num_cores_per_host(self): member in TPUEmbedding