From 349c26fe528c0f2ef68291022921d8458a912965 Mon Sep 17 00:00:00 2001 From: Ankur Taly Date: Wed, 23 May 2018 14:34:23 -0700 Subject: [PATCH] fixed DirectSessionWithTrackingAllocTest keyboard_arrow_right CostModelTest --- .../core/common_runtime/direct_session_with_tracking_alloc_test.cc | 4 ++-- 1 file 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 e192a95..9028e62 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 @@ -113,13 +113,13 @@ TEST(DirectSessionWithTrackingAllocTest, CostModelTest) { // MKL is used. Now they are 9 and 10 for MKL. EXPECT_EQ(19, cm->AllocationId(node, 0)); #else - EXPECT_EQ(13, cm->AllocationId(node, 0)); + EXPECT_EQ(21, cm->AllocationId(node, 0)); #endif } else { #ifdef INTEL_MKL EXPECT_EQ(20, cm->AllocationId(node, 0)); #else - EXPECT_EQ(14, cm->AllocationId(node, 0)); + EXPECT_EQ(22, cm->AllocationId(node, 0)); #endif } } -- 2.7.4