tvm
Loading...
Searching...
No Matches
Namespaces | Functions
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

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::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.
 

Detailed Description

local response normalization op constructions