1 | #pragma once |
---|---|
2 | |
3 | #include <torch/csrc/api/include/torch/jit.h> |
4 | #include <torch/csrc/lazy/backend/lowering_context.h> |
5 | |
6 | namespace torch { |
7 | namespace lazy { |
8 | using TSOpVector = std::vector<torch::jit::Value*>; |
9 | |
10 | TORCH_API TSOpVector LowerTSBuiltin( |
11 | std::shared_ptr<torch::jit::GraphFunction> function, |
12 | c10::Symbol sym, |
13 | const std::vector<torch::jit::NamedValue>& arguments, |
14 | const std::vector<torch::jit::NamedValue>& kwarguments = {}); |
15 | |
16 | } // namespace lazy |
17 | } // namespace torch |
18 |