mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b329e03684
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15899 Add PyTorchPredictorContainer to support multiple jit script modules Reviewed By: pritamdamania87 Differential Revision: D13596139 fbshipit-source-id: 3ce0bdf2f4dbba7aa1d20e824d03e5ac98f5d887
29 lines
687 B
C++
29 lines
687 B
C++
#pragma once
|
|
|
|
#include <fstream>
|
|
#include <memory>
|
|
|
|
#include "c10/macros/Macros.h"
|
|
#include "caffe2/serialize/istream_adapter.h"
|
|
#include "caffe2/serialize/read_adapter_interface.h"
|
|
|
|
namespace caffe2 {
|
|
namespace serialize {
|
|
|
|
class CAFFE2_API FileAdapter final : public ReadAdapterInterface {
|
|
public:
|
|
C10_DISABLE_COPY_AND_ASSIGN(FileAdapter);
|
|
explicit FileAdapter(const std::string& file_name);
|
|
size_t size() const override;
|
|
size_t read(uint64_t pos, void* buf, size_t n, const char* what = "")
|
|
const override;
|
|
~FileAdapter();
|
|
|
|
private:
|
|
std::ifstream file_stream_;
|
|
std::unique_ptr<IStreamAdapter> istream_adapter_;
|
|
};
|
|
|
|
} // namespace serialize
|
|
} // namespace caffe2
|