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

Automated rollback of commit 85a4cef8b4

Browse Source 
PiperOrigin-RevId: 293937232
Change-Id: Iae3f40d6b05ff380d2fd345f4a1a3b5957c8062b
This commit is contained in:
A. Unique TensorFlower 2020-02-07 18:33:38 -08:00 committed by TensorFlower Gardener
parent 25dbdec761
commit 13f0092fa3
1 changed files with 3 additions and 6 deletions

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

@ -141,12 +141,9 @@ mlir::OwningModuleRef SavedModelV1ToMlirImport(
absl::string_view saved_model_dir,
const std::unordered_set<std::string>& tags, mlir::MLIRContext* context) {
tensorflow::SavedModelBundle bundle;
tensorflow::SessionOptions session_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);
auto load_status = tensorflow::LoadSavedModel(
/* session_options = */ {}, /* run_options = */ {},
std::string(saved_model_dir), tags, &bundle);
if (!load_status.ok()) {
LOG(ERROR) << "Failed to load saved model v1 '" << saved_model_dir
<< "': " << load_status;