Searched refs:set_feature_group_count (Results 1 – 7 of 7) sorted by relevance
62 custom_call->set_feature_group_count(feature_group_count); in CreateCudnnConv()
1891 proto.set_feature_group_count(feature_group_count_); in ToProto()2086 proto.set_feature_group_count(feature_group_count_); in ToProto()2191 cloned->set_feature_group_count(feature_group_count_); in CloneWithNewOperandsImpl()
471 ->set_feature_group_count( in CreateFromProto()3467 void HloInstruction::set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function in xla::HloInstruction3468 Cast<HloCustomCallInstruction>(this)->set_feature_group_count( in set_feature_group_count()
1561 void set_feature_group_count(int64 feature_group_count);
1203 void set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() function
1554 instruction->set_feature_group_count(*feature_group_count); in ParseInstructionRhs()
1198 instr.set_feature_group_count(feature_group_count); in ConvGeneralDilated()