diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-26 14:27:25 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 14:37:18 -0700 |
| commit | 2116c6649cfe339ce8a3859eb425806db8ae32b9 (patch) | |
| tree | fa151a4dae84ad09e8ebd7c5e509c11c8c594e28 /tensorflow/core/common_runtime | |
| parent | c551a7dbd08685160c233ccecd444f774666f98e (diff) | |
Misc. micro-optimizations in Grappler optimizers.
Make shape inference lazy in optimizers that may not trigger.
PiperOrigin-RevId: 214669034
Diffstat (limited to 'tensorflow/core/common_runtime')
| -rw-r--r-- | tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | 8 |
1 files changed, 4 insertions, 4 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 efd6185f8b..2c63b8704e 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 @@ -117,15 +117,15 @@ TEST(DirectSessionWithTrackingAllocTest, CostModelTest) { // which increments the value of AllocationId. // Thus AllocationId becomes more than TF if MKL // is used. Now IDs for MKL are 8 more than TF. - EXPECT_EQ(29, cm->AllocationId(node, 0)); -#else EXPECT_EQ(21, cm->AllocationId(node, 0)); +#else + EXPECT_EQ(13, cm->AllocationId(node, 0)); #endif // INTEL_MKL && ENABLE_MKL } else { #if defined(INTEL_MKL) && defined(ENABLE_MKL) - EXPECT_EQ(30, cm->AllocationId(node, 0)); -#else EXPECT_EQ(22, cm->AllocationId(node, 0)); +#else + EXPECT_EQ(14, cm->AllocationId(node, 0)); #endif // INTEL_MKL && ENABLE_MKL } } |