1 #pragma once 2 3 #include <torch/csrc/jit/api/module.h> 4 #include <torch/csrc/jit/ir/ir.h> 5 #include <torch/csrc/onnx/onnx.h> 6 7 #include <memory> 8 9 namespace torch::jit { 10 11 TORCH_API void UnpackQuantizedWeights( 12 std::shared_ptr<Graph>& graph, 13 std::map<std::string, IValue>& paramsDict); 14 TORCH_API void insertPermutes( 15 std::shared_ptr<Graph>& graph, 16 std::map<std::string, IValue>& paramsDict); 17 } // namespace torch::jit 18