From 6fa593f46d4efc8138a9c76ebcc9c906b82a3e04 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <bsteiner@google.com>
Date: Sat, 30 Jul 2016 07:23:05 -0800
Subject: [PATCH] Mark the ExpandDimOp and the SqueezeOp as non expensive
 Change: 128892723

---
 tensorflow/core/kernels/shape_ops.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/core/kernels/shape_ops.cc b/tensorflow/core/kernels/shape_ops.cc
index 0861fa99821..63ad0059d45 100644
--- a/tensorflow/core/kernels/shape_ops.cc
+++ b/tensorflow/core/kernels/shape_ops.cc
@@ -253,6 +253,8 @@ class ExpandDimsOp : public OpKernel {
                            " and output shape ", output_shape.DebugString()));
     }
   }
+
+  bool IsExpensive() override { return false; }
 };
 REGISTER_KERNEL_BUILDER(Name("ExpandDims").Device(DEVICE_CPU).HostMemory("dim"),
                         ExpandDimsOp);
@@ -342,6 +344,8 @@ class SqueezeOp : public OpKernel {
     }
   }
 
+  bool IsExpensive() override { return false; }
+
  private:
   std::unordered_set<int32> squeeze_dims_;
 };