• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #pragma once
2 
3 #include <memory>
4 
5 #include <torch/csrc/jit/ir/ir.h>
6 
7 namespace torch::jit {
8 
9 void DeduplicateInitializers(
10     std::shared_ptr<Graph>& g,
11     std::map<std::string, IValue>& paramsDict,
12     bool is_train);
13 
14 } // namespace torch::jit
15