| #pragma once | |
| #include <torch/csrc/jit/ir.h> | |
| namespace torch { | |
| namespace jit { | |
| struct HashNode { | |
| size_t operator()(const Node* k) const; | |
| }; | |
| struct EqualNode { | |
| bool operator()(const Node* lhs, const Node* rhs) const; | |
| }; | |
| } // namespace jit | |
| } // namespace torch |