mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8967d55b01
Follows #132753 Pull Request resolved: https://github.com/pytorch/pytorch/pull/132963 Approved by: https://github.com/Skylion007
22 lines
740 B
C++
22 lines
740 B
C++
#pragma once
|
|
#include <torch/csrc/jit/mobile/function.h>
|
|
|
|
namespace torch::jit::mobile {
|
|
using c10::IValue;
|
|
TORCH_API void parseInstructions(
|
|
const std::string& function_name,
|
|
c10::ivalue::TupleElements&& ins_list,
|
|
c10::ivalue::TupleElements& debug_handles_m_tuple,
|
|
mobile::Function* function);
|
|
TORCH_API void parseConstants(
|
|
const c10::ivalue::TupleElements& consts_list,
|
|
mobile::Function* function);
|
|
TORCH_API void parseTypes(
|
|
const c10::ivalue::TupleElements& types_list,
|
|
mobile::Function* function);
|
|
TORCH_API void parseRegisterSize(size_t rsize, mobile::Function* function);
|
|
TORCH_API void applyUpgrader(
|
|
mobile::Function* function,
|
|
uint64_t operator_version);
|
|
} // namespace torch::jit::mobile
|