Searched defs:set_batch_group_count (Results 1 – 2 of 2) sorted by relevance
1248 void set_batch_group_count(int64 num_batch_groups) { in set_batch_group_count() function1449 void set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function
4188 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function in xla::HloInstruction