-
Amy Zhuang authored
* [MLIR] Use call back for MatMul. * Use callback for Gemm. * Use mkldnn callback for Softmax. * Address PR feedback. * Fix merge errors. * Change to tail allocation struct. * Use mkldnn callback for AvgPool. * Add callbacks for AvgPoolBackprop, MaxPool, and MaxPoolBackprop. * Fix merge errors. * Use UnrankedMemRefType for callbacks. * Address PR feedback. * Cleanup. * Address PR feedback. * Fix a bug. * Use global variable to hold attributes. * Convert layout if needed for pooling. * Address PR feedback. * Add header. * Address PR feedback. * Update Copyright to 2017-2020. * Address PR feedback. Co-authored-by: Scott Cyphers <diyessi@users.noreply.github.com>
c737a573
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
analysis | ||
cpu | ||
pass | ||
backend.hpp |