Searched refs:set_feature_group_count (Results 1 – 8 of 8) sorted by relevance
63 custom_call->set_feature_group_count(feature_group_count); in CreateGpuConv()115 new_conv->set_feature_group_count(num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
240 new_conv->set_feature_group_count(conv->feature_group_count()); in TryRewriteToCudnnForwardRelu()
1243 void set_feature_group_count(int64 num_feature_groups) { in set_feature_group_count() function1446 void set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function
593 custom_call_instr->set_feature_group_count( in CreateFromProto()4173 void HloInstruction::set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function in xla::HloInstruction4175 return convolution->set_feature_group_count(feature_group_count); in set_feature_group_count()4177 Cast<HloCustomCallInstruction>(this)->set_feature_group_count( in set_feature_group_count()
2170 proto.set_feature_group_count(feature_group_count_); in ToProto()2421 proto.set_feature_group_count(feature_group_count_); in ToProto()2600 cloned->set_feature_group_count(feature_group_count_); in CloneWithNewOperandsImpl()
1847 void set_feature_group_count(int64 feature_group_count);
2349 custom_call_instr->set_feature_group_count(*feature_group_count); in ParseInstructionRhs()
1517 instr.set_feature_group_count(feature_group_count); in DynamicConvInstruction()1611 instr.set_feature_group_count(feature_group_count); in ConvGeneralDilatedInternal()