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

rocblas_batch_matmul()

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

inline

Create an op that batch multiplies lhs and rhs with rocBLAS.

Parameters

lhs	The left matrix operand e.g. (batch_size, M, K)
rhs	The right matrix operand e.g. (batch_size, K, N)
transa	Whether to transpose lhs
transb	Whether to transpose rhs

Returns

The output tensor

rocblas_matmul()

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

inline

Create an op that multiplies lhs and rhs with rocBLAS.

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