Searched defs:set_batch_group_count (Results 1 – 2 of 2) sorted by relevance
1617 void set_batch_group_count(int64_t num_batch_groups) { in set_batch_group_count() function1852 void set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count() function
4761 void HloInstruction::set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count() function in xla::HloInstruction