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

Detailed Description

Mapping op constructions.