xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/import_export_functions.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 #include <ATen/core/ivalue.h>
3 
4 // Functions that are used in both import and export processes
5 
6 namespace torch::jit {
7 using c10::IValue;
8 IValue expect_field(
9     c10::ivalue::TupleElements& elements,
10     const std::string& expected_name,
11     size_t entry);
12 std::string operator_str(
13     const std::string& name,
14     const std::string& overloadname);
15 } // namespace torch::jit
16