tvm
Namespaces | Functions
instance_norm.h File Reference

instance normalization op constructions More...

#include <tvm/te/operation.h>
#include <tvm/topi/nn/layer_norm.h>
#include <tvm/topi/tags.h>
#include <string>
Include dependency graph for instance_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::instance_norm (const Tensor &data, const Tensor &gamma, const Tensor &beta, const Array< Integer > &axis, double epsilon, std::string name="T_instance_norm", std::string tag=kInjective)
 Instance normalization. More...
 

Detailed Description

instance normalization op constructions