Home
last modified time | relevance | path

Searched refs:mesh_split_sharding (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py409 def mesh_split_sharding(device_mesh, tensor_split_dims_mapping): function
469 sharding = mesh_split_sharding(device_mesh, tensor_split_dims_mapping)