mirror of https://github.com/llvm/torch-mlir
6c702b149f
* 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 |
||
|---|---|---|
| .. | ||
| ATenLayerNamePass.cpp | ||
| ATenLoweringPass.cpp | ||
| ATenOpReport.cpp | ||
| ATenToStd.cpp | ||
| CMakeLists.txt | ||
| LivenessReport.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| RecognizeKernelsPass.cpp | ||
| ReturnEliminationPass.cpp | ||