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 <ATen/core/dispatch/Dispatcher.h>
#include <ATen/core/ivalue.h>
#include <ATen/core/stack.h>
#include <functional>
namespace torch::lazy {
bool force_eager_fallback(c10::Symbol op);
void ltc_eager_fallback(
const c10::OperatorHandle& op,
torch::jit::Stack* stack);
void ts_eager_fallback(
const c10::OperatorHandle& op,
torch::jit::Stack* stack,
c10::DeviceType device_type);
// The TorchScript backend does not register itself with pytorch dispatcher
// until it is explicitly initialized. This function should only be called
// by the main Torchscript backend init function.
void register_ts_ltc_eager_fallback();
} // namespace torch::lazy
|
