softmax.h File Reference

Softmax op constructions. More...

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

Include dependency graph for softmax.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::softmax (const Tensor &x, int axis=-1, std::string name="tensor", std::string tag="softmax_output")
	Softmax activation. More...
Tensor	tvm::topi::nn::log_softmax (const Tensor &x, std::string name="tensor", std::string tag="log_softmax_output")
	Log softmax activation. More...

Detailed Description

Softmax op constructions.