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