tvm::topi::contrib Namespace Reference

Functions
Tensor	cublas_matmul (const Tensor &lhs, const Tensor &rhs, bool transa, bool transb)
	Create an op that multiplies lhs and rhs with cuBLAS. More...
Tensor	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...

Function Documentation

cublas_batch_matmul()

Tensor tvm::topi::contrib::cublas_batch_matmul ( const Tensor &  lhs, const Tensor &  rhs, bool  transa, bool  transb  )

inline

Create an op that multiplies batch matrices lhs and rhs with cuBLAS.

Parameters

lhs	The left matrix operand
rhs	The right matrix operand
transa	Whether to transpose lhs
transb	Whether to transpose rhs

Returns

The output tensor

cublas_matmul()

Tensor tvm::topi::contrib::cublas_matmul ( const Tensor &  lhs, const Tensor &  rhs, bool  transa, bool  transb  )

inline

Create an op that multiplies lhs and rhs with cuBLAS.

Parameters

lhs	The left matrix operand
rhs	The right matrix operand
transa	Whether to transpose lhs
transb	Whether to transpose rhs

Returns

The output tensor