1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 // Functions used by both encapsulation and conversion. 6 7 namespace torch { 8 namespace jit { 9 10 struct IndexingPatternFinder { 11 public: 12 static std::vector<Node*> FetchSliceAndSelect(const Node* node); 13 14 private: 15 static bool IsSameSource(const Node* n, const Node* m); 16 }; 17 18 } // namespace jit 19 } // namespace torch 20