diff options
author | 2018-09-04 18:10:09 -0700 | |
---|---|---|
committer | 2018-09-04 18:13:56 -0700 | |
commit | fd28fee75f141345c3e862bc1115ff4a2b478eb0 (patch) | |
tree | 2f945618866880bdc45830b25ae5912f38d170d6 /tensorflow/compiler/xla/service/hlo_instructions.cc | |
parent | bfde272cf661d942b11877a8709739a09c5d41fd (diff) |
[XLA] Don't show trivial feature_group_count attributes
If the feature_group_count is 1, don't bother showing it as it is not very
informative and a very common scenario. This is consistent with the
HloCustomCall's feature_group_count attribute.
PiperOrigin-RevId: 211560372
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instructions.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_instructions.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instructions.cc b/tensorflow/compiler/xla/service/hlo_instructions.cc index bed273149b..e3683aaec9 100644 --- a/tensorflow/compiler/xla/service/hlo_instructions.cc +++ b/tensorflow/compiler/xla/service/hlo_instructions.cc @@ -1674,7 +1674,9 @@ std::vector<string> HloConvolutionInstruction::ExtraAttributesToStringImpl( } extra.push_back(StrCat("dim_labels=", ConvolutionDimensionNumbersToString( convolution_dimension_numbers_))); - extra.push_back(StrCat("feature_group_count=", feature_group_count_)); + if (feature_group_count_ != 1) { + extra.push_back(StrCat("feature_group_count=", feature_group_count_)); + } return extra; } |