Searched defs:set_batch_group_count (Results 1 – 2 of 2) sorted by relevance
1206 void set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function
3479 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function in xla::HloInstruction