1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#pragma once
#include <torch/csrc/jit/mobile/function.h>
namespace torch::jit {
using c10::IValue;
enum MobileModuleLoadOptions {
OPERATOR_CHECK = 1,
// PARSE_ALL_EXTRA_FILE_MAPS is used to gate for ExtraFileMaps to pull all
// files automatically without explicit entries mapping. Refer to PR for a
// detail: https://github.com/pytorch/pytorch/pull/99747
PARSE_ALL_EXTRA_FILE_MAPS = 2,
};
const uint64_t kDefaultMobileLoadOptions =
MobileModuleLoadOptions::OPERATOR_CHECK;
namespace mobile {
TORCH_API void parseOperators(
c10::ivalue::TupleElements&& ops_list,
const uint64_t& module_load_options,
mobile::Function* function);
} // namespace mobile
} // namespace torch::jit
|
