Searched defs:split_count (Results 1 – 4 of 4) sorted by relevance
47 split_count, argument
500 int64 split_count, absl::Span<const ReplicaGroup> replica_groups) { in AllToAll()
1993 int64 split_count) { in InferAllToAllShape()
2160 int64 concat_dimension, int64 split_count, in AllToAll()3245 int64 concat_dimension, int64 split_count, in AllToAll()