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
26
27
28
29
30
31
32
|
#pragma once
#include <memory>
#include <torch/csrc/jit/ir/ir.h>
#include <optional>
namespace torch::jit {
const int ONNX_OPSET_9 = 9;
const int ONNX_OPSET_10 = 10;
const int ONNX_OPSET_11 = 11;
const int ONNX_OPSET_12 = 12;
const int ONNX_OPSET_13 = 13;
const int ONNX_OPSET_14 = 14;
namespace onnx_constant_fold {
at::Tensor IntToTensor(int64_t value);
std::optional<at::Tensor> runTorchBackendForOnnx(
const Node* node,
std::vector<at::Tensor>& inputTensorValues,
int opset_version);
} // namespace onnx_constant_fold
void ConstantFoldONNX(
std::shared_ptr<Graph>& g,
std::map<std::string, IValue>& paramDict,
int opset_version);
} // namespace torch::jit
|
