1#pragma once
2
3#include <torch/csrc/Export.h>
4
5#include <memory>
6
7namespace torch {
8namespace jit {
9
10struct Graph;
11struct ArgumentSpec;
12
13TORCH_API void PropagateRequiresGrad(std::shared_ptr<Graph>& graph);
14
15} // namespace jit
16} // namespace torch
17