Skip to content

GitLab

"megatron/training/arguments.py" did not exist on "a3fbac58fc042daf219a2bd1878a90a48143a65f"
Switch branch/tag
  1. 17 Jan, 2023 1 commit
    • Charlie Lin's avatar
      Dynamic ONNX Gemm (#1459) · 8b651eee
      Charlie Lin authored 
      Extends ONNX Gemm parser to handle dynamic input shapes
Limits ONNX Gemm parsing to 2D input tensors for A and B inputs
As per the ONNX specifications
Changed Gemm ONNX tests to 2D input versions
Add onnx_verify tests for Gemm
Parsing ONNX Gemm links to more than one operator, checking that it produces the correct result
      8b651eee