torch-mlir/frontends/pytorch/csrc
Stella Laurenzo 6c702b149f Add a number of kernels and new patterns.
* convolution, convolution_backward, _log_softmax, _log_softmax_backward_data, nll_loss_forward, nll_loss_backward, nll_loss2d_forward, nll_loss2d_backward, copy_
* Extends the recognition logic and metadata for handling inplace transformations, optional tensors, ints, lists and dropped args.
* The kernel_calls generated by test_conv_nllloss_grads.py now convert to ATen.
* The result *almost* comes out as a pure tensor program with the exception of the copy_ op, which I will do some followup work to deal with.
* More progress on #97
2020-11-04 14:36:59 -08:00
..
c10_dispatch Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
type_dispatch Add capture function arguments. 2020-10-01 18:59:58 -07:00
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
init_python_bindings.cpp NFC: Transition ATen passes to tablegen registration. 2020-10-22 17:12:44 -07:00
init_python_bindings.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
pybind.h Add capture function arguments. 2020-10-01 18:59:58 -07:00