1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7struct Graph;
8
9// Propagates Device type info throughout the given graph.
10TORCH_API bool DeviceTypePropagation(std::shared_ptr<Graph>& graph);
11
12} // namespace jit
13} // namespace torch
14