dense.h File Reference

rocm schedule for dense operation More...

#include <tvm/target/generic_func.h>
#include <tvm/te/operation.h>
#include <tvm/topi/contrib/rocblas.h>
#include <tvm/topi/cuda/dense.h>
#include <tvm/topi/detail/array_utils.h>
#include <tvm/topi/generic/extern.h>
#include <tvm/topi/nn/dense.h>
#include <tvm/topi/tags.h>

Include dependency graph for dense.h:

Go to the source code of this file.

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

Functions
tvm::te::Tensor	tvm::topi::rocm::dense_rocm (const Target &target, const tvm::te::Tensor &data, const tvm::te::Tensor &weight, const tvm::te::Tensor &bias, const DataType &out_dtype)
	Implementation of dense for rocm backend. More...
Schedule	tvm::topi::rocm::schedule_dense (const Target &target, const Array< Tensor > &outs)
	Create a rocm schedule for dense. More...

Detailed Description

rocm schedule for dense operation