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

Automated rollback of commit 13f0092fa3

Browse Source 
PiperOrigin-RevId: 294292388
Change-Id: Ie74a6c898b9d8866bb89644f5b19bc9d7867a8a1
This commit is contained in:
A. Unique TensorFlower 2020-02-10 13:35:55 -08:00 committed by TensorFlower Gardener
parent 7343cfb70a
commit 036b77ad19
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc
View File

@ -141,8 +141,11 @@ mlir::OwningModuleRef SavedModelV1ToMlirImport(
absl::string_view saved_model_dir, absl::string_view saved_model_dir,
const std::unordered_set<std::string>& tags, mlir::MLIRContext* context) { const std::unordered_set<std::string>& tags, mlir::MLIRContext* context) {
tensorflow::SavedModelBundle bundle; tensorflow::SavedModelBundle bundle;
auto load_status = tensorflow::LoadSavedModel( tensorflow::SessionOptions session_options;
/* session_options = */ {}, /* run_options = */ {}, // Force saved model states to be restored to CPU.
(*session_options.config.mutable_device_count())["GPU"] = 0;
auto load_status =
tensorflow::LoadSavedModel(session_options, /* run_options = */ {},
std::string(saved_model_dir), tags, &bundle); std::string(saved_model_dir), tags, &bundle);
if (!load_status.ok()) { if (!load_status.ok()) {
LOG(ERROR) << "Failed to load saved model v1 '" << saved_model_dir LOG(ERROR) << "Failed to load saved model v1 '" << saved_model_dir