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

Fix legalizeTF(). Currently definition of this function is inside an anonymous namespace.

Browse Source 
PiperOrigin-RevId: 310040842
Change-Id: Ia37716447a9cb307c26f8c2f6efaee4915e35c19
This commit is contained in:
A. Unique TensorFlower 2020-05-05 16:12:11 -07:00 committed by TensorFlower Gardener
parent ae422505db
commit d71b3dfeca
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
View File

@ -4785,6 +4785,17 @@ class ConvertQrOp : public OpRewritePattern<TF::QrOp> {
} }
}; };
// Performs the lowering to XLA dialect.
void LegalizeTF::runOnFunction() {
if (failed(legalizeTF(getFunction(), allow_partial_conversion_)))
signalPassFailure();
}
static PassRegistration<LegalizeTF> pass(
"xla-legalize-tf", "Legalize from TensorFlow to the XLA dialect");
} // end namespace
#include "tensorflow/compiler/mlir/xla/transforms/generated_legalize_tf.inc" #include "tensorflow/compiler/mlir/xla/transforms/generated_legalize_tf.inc"
LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) { LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) {
@ -4836,17 +4847,6 @@ LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) {
return applyPartialConversion(op, target, patterns); return applyPartialConversion(op, target, patterns);
} }
/// Performs the lowering to XLA dialect.
void LegalizeTF::runOnFunction() {
if (failed(legalizeTF(getFunction(), allow_partial_conversion_)))
signalPassFailure();
}
static PassRegistration<LegalizeTF> pass(
"xla-legalize-tf", "Legalize from TensorFlow to the XLA dialect");
} // end namespace
std::unique_ptr<OperationPass<FuncOp>> createLegalizeTFPass( std::unique_ptr<OperationPass<FuncOp>> createLegalizeTFPass(
bool allow_partial_conversion) { bool allow_partial_conversion) {
return std::make_unique<LegalizeTF>(allow_partial_conversion); return std::make_unique<LegalizeTF>(allow_partial_conversion);