Searched refs:feature_group_count_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 2139 feature_group_count_(feature_group_count), in HloConvolutionInstruction()
2170 proto.set_feature_group_count(feature_group_count_); in ToProto()
2184 if (feature_group_count_ != 1) { in ExtraAttributesToStringImpl()
2185 extra.push_back(StrCat("feature_group_count=", feature_group_count_)); in ExtraAttributesToStringImpl()
2205 if (feature_group_count_ != other.feature_group_count()) { in IdenticalSlowPath()
2225 shape, new_operands[0], new_operands[1], feature_group_count_, in CloneWithNewOperandsImpl()
2343 feature_group_count_(1), in HloCustomCallInstruction()
2360 feature_group_count_(1), in HloCustomCallInstruction()
2378 feature_group_count_(1), in HloCustomCallInstruction()
2398 feature_group_count_(1), in HloCustomCallInstruction()
[all …]
|
| D | hlo_instructions.h | 1242 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count()
1244 feature_group_count_ = num_feature_groups; in set_feature_group_count()
1279 int64 feature_group_count_; variable
1447 feature_group_count_ = feature_group_count; in set_feature_group_count()
1457 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count()
1524 int64 feature_group_count_; variable
|