当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.tpu.experimental.DeviceAssignment用法及代码示例


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

用法

tf.tpu.experimental.DeviceAssignment(
    topology:tf.tpu.experimental.Topology,
    core_assignment:np.ndarray
)

参数

  • topology 说明物理 TPU 拓扑的 Topology 对象。
  • core_assignment 逻辑到物理核心映射,表示为 rank 3 numpy 数组。有关详细信息,请参阅core_assignment 属性的说明。

抛出

  • ValueError 如果 topology 不是 Topology 对象。
  • ValueError 如果 core_assignment 不是 3 级 numpy 数组。

属性

  • core_assignment 逻辑到物理核心映射。
  • num_cores_per_replica 每个副本的核心数。
  • num_replicas 计算的副本数。
  • topology 说明 TPU 拓扑的Topology

更喜欢使用 DeviceAssignment.build() 帮助器来构造 DeviceAssignment ;如果灵活性不如直接构建DeviceAssignment,那么它会更容易。

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.tpu.experimental.DeviceAssignment。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。