1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
| #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
|
|