flatten.h File Reference

Softmax op constructions. More...

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

Include dependency graph for flatten.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
Tensor	tvm::topi::nn::flatten (const Tensor &x, std::string name="tensor", std::string tag=kInjective)
	Flattens the input tensor into a 2-D tensor by collapsing higher dimensions. This requires the input tensor to have constant sized dimensions. More...

Detailed Description

Softmax op constructions.