Searched refs:set_feature_group_count (Results 1 – 7 of 7) sorted by relevance
583 custom_call->set_feature_group_count(feature_group_count); in CreateGpuConv()651 new_conv->set_feature_group_count(num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
1612 void set_feature_group_count(int64_t num_feature_groups) { in set_feature_group_count() function1849 void set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count() function
764 custom_call_instr->set_feature_group_count(std::max( in CreateFromProto()4746 void HloInstruction::set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count() function in xla::HloInstruction4748 return convolution->set_feature_group_count(feature_group_count); in set_feature_group_count()4750 Cast<HloCustomCallInstruction>(this)->set_feature_group_count( in set_feature_group_count()
2468 proto.set_feature_group_count(feature_group_count_); in ToProto()2722 proto.set_feature_group_count(feature_group_count_); in ToProto()2922 cloned->set_feature_group_count(feature_group_count_); in CloneWithNewOperandsImpl()
2063 void set_feature_group_count(int64_t feature_group_count);
2708 custom_call_instr->set_feature_group_count(*feature_group_count); in CreateInstruction()
1597 instr.set_feature_group_count(feature_group_count); in DynamicConvInstruction()1696 instr.set_feature_group_count(feature_group_count); in ConvGeneralDilatedInternal()