reduction.h File Reference

rocm schedule for reduction operations More...

#include <tvm/target/generic_func.h>
#include <tvm/te/operation.h>
#include <tvm/topi/cuda/reduction.h>
#include <tvm/topi/detail/fuse.h>
#include <tvm/topi/tags.h>

Include dependency graph for reduction.h:

Go to the source code of this file.

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

Functions
Schedule	tvm::topi::rocm::schedule_reduce (const Target &target, Array< Tensor > outs)
	Create a rocm schedule for a reduce operation. More...

Detailed Description

rocm schedule for reduction operations