torch-mlir/frontends/pytorch/test/acap_export
Stella Laurenzo 8d98dd4551 Support optional args/returns and other odds and ends.
* None's out Device? args.
* Emits bool tensors if needed.
* Adds some stderr tracing to better see what is going on.
* Test case that exercises NLLLoss.
* This test case emits something for backward calculations but there are some issues still to be worked out, so that part is left out of the test case.
* Progress on #97
2020-10-30 14:50:28 -07:00
..
test_export_ResA.py NFC: Use new print with large_elements_limit in tests. 2020-10-22 13:04:24 -07:00
test_export_add3.py Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
test_export_batchnorm.py More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
test_export_conv2d_fwd.py Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
test_export_multi_out.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_nllloss.py Support optional args/returns and other odds and ends. 2020-10-30 14:50:28 -07:00