Searched defs:subsharding (Results 1 – 3 of 3) sorted by relevance
132 absl::optional<xla::OpSharding> subsharding = in ParseShardingFromEdgeSource() local
187 auto subsharding = sharding ? sharding->tuple_shardings(i) : sharding; in ReshapeWithCorrectRepresentationAndSharding() local
1101 for (const auto& subsharding : sharding.tuple_elements()) { in DevicesForShardingInternal() local