Searched defs:spmd_output_sharding (Results 1 – 2 of 2) sorted by relevance
456 const HloSharding& spmd_output_sharding() const { in spmd_output_sharding() function
490 xla.OpSharding spmd_output_sharding = 12; field