Searched defs:set_batch_group_count (Results 1 – 2 of 2) sorted by relevance
1317 void set_batch_group_count(int64_t num_batch_groups) { in set_batch_group_count() function1534 void set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count() function
4486 void HloInstruction::set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count() function in xla::HloInstruction