/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes/prepack_folding.h (358B)
#pragma once
#include
#include
namespace torch {
namespace jit {
using PrePackingOpsFilterFn = std::function;
void PrePackingOpsFolder(
script::Module& m,
const PrePackingOpsFilterFn& is_foldable_op,
const std::string& attr_prefix);
} // namespace jit
} // namespace torch