Searched defs:set_feature_group_count (Results 1 – 2 of 2) sorted by relevance
1312 void set_feature_group_count(int64_t num_feature_groups) { in set_feature_group_count() function1531 void set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count() function
4471 void HloInstruction::set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count() function in xla::HloInstruction