此页面由 Cloud Translation API 翻译。
Switch to English

tf.tpu.experimental.DeviceAssignment

TensorFlow 1版 GitHub上查看源代码

从逻辑核心中的计算到物理TPU拓扑映射。

喜欢使用DeviceAssignment.build()辅助以构建DeviceAssignment ;它是比如果构建一个较不灵活容易DeviceAssignment直接。

topology 一个Topology描述物理拓扑TPU对象。
core_assignment 甲逻辑到物理映射的核心,表示为秩3 numpy的阵列。见的描述core_assignment属性的更多细节。

ValueError 如果topology是不是Topology对象。
ValueError 如果core_assignment不是秩3 numpy的阵列。

core_assignment 逻辑到物理映射的核心。
num_cores_per_replica 每个副本核心的数目。
num_replicas 计算的副本的数量。
topology 一个Topology描述TPU拓扑。

方法

build

查看源代码

coordinates

查看源代码

返回一个逻辑核心的物理拓扑坐标。

host_device

查看源代码

返回附连到一个逻辑核心的CPU的设备。

lookup_replicas

查看源代码

通过任务号和逻辑核心查找副本标识。

ARGS
task_id TensorFlow任务数。
logical_core 的整数,确定逻辑核心。

返回
一个有序附加到该任务和logical_core副本列表。

加薪
ValueError 如果没有副本存在,其中包含逻辑核心的任务。

tpu_device

查看源代码

返回分配给一个逻辑核心的TPU设备的名称。

tpu_ordinal

查看源代码

返回分配给一个逻辑核心的TPU装置的序号。