aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/util/tensor_slice_writer.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <nobody@tensorflow.org>2016-04-21 10:08:01 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-04-21 11:13:27 -0700
commit3883578ebff0e8f901386d6f2b5870de517de4a1 (patch)
treef40fff6794ad2ddcb0e891b9965a53201feacc4c /tensorflow/core/util/tensor_slice_writer.h
parent52500e395d526ac932a6de46ecec1ad45082f43c (diff)
Change tensorflow code to use proto_text for DebugString and ShortDebugString
calls, for call sites that are needed on android. Also changed protobuf.h to use MessageLite. Change: 120464266
Diffstat (limited to 'tensorflow/core/util/tensor_slice_writer.h')
-rw-r--r--tensorflow/core/util/tensor_slice_writer.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/util/tensor_slice_writer.h b/tensorflow/core/util/tensor_slice_writer.h
index d3ae7754af..4f7be6896f 100644
--- a/tensorflow/core/util/tensor_slice_writer.h
+++ b/tensorflow/core/util/tensor_slice_writer.h
@@ -33,6 +33,7 @@ limitations under the License.
#include "tensorflow/core/platform/macros.h"
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/util/saved_tensor_slice.pb.h"
+#include "tensorflow/core/util/saved_tensor_slice.pb_text.h"
#include "tensorflow/core/util/saved_tensor_slice_util.h"
namespace tensorflow {
@@ -97,7 +98,7 @@ Status TensorSliceWriter::Add(const string& name, const TensorShape& shape,
// The same tensor has been registered -- we verify that the shapes and the
// type agree.
const SavedSliceMeta& ssm = sts_.meta().tensor(index);
- CHECK_EQ(name, ssm.name()) << ssm.ShortDebugString();
+ CHECK_EQ(name, ssm.name()) << ProtoShortDebugString(ssm);
TensorShape ssm_shape(ssm.shape());
if (!shape.IsSameSize(ssm_shape)) {
return errors::Internal("Mismatching shapes: existing tensor = ",