tvm::topi::vision Namespace Reference

Functions
Tensor	reorg (const Tensor &data, int stride=1, std::string name="tensor", std::string tag="reorg_output")
	Reorg operation. More...

Function Documentation

reorg()

Tensor tvm::topi::vision::reorg ( const Tensor &  data, int  stride = 1, std::string  name = "tensor", std::string  tag = "reorg_output"  )

inline

Reorg operation.

Parameters

data	The input tensor. Can be any dimension
stride	The input integer used as stride in reorg operation
name	The name of the operation
tag	The tag to mark the operation

Returns

A Tensor whose op member is the reorg operation