Python MeshTensorFlow是一种用于分布式深度学习的语言能够指定一大类分布式张量计算
Mesh TensorFlow的目的是为您的硬件/处理器上的计算图形化和实现分发策略。 例如:“将批处理拆分为多行处理器,并将处理列中的单元拆分为隐藏层。” Mesh TensorFlow实现为TensorFlow上的一个层。
文件列表
Python-MeshTensorFlow是一种用于分布式深度学习的语言能够指定一大类分布式张量计算.zip
(预估有个33文件)
tensorflow-mesh-12743d8
.travis.yml
482B
mesh_tensorflow
placement_mesh_impl.py
18KB
simd_mesh_impl.py
16KB
utils.py
2KB
import_test.py
944B
transformer
moe.py
30KB
transformer_layers.py
9KB
暂无评论