diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-06-07 16:51:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-07 16:55:08 -0700 |
| commit | 187404eac001d67e8f0b3a7aeac2fc2b2ba65e0a (patch) | |
| tree | d0fab60667ac3bb57bb0e5bba70225c07908fe49 | |
| parent | 2741561c8eea7748ba04f6b47076bdfd22e3a915 (diff) | |
Setup the env to since ops such as MatchFileOp rely on it.
PiperOrigin-RevId: 158336344
| -rw-r--r-- | tensorflow/core/grappler/optimizers/constant_folding.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/constant_folding.cc b/tensorflow/core/grappler/optimizers/constant_folding.cc index 8bf6a081e3..33b0e16093 100644 --- a/tensorflow/core/grappler/optimizers/constant_folding.cc +++ b/tensorflow/core/grappler/optimizers/constant_folding.cc @@ -30,6 +30,7 @@ limitations under the License. #include "tensorflow/core/lib/gtl/inlined_vector.h" #include "tensorflow/core/lib/strings/numbers.h" #include "tensorflow/core/lib/strings/strcat.h" +#include "tensorflow/core/platform/env.h" #include "tensorflow/core/public/version.h" namespace tensorflow { @@ -53,7 +54,7 @@ class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class DeviceSimple : public DeviceBase { public: - DeviceSimple() : DeviceBase(nullptr) { + DeviceSimple() : DeviceBase(Env::Default()) { eigen_worker_threads_.num_threads = 1; eigen_worker_threads_.workers = new thread::ThreadPool( Env::Default(), "constant_folding", eigen_worker_threads_.num_threads); |