/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/ir
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/ir/irparser.h (767B)
#pragma once
#include
#include
#include
#include
namespace torch {
namespace jit {
struct Graph;
struct Value;
// \brief Parse IR from \p STR constructing the corresponding IR in\ GRAPH.
TORCH_API void parseIR(const std::string& str, torch::jit::Graph* graph);
/** \brief Parse IR from \p STR constructing the corresponding IR in\ GRAPH.
*
* \p VMAP is filled with String to Value pairs allowing to index Values in the
* newly created graph by their name in the original IR string.
*/
TORCH_API void parseIR(
const std::string& str,
torch::jit::Graph* graph,
std::unordered_map& vmap);
} // namespace jit
} // namespace torch