1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 namespace torch::jit { 6 7 // EliminateUnusedItemsONNX pass is removing unused 8 // initializers and inputs, this is needed because 9 // dce pass is only removing unused fork inputs 10 void EliminateUnusedItemsONNX( 11 Block* b, 12 std::map<std::string, IValue>& paramDict); 13 14 } // namespace torch::jit 15