experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update kMaxListSummarySize to 50

Browse Source 
PiperOrigin-RevId: 322601670
Change-Id: I30335a012b66366afc17dceeb19253f17bc666b8
This commit is contained in:
James Qin 2020-07-22 10:22:06 -07:00 committed by TensorFlower Gardener
parent 177790aafc
commit f97ced1b20
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/framework/attr_value_util.cc
View File

@ -278,7 +278,7 @@ string SummarizeAttrValue(const AttrValue& attr_value) {
pieces.push_back(SummarizeFunc(attr_value.list().func(i))); pieces.push_back(SummarizeFunc(attr_value.list().func(i)));
} }
} }
constexpr int kMaxListSummarySize = 15; constexpr int kMaxListSummarySize = 50;
if (pieces.size() >= kMaxListSummarySize) { if (pieces.size() >= kMaxListSummarySize) {
pieces.erase(pieces.begin() + 5, pieces.begin() + (pieces.size() - 6)); pieces.erase(pieces.begin() + 5, pieces.begin() + (pieces.size() - 6));
pieces[5] = "..."; pieces[5] = "...";

4
tensorflow/core/framework/attr_value_util_test.cc
View File

@ -160,12 +160,12 @@ TEST(AttrValueUtil, SummarizeAttrValueDoesNotElideShortLists) {
} }
TEST(AttrValueUtil, SummarizeAttrValueElidesLongLists) { TEST(AttrValueUtil, SummarizeAttrValueElidesLongLists) {
std::vector<int> alist(30); std::vector<int> alist(60);
std::iota(alist.begin(), alist.end(), 0); std::iota(alist.begin(), alist.end(), 0);
AttrValue attr_value; AttrValue attr_value;
SetAttrValue(alist, &attr_value); SetAttrValue(alist, &attr_value);
EXPECT_EQ("[0, 1, 2, 3, 4, ..., 25, 26, 27, 28, 29]", EXPECT_EQ("[0, 1, 2, 3, 4, ..., 55, 56, 57, 58, 59]",
SummarizeAttrValue(attr_value)); SummarizeAttrValue(attr_value));
} }