dense.h File Reference

Dense op constructions. More...

#include <tvm/te/operation.h>
#include <tvm/topi/tags.h>
#include <string>

Include dependency graph for dense.h:

Go to the source code of this file.

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::topi
namespace	tvm::topi::nn

Functions
tvm::te::Tensor	tvm::topi::nn::dense (const tvm::te::Tensor &data, const tvm::te::Tensor &weight, const tvm::te::Tensor &bias, const PrimType &out_dtype)
	Creates an operation that calculates data * weight^T + bias.

Detailed Description

Dense op constructions.