Searched defs:hlo_sharding (Results 1 – 3 of 3) sorted by relevance
770 py::class_<HloSharding> hlo_sharding(m, "HloSharding"); in BuildXlaCompilerSubmodule() local
107 std::optional<xla::HloSharding> hlo_sharding; in MaybeRewriteLayoutWithShardedShape() local
2364 auto hlo_sharding = xla::HloSharding::FromProto(*arg_sharding.value()); in SetEntryTupleShardings() local