diff options
Diffstat (limited to 'tensorflow/core/util/tensor_slice_reader_test.cc')
| -rw-r--r-- | tensorflow/core/util/tensor_slice_reader_test.cc | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/tensorflow/core/util/tensor_slice_reader_test.cc b/tensorflow/core/util/tensor_slice_reader_test.cc index 8545697886..f4859262e1 100644 --- a/tensorflow/core/util/tensor_slice_reader_test.cc +++ b/tensorflow/core/util/tensor_slice_reader_test.cc @@ -13,6 +13,8 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ +#include <utility> + #include "tensorflow/core/util/tensor_slice_reader.h" #include "tensorflow/core/framework/types.h" @@ -48,8 +50,9 @@ namespace { // // We assume this is a row-major matrix. -void SimpleFloatHelper(TensorSliceWriter::CreateBuilderFunction create_function, - TensorSliceReader::OpenTableFunction open_function) { +void SimpleFloatHelper( + const TensorSliceWriter::CreateBuilderFunction& create_function, + TensorSliceReader::OpenTableFunction open_function) { const string fname_base = io::JoinPath(testing::TmpDir(), "float_checkpoint"); TensorShape shape({4, 5}); @@ -108,7 +111,7 @@ void SimpleFloatHelper(TensorSliceWriter::CreateBuilderFunction create_function, // Now we need to read the tensor slices const string filepattern = strings::StrCat(fname_base, "_*"); - TensorSliceReader reader(filepattern, open_function); + TensorSliceReader reader(filepattern, std::move(open_function)); TF_EXPECT_OK(reader.status()); EXPECT_EQ(2, reader.num_files()); @@ -171,9 +174,10 @@ TEST(TensorSliceReaderTest, SimpleFloat) { } template <typename T, typename U> -void SimpleIntXHelper(TensorSliceWriter::CreateBuilderFunction create_function, - TensorSliceReader::OpenTableFunction open_function, - const string& checkpoint_file) { +void SimpleIntXHelper( + const TensorSliceWriter::CreateBuilderFunction& create_function, + TensorSliceReader::OpenTableFunction open_function, + const string& checkpoint_file) { const string fname_base = io::JoinPath(testing::TmpDir(), checkpoint_file); TensorShape shape({4, 5}); @@ -232,7 +236,7 @@ void SimpleIntXHelper(TensorSliceWriter::CreateBuilderFunction create_function, // Now we need to read the tensor slices const string filepattern = strings::StrCat(fname_base, "_*"); - TensorSliceReader reader(filepattern, open_function); + TensorSliceReader reader(filepattern, std::move(open_function)); TF_EXPECT_OK(reader.status()); EXPECT_EQ(2, reader.num_files()); @@ -304,8 +308,8 @@ TEST_SIMPLE_INT(int8, int32) TEST_SIMPLE_INT(uint8, int32) void CachedTensorSliceReaderTesterHelper( - TensorSliceWriter::CreateBuilderFunction create_function, - TensorSliceReader::OpenTableFunction open_function) { + const TensorSliceWriter::CreateBuilderFunction& create_function, + const TensorSliceReader::OpenTableFunction& open_function) { const string fname_base = io::JoinPath(testing::TmpDir(), "float_checkpoint"); TensorShape shape({4, 5}); |