torch-mlir/frontends/pytorch/csrc/c10_dispatch
Stella Laurenzo 58adb6bd8e Work around various PyTorch issues in support of convolution.
* Enables the conv2d fwd test and ResA (which are both small).
* Deletes resnet18 and vgg, which both run but generate output that crashes FileCheck and lit (or at least makes them take an eternity).
2020-10-21 12:44:31 -07:00
..
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
acap_dispatch.cpp Work around various PyTorch issues in support of convolution. 2020-10-21 12:44:31 -07:00
acap_dispatch.h Work around various PyTorch issues in support of convolution. 2020-10-21 12:44:31 -07:00
func_builder.cpp Add remaining pieces to capture full example models. 2020-10-19 22:16:59 -07:00
func_builder.h Add remaining pieces to capture full example models. 2020-10-19 22:16:59 -07:00
module_builder.cpp Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
module_builder.h Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
python_bindings.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00