Searched defs:num_tpus_per_task (Results 1 – 2 of 2) sorted by relevance
214 def num_tpus_per_task(self): member in Topology
170 const int num_tpus_per_task = tpu_devices[0].size(); in GetTPUCompilationAndExecutionDevices() local