Searched defs:set_feature_group_count (Results 1 – 2 of 2) sorted by relevance
1203 void set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function
3467 void HloInstruction::set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function in xla::HloInstruction