commit | fe3f150eef1e3a40cbd7337f92a35111e7cb0ae4 | [log] [tgz] |
---|---|---|
author | Brian Atkinson <bca@google.com> | Fri Feb 07 11:12:05 2020 -0800 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Feb 07 11:18:11 2020 -0800 |
tree | d3dbd398733d36378cb168964f440ee7c347c216 | |
parent | 930c651dd6ed2b3b2f65e93404bbcb6312fe86e1 [diff] |
Include passes.h to make declarations available while compiling the definitions. PiperOrigin-RevId: 293853109 Change-Id: I94a3e1095c6b790409e6e396547444bc02e8d2f1
diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc b/tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc index 8d83b5c..8b5f93b 100644 --- a/tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc +++ b/tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
@@ -16,6 +16,7 @@ #include "mlir/IR/PatternMatch.h" // TF:llvm-project #include "mlir/Pass/Pass.h" // TF:llvm-project #include "tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.h" +#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" namespace mlir { namespace TFDevice {
diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc b/tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc index 3bc2f74..cf7ae8e 100644 --- a/tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc +++ b/tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
@@ -41,6 +41,7 @@ #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.h" +#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_type.h" #include "tensorflow/compiler/mlir/tensorflow/utils/mangling_util.h"