mapping.h File Reference

Mapping op constructions. More...

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

Include dependency graph for mapping.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::scale_shift_nchw (const Tensor &x, const Tensor &scale, const Tensor &shift, std::string name="ScaleShift", std::string tag=kBroadcast)
	Scale and shift with NCHW order. More...
Tensor	tvm::topi::nn::scale_shift_nhwc (const Tensor &x, const Tensor &scale, const Tensor &shift, std::string name="ScaleShift", std::string tag=kBroadcast)
	Scale and shift with NHWC order. More...

Detailed Description

Mapping op constructions.