bias_add.h File Reference

bias_add op constructions More...

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

Include dependency graph for bias_add.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::bias_add (const tvm::te::Tensor &data, const tvm::te::Tensor &bias, int axis)
	Creates an operation that calculates data + bias.

Detailed Description

bias_add op constructions