Searched defs:group_dim_shards (Results 1 – 3 of 3) sorted by relevance
213 absl::InlinedVector<int64_t, 1> group_dim_shards = { in PartitionGatherIndexPassthroughPartition() local319 absl::InlinedVector<int64_t, 1> group_dim_shards = { in PartitionGatherPassthroughOperand() local
2329 std::vector<int64_t> group_dim_shards = { in GetNonContractingPartitionGroupedShardingForOtherOperand() local2472 std::vector<int64_t> group_dim_shards = { in GetDotGroupPartitionContractingOutputShardings() local
1628 std::vector<int64_t> group_dim_shards(group_dims.size(), 1); in GroupShardingOnDims() local1635 absl::Span<const int64_t> group_dim_shards, in GroupShardingOnDims()