We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 50104f1 commit 5d45793Copy full SHA for 5d45793
paddle/fluid/framework/reader.h
@@ -62,9 +62,9 @@ class FileReader : public ReaderBase {
62
// making it easier to access different type reader in Variables.
63
class ReaderHolder {
64
public:
65
- void Reset(ReaderBase* reader) { reader_.reset(reader); }
+ void Reset(ReaderBase* reader) { reader_ = reader; }
66
67
- ReaderBase* Get() const { return reader_.get(); }
+ ReaderBase* Get() const { return reader_; }
68
69
void ReadNext(std::vector<LoDTensor>* out) {
70
PADDLE_ENFORCE_NOT_NULL(reader_);
@@ -76,7 +76,7 @@ class ReaderHolder {
76
}
77
78
private:
79
- std::unique_ptr<ReaderBase> reader_;
+ ReaderBase* reader_;
80
};
81
82
} // namespace framework
0 commit comments