1#pragma once
2
3#include <torch/csrc/jit/api/module.h>
4#include <torch/csrc/jit/ir/ir.h>
5#include <memory>
6
7namespace torch {
8namespace jit {
9
10using PrePackParamFilterFn = std::function<bool(Node*)>;
11
12TORCH_API std::unordered_set<std::string> RegisterPrePackParams(
13 Module& m,
14 const std::string& method_name,
15 const PrePackParamFilterFn& is_packed_param,
16 const std::string& attr_prefix);
17
18TORCH_API std::string joinPaths(const std::vector<std::string>& paths);
19} // namespace jit
20} // namespace torch
21