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
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::topi
namespace	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.
Tensor	tvm::topi::nn::log_softmax (const Tensor &x, std::string name="tensor", std::string tag="log_softmax_output")
	Log softmax activation.

Detailed Description

Softmax op constructions.