1#pragma once
2
3#include <torch/csrc/Export.h>
4#include <torch/csrc/jit/ir/ir.h>
5#include <memory>
6
7namespace torch {
8namespace jit {
9struct Graph;
10
11// Propagate tensor properties (e.g., dtype, device, is_contiguous, layout)
12// propagation on all tensor objects. Currently, we only support dtype
13// propagation
14TORCH_API bool DtypePropagation(std::shared_ptr<Graph>& graph);
15
16} // namespace jit
17} // namespace torch
18