cublas.h File Reference

External function interface to cuBLAS libraries. More...

#include <tvm/te/operation.h>
#include <tvm/topi/detail/extern.h>

Include dependency graph for cublas.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi
	tvm::topi::contrib

Functions
Tensor	tvm::topi::contrib::cublas_matmul (const Tensor &lhs, const Tensor &rhs, bool transa, bool transb)
	Create an op that multiplies lhs and rhs with cuBLAS. More...
Tensor	tvm::topi::contrib::cublas_batch_matmul (const Tensor &lhs, const Tensor &rhs, bool transa, bool transb)
	Create an op that multiplies batch matrices lhs and rhs with cuBLAS. More...

Detailed Description

External function interface to cuBLAS libraries.