tvm
|
instance normalization op constructions More...
#include <tvm/te/operation.h>
#include <tvm/topi/nn/layer_norm.h>
#include <tvm/topi/tags.h>
#include <string>
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... | |
instance normalization op constructions