xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/eliminate_unused_items.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/jit/passes/onnx/eliminate_unused_items.h>
2 #include <torch/csrc/jit/passes/onnx/helper.h>
3 
4 namespace torch::jit {
5 
6 namespace onnx {
7 using namespace ::c10::onnx;
8 }
9 
EliminateUnusedItemsONNX(Block * b,ParamMap & paramsDict)10 void EliminateUnusedItemsONNX(Block* b, ParamMap& paramsDict) {
11   auto valsToParamsMap = buildValueToParamsMap(b, paramsDict);
12   eraseUnusedValuesFromMap(valsToParamsMap);
13   eraseUnusedBlockInputs(b);
14   buildParamsMapFromValueToParamsMap(valsToParamsMap, paramsDict);
15   return;
16 }
17 
18 } // namespace torch::jit
19