Files
apache--tvm/python
Matthew Barrett 5cacecc0c0 [CUBLAS] Add support for nn.dense and nn.batch_matmul (#10826)
* [CUBLAS] Add support for nn.dense and nn.batch_matmul

This commit includes a fix for cublas.batch_matmul
when mixed precision is being used.

* Specify args in dense
2022-03-31 13:46:31 +09:00
..