mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
1bc3571078
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70201 Included functions: save_mobile_module -> saves a mobile::Module to flatbuffer load_mobile_module_from_file -> loads a flatbuffer into mobile::Module parse_mobile_module -> parses from bytes or deserialized flatbuffer module object Compared to previous attempts, this diff only adds flatbuffer to cmake target and leaves fbcode/xplat ones unchanged. Test Plan: unittest Reviewed By: malfet, gmagogsfm Differential Revision: D33239362 fbshipit-source-id: b9ca36b83d6af2d78cc50b9eb9e2a6fa7fce0763
27 lines
710 B
C++
27 lines
710 B
C++
#pragma once
|
|
|
|
#include <ATen/core/qualified_name.h>
|
|
#include <flatbuffers/flatbuffers.h>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <ATen/core/ivalue.h>
|
|
#include <ATen/core/jit_type.h>
|
|
#include <torch/csrc/jit/backends/backend_debug_handler.h>
|
|
#include <torch/csrc/jit/mobile/module.h>
|
|
#include <torch/csrc/jit/serialization/type_name_uniquer.h>
|
|
|
|
#include <torch/csrc/jit/serialization/mobile_bytecode_generated.h> // NOLINT
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
TORCH_API void save_mobile_module(
|
|
const mobile::Module& module,
|
|
const std::string& filename);
|
|
TORCH_API flatbuffers::DetachedBuffer save_mobile_module_to_bytes(
|
|
const mobile::Module& module);
|
|
|
|
} // namespace jit
|
|
} // namespace torch
|