mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 21:49:24 +08:00
8967d55b01
Follows #132753 Pull Request resolved: https://github.com/pytorch/pytorch/pull/132963 Approved by: https://github.com/Skylion007
36 lines
1.1 KiB
C++
36 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include <ATen/core/ivalue.h>
|
|
#include <ATen/core/operator_name.h>
|
|
#include <torch/csrc/jit/runtime/instruction.h>
|
|
|
|
namespace torch::jit::mobile {
|
|
|
|
using Stack = std::vector<c10::IValue>;
|
|
using DebugHandle = int64_t;
|
|
|
|
class Function;
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
struct Code {
|
|
std::vector<Instruction> instructions_;
|
|
std::vector<DebugHandle> debug_handles_;
|
|
std::vector<c10::OperatorName> op_names_;
|
|
std::vector<int> operator_input_sizes_;
|
|
std::vector<std::function<void(Stack&)>> operators_;
|
|
std::vector<c10::IValue> constants_;
|
|
std::vector<c10::TypePtr> types_;
|
|
// TODO After we actually export CALL instructions we can remove this.
|
|
// We may need a two-stage importing scheme, where we firstly construct all
|
|
// function objects, and then append referenced function pointers. This could
|
|
// be done in parseMethods().
|
|
std::vector<mobile::Function*> functions_;
|
|
size_t register_size_ = 0; // Aggregated output size.
|
|
// initialized means operators_ array is filled with operators
|
|
bool initialized = false;
|
|
};
|
|
|
|
} // namespace torch::jit::mobile
|