Commit 3629b2d9 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TPU] Move methods to map between TensorFlow task/device names and TPU device... 

[TPU] Move methods to map between TensorFlow task/device names and TPU device coordinates out of DeviceAssignment and into Topology.

The mapping is a property of the Topology (which describes the machine and how to address it), not the DeviceAssignment, which describes a choice of devices for a particular job.

PiperOrigin-RevId: 219141156
parent 36bf15f2
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment