1 | #pragma once |
2 | |
3 | #include <ATen/core/dispatch/Dispatcher.h> |
4 | #include <ATen/core/ivalue.h> |
5 | #include <ATen/core/stack.h> |
6 | #include <functional> |
7 | |
8 | namespace torch { |
9 | namespace lazy { |
10 | |
11 | bool force_eager_fallback(c10::Symbol op); |
12 | void ltc_eager_fallback( |
13 | const c10::OperatorHandle& op, |
14 | torch::jit::Stack* stack); |
15 | |
16 | void ts_eager_fallback( |
17 | const c10::OperatorHandle& op, |
18 | torch::jit::Stack* stack, |
19 | c10::DeviceType device_type); |
20 | |
21 | // The TorchScript backend does not register itself with pytorch dispatcher |
22 | // until it is explicitly initialized. This function should only be called |
23 | // by the main Torchscript backend init function. |
24 | void register_ts_ltc_eager_fallback(); |
25 | |
26 | } // namespace lazy |
27 | } // namespace torch |
28 | |