Skip to content
  • Justin Ngo's avatar
    Add Torch Unit Test Suite to Reference Model · 49c51ad8
    Justin Ngo authored and Eric Kunze's avatar Eric Kunze committed
    
    
    Unit test test suite for PyTorch utilizing FxImporter to lower PyTorch
    graphs to Torch-MLIR then to TOSA IR. Current infrastructure is mostly
    based on the existing TF/TFL frameworks test suite.
    
    This is the initial work for the Torch unit test suite. The end goal
    is to add more Torch unit tests for a more robust test suite.
    
    Signed-off-by: default avatarJustin Ngo <justin.ngo@arm.com>
    Change-Id: I829ddf708556488166bc2bfdd4b52a1cfb4fc34d
    49c51ad8
Loading