diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-10 11:23:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-10 11:26:37 -0700 |
commit | bb17f04f59ca0ba1484ffef78936428d6852a8e7 (patch) | |
tree | 5de5f47b8710d0db69d4dcd3d48e2f7bfbb24bf1 /tensorflow/core/util/memmapped_file_system_test.cc | |
parent | a7e8cbc251f9adc530609fa5c06ca933ef526098 (diff) |
Update use of deprecated GraphDef.version to GraphDef.versions. This change only affects a unit test and has no impact on functionality.
PiperOrigin-RevId: 158619477
Diffstat (limited to 'tensorflow/core/util/memmapped_file_system_test.cc')
-rw-r--r-- | tensorflow/core/util/memmapped_file_system_test.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/util/memmapped_file_system_test.cc b/tensorflow/core/util/memmapped_file_system_test.cc index db5ce2e50b..1d01c6b083 100644 --- a/tensorflow/core/util/memmapped_file_system_test.cc +++ b/tensorflow/core/util/memmapped_file_system_test.cc @@ -45,7 +45,8 @@ Status CreateMemmappedFileSystemFile(const string& filename, bool corrupted, // Create a proto with some fields. GraphDef graph_def; - graph_def.set_version(kTestGraphDefVersion); + graph_def.mutable_versions()->set_producer(kTestGraphDefVersion); + graph_def.mutable_versions()->set_min_consumer(kTestGraphDefVersion); TF_RETURN_IF_ERROR(writer.SaveProtobuf(graph_def, kProtoFileName)); // Save a tensor after the proto to check that alignment works. @@ -74,7 +75,8 @@ TEST(MemmappedFileSystemTest, SimpleTest) { GraphDef test_graph_def; TF_EXPECT_OK( ReadBinaryProto(&memmapped_env, kProtoFileName, &test_graph_def)); - EXPECT_EQ(kTestGraphDefVersion, test_graph_def.version()); + EXPECT_EQ(kTestGraphDefVersion, test_graph_def.versions().producer()); + EXPECT_EQ(kTestGraphDefVersion, test_graph_def.versions().min_consumer()); // Check that we can correctly get a tensor memory. std::unique_ptr<ReadOnlyMemoryRegion> memory_region; TF_ASSERT_OK(memmapped_env.NewReadOnlyMemoryRegionFromFile(kTensor2FileName, |