reorg.h File Reference

Reorg op constructions. More...

#include <tvm/te/operation.h>
#include <tvm/topi/detail/constant_utils.h>
#include <tvm/topi/reduction.h>
#include <tvm/topi/tags.h>
#include <tvm/topi/transform.h>
#include <algorithm>
#include <string>

Include dependency graph for reorg.h:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi
	tvm::topi::vision

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

Detailed Description

Reorg op constructions.