local_response_norm.h File Reference

local response normalization op constructions More...

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

Include dependency graph for local_response_norm.h:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi
	tvm::topi::nn

Functions
Tensor	tvm::topi::nn::lrn (const Tensor &data, int size, int axis=1, float alpha=0.0001, float beta=0.75, float bias=2, std::string name="tensor", std::string tag=kBroadcast)
	Local response normalization inference operator. More...

Detailed Description

local response normalization op constructions