1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#pragma once
#include <torch/csrc/jit/ir/ir.h>
namespace torch {
namespace jit {
TORCH_API void ConstantPropagation(std::shared_ptr<Graph>& graph);
// runs constant propagation only on ops that have non-aliasing inputs & outputs
TORCH_API void ConstantPropagationImmutableTypes(std::shared_ptr<Graph>& graph);
// Runs the node if its inputs are constants. Callers of this function must
// make their own determination if constant prop is appropriate - for example
// non-deterministic ops or ops with side effects
TORCH_API c10::optional<Stack> runNodeIfInputsAreConstant(const Node* node);
} // namespace jit
} // namespace torch
|
