tvm
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Namespaces | Functions
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

 tvm
 Performance counters for profiling via the PAPI library.
 
 tvm::topi
 
 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 DataType &out_dtype)
 Creates an operation that calculates data * weight^T + bias. More...
 

Detailed Description

Dense op constructions.