blob: 112e56561a0b91460370bb8d3124b60b86954086 [file] [log] [blame]
#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