タグ

ブックマーク / natsutan.hatenablog.com (1)

  • ONNXの最適化まとめ - ぱたへね

    ONNXの最適化を一通り試してみたのでまとめ。 サポートしている最適化一覧の取得 サポートしている最適化は、get_available_passesで取得できます。 from onnx import optimizer all_passes = optimizer.get_available_passes() 大きく分けると、このように分類できます。 意味のないOpの削除 (eliminate_deadend等) 2つのOpのfusion (fuse_matmul_add_bias_into_gemm等) Convへのfusion (fuse_add_bias_into_conv等) その他 convへのfuseは全く動かず、バージョンアップ待ちです。 最適化の結果 Qiitaにそれぞれまとめました。 ONNXでeliminate_deadend 最適化 ONNXで eliminate_i

    ONNXの最適化まとめ - ぱたへね
  • 1