• 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
parse_gemm.cpp 5.2 KB