Searched defs:num_tpu_devices_per_task (Results 1 – 2 of 2) sorted by relevance
37 int32 num_tpu_devices_per_task = 3; field
123 int num_tpu_devices_per_task) { in TopologyWithMeshShapeAndTasks()