mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
5b3b2b9cc7
Follows #131986 Pull Request resolved: https://github.com/pytorch/pytorch/pull/131996 Approved by: https://github.com/ezyang
16 lines
265 B
C++
16 lines
265 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
|
|
namespace torch::jit {
|
|
|
|
struct TORCH_API HashNode {
|
|
size_t operator()(const Node* k) const;
|
|
};
|
|
|
|
struct TORCH_API EqualNode {
|
|
bool operator()(const Node* lhs, const Node* rhs) const;
|
|
};
|
|
|
|
} // namespace torch::jit
|