Searched defs:split_count (Results 1 – 7 of 7) sorted by relevance
43 int split_count; in __anon481f6f7f0102() local
48 split_count, argument
241 const int64 split_count = hlo->replica_groups().empty() in HandleAllToAll() local
2019 int64 split_count) { in InferAllToAllShape()
337 auto split_count = op.split_count().getSExtValue(); in Verify() local
2190 int64 concat_dimension, int64 split_count, in AllToAll()3315 int64 concat_dimension, int64 split_count, in AllToAll()
176 int split_count; member