experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

NFC: Add headers for MLIR builders and standard types

Browse Source 
PiperOrigin-RevId: 282799238
Change-Id: Ifb73b4953dc85ba79670348f34afc6fb36c33bfb
This commit is contained in:
Lei Zhang 2019-11-27 10:55:59 -08:00 committed by TensorFlower Gardener
parent fb28210ab0
commit b1e14472c8
2 changed files with 3 additions and 0 deletions
tensorflow/compiler/mlir/tensorflow/ir
tf_device.cctf_saved_model.cc

2
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
View File

@ -26,11 +26,13 @@ limitations under the License.
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/SMLoc.h"
#include "mlir/IR/Attributes.h" // TF:local_config_mlir
#include "mlir/IR/Builders.h" // TF:local_config_mlir
#include "mlir/IR/MLIRContext.h" // TF:local_config_mlir
#include "mlir/IR/OpDefinition.h" // TF:local_config_mlir
#include "mlir/IR/OpImplementation.h" // TF:local_config_mlir
#include "mlir/IR/OperationSupport.h" // TF:local_config_mlir
#include "mlir/IR/PatternMatch.h" // TF:local_config_mlir
#include "mlir/IR/StandardTypes.h" // TF:local_config_mlir
#include "mlir/IR/TypeUtilities.h" // TF:local_config_mlir
#include "mlir/IR/Types.h" // TF:local_config_mlir
#include "mlir/IR/Value.h" // TF:local_config_mlir

1
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
View File

@ -23,6 +23,7 @@ limitations under the License.
#include "llvm/Support/Casting.h"
#include "llvm/Support/raw_ostream.h"
#include "mlir/IR/Attributes.h" // TF:local_config_mlir
#include "mlir/IR/Builders.h" // TF:local_config_mlir
#include "mlir/IR/Function.h" // TF:local_config_mlir
#include "mlir/IR/Identifier.h" // TF:local_config_mlir
#include "mlir/IR/Module.h" // TF:local_config_mlir