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

Internal protobuf refactoring

Browse Source 
PiperOrigin-RevId: 321197814
Change-Id: If6023874e67199f9d66b6380a73838da16837005
This commit is contained in:
Gunhan Gulsoy 2020-07-14 11:24:45 -07:00 committed by TensorFlower Gardener
parent ed6ff2af96
commit 9fd2e39518
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/xla/tools/hlo_module_loader.cc
View File

@ -21,7 +21,6 @@ limitations under the License.
#include <string>
#include <utility>
#include "google/protobuf/text_format.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/str_join.h"
#include "absl/strings/str_split.h"
@ -32,6 +31,7 @@ limitations under the License.
#include "tensorflow/core/lib/io/path.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/protobuf.h"
#include "tensorflow/core/platform/regexp.h"
namespace xla {
@ -87,9 +87,10 @@ StatusOr<std::unique_ptr<HloModule>> LoadModuleFromData(
return InvalidArgument("Failed to parse input as HLO protobuf binary");
}
} else if (format == "pbtxt") {
if (!google::protobuf::TextFormat::ParseFromString(data, &proto) &&
!google::protobuf::TextFormat::ParseFromString(data, proto.mutable_hlo()) &&
!google::protobuf::TextFormat::ParseFromString(
if (!tensorflow::protobuf::TextFormat::ParseFromString(data, &proto) &&
!tensorflow::protobuf::TextFormat::ParseFromString(
data, proto.mutable_hlo()) &&
!tensorflow::protobuf::TextFormat::ParseFromString(
data, proto.mutable_hlo()->mutable_hlo_module())) {
return InvalidArgument("Failed to parse input as HLO protobuf text");
}