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 <torch/csrc/jit/mobile/function.h>
namespace torch {
namespace jit {
using c10::IValue;
enum MobileModuleLoadOptions {
OPERATOR_CHECK = 1,
};
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 jit
} // namespace torch
|
