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

Remove the use of the global dialect registry in tf-mlir-translate (NFC)

Browse Source 
We've been moving to explicit registration for every users.

PiperOrigin-RevId: 328274615
Change-Id: I3c3f83d107ff8131d441640d23b82aaeff86a0b5
This commit is contained in:
Mehdi Amini 2020-08-24 23:01:07 -07:00 committed by TensorFlower Gardener
parent 2856a636b1
commit 63696674f3
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/mlir/tf_mlir_translate_main.cc
View File

@ -111,8 +111,6 @@ int main(int argc, char** argv) {
if (import_saved_model_object_graph) {
mlir::MLIRContext context;
context.loadAllGloballyRegisteredDialects();
auto module_or = tensorflow::SavedModelObjectGraphToMlirImport(
input_filename, tags, exported_names, &context);
if (!module_or.status().ok()) return 1;
@ -120,8 +118,6 @@ int main(int argc, char** argv) {
module_or.ConsumeValueOrDie()->print(output->os());
} else if (import_saved_model_signature_defs) {
mlir::MLIRContext context;
context.loadAllGloballyRegisteredDialects();
auto module_or = tensorflow::SavedModelSignatureDefsToMlirImport(
input_filename, tags, exported_names, &context, upgrade_legacy);
if (!module_or.status().ok()) return 1;
@ -141,7 +137,6 @@ int main(int argc, char** argv) {
llvm::SourceMgr sourceMgr;
sourceMgr.AddNewSourceBuffer(std::move(ownedBuffer), llvm::SMLoc());
mlir::MLIRContext context;
context.loadAllGloballyRegisteredDialects();
mlir::SourceMgrDiagnosticHandler diagnostic_handler(sourceMgr, &context);
return (*requested_translation)(sourceMgr, os, &context);
};