mirror of https://github.com/llvm/torch-mlir
c2d3820e48
The current code was inserting all build_list ops after the last constant op since it was assuming that all elements being passed in were constants. This patch replaces that patch with a new function that inserts the build_list ops before the terminator. Also modifies test_export_conv2d_fwd.py since its output no longer matches. TEST: Added test_export_cat.py which is the code in #102 |
||
---|---|---|
.. | ||
test_conv_nllloss_grads.py | ||
test_export_ResA.py | ||
test_export_add3.py | ||
test_export_batchnorm.py | ||
test_export_cat.py | ||
test_export_conv2d_fwd.py | ||
test_export_multi_out.py |