mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
2f30473fba
Follows #132963 Pull Request resolved: https://github.com/pytorch/pytorch/pull/133067 Approved by: https://github.com/Skylion007
40 lines
905 B
C++
40 lines
905 B
C++
#pragma once
|
|
|
|
#include <ATen/core/ivalue_inl.h>
|
|
|
|
#include <torch/csrc/jit/mobile/code.h>
|
|
#include <torch/csrc/jit/mobile/function.h>
|
|
#include <torch/csrc/jit/serialization/import_export_functions.h>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
namespace torch::jit {
|
|
struct Instruction;
|
|
struct Upgrader {
|
|
int min_version;
|
|
int max_version;
|
|
std::string upgrader_name;
|
|
int index;
|
|
};
|
|
|
|
// From operator_versions.yaml
|
|
TORCH_API const std::unordered_map<std::string, std::vector<Upgrader>>
|
|
getOperatorVersionMapForMobile();
|
|
|
|
struct OperatorString {
|
|
const std::string name;
|
|
const std::string overload_name;
|
|
const std::optional<int> num_specified_args;
|
|
};
|
|
|
|
struct ByteCodeFunctionWithOperator {
|
|
mobile::Function& function;
|
|
std::vector<OperatorString> operators;
|
|
};
|
|
|
|
TORCH_API const std::vector<ByteCodeFunctionWithOperator>&
|
|
getUpgraderBytecodeList();
|
|
|
|
} // namespace torch::jit
|