Searched refs:feature_group_count_ (Results 1 – 2 of 2) sorted by relevance
1860 feature_group_count_(feature_group_count), in HloConvolutionInstruction()1891 proto.set_feature_group_count(feature_group_count_); in ToProto()1905 if (feature_group_count_ != 1) { in ExtraAttributesToStringImpl()1906 extra.push_back(StrCat("feature_group_count=", feature_group_count_)); in ExtraAttributesToStringImpl()1927 if (feature_group_count_ != other.feature_group_count()) { in IdenticalSlowPath()1947 shape, new_operands[0], new_operands[1], feature_group_count_, in CloneWithNewOperandsImpl()2050 feature_group_count_(1), in HloCustomCallInstruction()2065 feature_group_count_(1), in HloCustomCallInstruction()2086 proto.set_feature_group_count(feature_group_count_); in ToProto()2108 if (feature_group_count_ != 1) { in ExtraAttributesToStringImpl()[all …]
1047 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count()1080 int64 feature_group_count_; variable1204 feature_group_count_ = feature_group_count; in set_feature_group_count()1209 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count()1244 int64 feature_group_count_; variable