mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
84ec5357d3
Summary: HashNode and CompareNode are useful functions for hanlding jit::Node. This is to unblock https://github.com/pytorch/glow/pull/4235. Pull Request resolved: https://github.com/pytorch/pytorch/pull/34045 Reviewed By: houseroad Differential Revision: D20184733 Pulled By: yinghai fbshipit-source-id: 6c829f2f111a490fd2d85017475c1731cd97fb20
18 lines
290 B
C++
18 lines
290 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
|
|
namespace torch {
|
|
namespace 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 jit
|
|
} // namespace torch
|