1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| #pragma once
|
| #include <cstddef>
| #include <cstdint>
|
| #include "c10/macros/Macros.h"
|
| namespace caffe2 {
| namespace serialize {
|
| // this is the interface for the (file/stream/memory) reader in
| // PyTorchStreamReader. with this interface, we can extend the support
| // besides standard istream
| class CAFFE2_API ReadAdapterInterface {
| public:
| virtual size_t size() const = 0;
| virtual size_t read(uint64_t pos, void* buf, size_t n, const char* what = "")
| const = 0;
| virtual ~ReadAdapterInterface();
| };
|
| } // namespace serialize
| } // namespace caffe2
|
|
