Home
last modified time | relevance | path

Searched defs:group_dim_sizes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h309 std::vector<int64> group_dim_sizes; member
Dspmd_partitioner_util.cc1390 std::vector<int64> group_dim_sizes(group_dims.size()); in GroupShardingOnDims() local
1696 absl::Span<const int64> group_dims, absl::Span<const int64> group_dim_sizes, in PerGroupSliceFromReplicated()