Remove explicit
keyword from snapshot_util Reader/Writer constructors.
PiperOrigin-RevId: 314717764 Change-Id: I5001ce573d04321c696bf41df08ee4c72093e68f
This commit is contained in:
parent
0ca0c442c0
commit
b794497e61
@ -82,8 +82,8 @@ class Writer {
|
|||||||
// Writes snapshots with the standard TFRecord file format.
|
// Writes snapshots with the standard TFRecord file format.
|
||||||
class TFRecordWriter : public Writer {
|
class TFRecordWriter : public Writer {
|
||||||
public:
|
public:
|
||||||
explicit TFRecordWriter(const std::string& filename,
|
TFRecordWriter(const std::string& filename,
|
||||||
const std::string& compression_type);
|
const std::string& compression_type);
|
||||||
|
|
||||||
Status WriteTensors(const std::vector<Tensor>& tensors) override;
|
Status WriteTensors(const std::vector<Tensor>& tensors) override;
|
||||||
|
|
||||||
@ -114,9 +114,8 @@ class CustomWriter : public Writer {
|
|||||||
static constexpr const char* const kWriteCord = "WriteCord";
|
static constexpr const char* const kWriteCord = "WriteCord";
|
||||||
static constexpr const char* const kSeparator = "::";
|
static constexpr const char* const kSeparator = "::";
|
||||||
|
|
||||||
explicit CustomWriter(const std::string& filename,
|
CustomWriter(const std::string& filename, const std::string& compression_type,
|
||||||
const std::string& compression_type,
|
const DataTypeVector& dtypes);
|
||||||
const DataTypeVector& dtypes);
|
|
||||||
|
|
||||||
Status WriteTensors(const std::vector<Tensor>& tensors) override;
|
Status WriteTensors(const std::vector<Tensor>& tensors) override;
|
||||||
|
|
||||||
@ -192,9 +191,8 @@ class Reader {
|
|||||||
// Reads snapshots previously written with `TFRecordWriter`.
|
// Reads snapshots previously written with `TFRecordWriter`.
|
||||||
class TFRecordReader : public Reader {
|
class TFRecordReader : public Reader {
|
||||||
public:
|
public:
|
||||||
explicit TFRecordReader(const std::string& filename,
|
TFRecordReader(const std::string& filename, const string& compression_type,
|
||||||
const string& compression_type,
|
const DataTypeVector& dtypes);
|
||||||
const DataTypeVector& dtypes);
|
|
||||||
|
|
||||||
Status ReadTensors(std::vector<Tensor>* read_tensors) override;
|
Status ReadTensors(std::vector<Tensor>* read_tensors) override;
|
||||||
|
|
||||||
@ -231,9 +229,8 @@ class CustomReader : public Reader {
|
|||||||
static constexpr const char* const kReadCord = "ReadCord";
|
static constexpr const char* const kReadCord = "ReadCord";
|
||||||
static constexpr const char* const kSeparator = "::";
|
static constexpr const char* const kSeparator = "::";
|
||||||
|
|
||||||
explicit CustomReader(const std::string& filename,
|
CustomReader(const std::string& filename, const string& compression_type,
|
||||||
const string& compression_type, const int version,
|
const int version, const DataTypeVector& dtypes);
|
||||||
const DataTypeVector& dtypes);
|
|
||||||
|
|
||||||
Status ReadTensors(std::vector<Tensor>* read_tensors) override;
|
Status ReadTensors(std::vector<Tensor>* read_tensors) override;
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user