Home
last modified time | relevance | path

Searched refs:RecordWriterOptions (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/lib/io/
Drecord_writer.cc26 bool IsZlibCompressed(const RecordWriterOptions& options) { in IsZlibCompressed()
27 return options.compression_type == RecordWriterOptions::ZLIB_COMPRESSION; in IsZlibCompressed()
30 bool IsSnappyCompressed(const RecordWriterOptions& options) { in IsSnappyCompressed()
31 return options.compression_type == RecordWriterOptions::SNAPPY_COMPRESSION; in IsSnappyCompressed()
35 RecordWriterOptions RecordWriterOptions::CreateRecordWriterOptions( in CreateRecordWriterOptions()
37 RecordWriterOptions options; in CreateRecordWriterOptions()
45 options.compression_type = io::RecordWriterOptions::ZLIB_COMPRESSION; in CreateRecordWriterOptions()
48 options.compression_type = io::RecordWriterOptions::ZLIB_COMPRESSION; in CreateRecordWriterOptions()
51 options.compression_type = io::RecordWriterOptions::SNAPPY_COMPRESSION; in CreateRecordWriterOptions()
61 const RecordWriterOptions& options) in RecordWriter()
[all …]
Drecord_reader_writer_test.cc40 const io::RecordWriterOptions& options) { in GetMatchingReaderOptions()
41 if (options.compression_type == io::RecordWriterOptions::ZLIB_COMPRESSION) { in GetMatchingReaderOptions()
54 void VerifyFlush(const io::RecordWriterOptions& options) { in VerifyFlush()
104 io::RecordWriterOptions options; in TEST()
109 io::RecordWriterOptions options; in TEST()
110 options.compression_type = io::RecordWriterOptions::ZLIB_COMPRESSION; in TEST()
129 io::RecordWriterOptions options; in TEST()
170 io::RecordWriterOptions options; in TEST()
207 io::RecordWriterOptions options; in TEST()
243 io::RecordWriterOptions options; in TEST()
[all …]
Drecord_writer.h39 struct RecordWriterOptions { struct
48 static RecordWriterOptions CreateRecordWriterOptions( argument
71 explicit RecordWriter(WritableFile* dest, const RecordWriterOptions& options = argument
72 RecordWriterOptions());
113 RecordWriterOptions options_;
Drecordio_test.cc251 void TestNonSequentialReads(const RecordWriterOptions& writer_options, in TestNonSequentialReads()
286 TestNonSequentialReads(RecordWriterOptions(), RecordReaderOptions()); in TEST_F()
292 TestNonSequentialReads(RecordWriterOptions(), options); in TEST_F()
297 RecordWriterOptions::CreateRecordWriterOptions("ZLIB"), in TEST_F()
307 void TestReadError(const RecordWriterOptions& writer_options, in TestReadError()
333 TestReadError(RecordWriterOptions(), RecordReaderOptions()); in TEST_F()
339 TestReadError(RecordWriterOptions(), options); in TEST_F()
343 TestReadError(RecordWriterOptions::CreateRecordWriterOptions("ZLIB"), in TEST_F()
/external/tensorflow/tensorflow/python/lib/io/
Drecord_io_wrapper.cc140 const tensorflow::io::RecordWriterOptions& options, in New()
284 using tensorflow::io::RecordWriterOptions; in PYBIND11_MODULE()
285 py::class_<RecordWriterOptions>(m, "RecordWriterOptions") in PYBIND11_MODULE()
286 .def(py::init(&RecordWriterOptions::CreateRecordWriterOptions)) in PYBIND11_MODULE()
287 .def_readonly("compression_type", &RecordWriterOptions::compression_type) in PYBIND11_MODULE()
288 .def_readonly("zlib_options", &RecordWriterOptions::zlib_options); in PYBIND11_MODULE()
294 [](const std::string& filename, const RecordWriterOptions& options) { in PYBIND11_MODULE()
Dtf_record.py129 options = _pywrap_record_io.RecordWriterOptions(
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc72 io::RecordWriterOptions::CreateRecordWriterOptions(compression_type)); in DoCompute()
Dsnapshot_util.cc111 dest_.get(), io::RecordWriterOptions::CreateRecordWriterOptions( in Initialize()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc160 auto options = io::RecordWriterOptions::CreateRecordWriterOptions( in WriteDataToTFRecordFile()