/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes/lower_graph.h (750B)
#pragma once
#include
namespace torch {
namespace jit {
using ModulePtr = c10::intrusive_ptr;
// Given a graph with of a method which first argument is %self, lower it to a
// graph where all attributes accesses are replaced with explicit inputs of the
// graph (rather than results of prim::GetAttr executed on %self).
//
// Returns a tuple (graph, parameters) where the last module.parameters.size()
// inputs to the graph are the trainable parameters used in this method. The
// remaining inputs are the true inputs to the function.
TORCH_API std::pair, std::vector> LowerGraph(
Graph& graph,
const ModulePtr& self);
} // namespace jit
} // namespace torch