pytorch/torch/csrc/jit/passes/inliner.h

13 lines
229 B
C++

#pragma once
#include <torch/csrc/jit/ir/ir.h>
namespace torch::jit {
// Inline function and method calls.
TORCH_API void Inline(Graph& graph);
TORCH_API GraphFunction* tryToGraphFunction(Node* n);
} // namespace torch::jit