diff options
author | 2018-05-14 10:43:08 -0700 | |
---|---|---|
committer | 2018-05-14 10:45:35 -0700 | |
commit | 5fb7401959391f7583087f404a48353ab21ef1ca (patch) | |
tree | 7413cfeec40ad33a8c4468219bbdf234fe9a6ed4 /tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | |
parent | 157c347f832413c29265e467cc733366b4b215a6 (diff) |
Use utility methods to compute AttrValue hash code and check for equality.
PiperOrigin-RevId: 196531355
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc index 695423b2cb..95093beced 100644 --- a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc +++ b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc @@ -102,9 +102,9 @@ TEST(DirectSessionWithTrackingAllocTest, CostModelTest) { EXPECT_EQ(2, shape.dim(0).size()); EXPECT_EQ(1, shape.dim(1).size()); if (node->name() == y->name()) { - EXPECT_EQ(9, cm->AllocationId(node, 0)); + EXPECT_EQ(13, cm->AllocationId(node, 0)); } else { - EXPECT_EQ(10, cm->AllocationId(node, 0)); + EXPECT_EQ(14, cm->AllocationId(node, 0)); } } EXPECT_LE(0, cm->MaxExecutionTime(node)); |