1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8TORCH_API bool canRunWithAutograd(Node* node);
9
10TORCH_API void InlineAutodiffSubgraphs(
11 std::shared_ptr<Graph>& graph,
12 size_t threshold = 5);
13
14} // namespace jit
15} // namespace torch
16