injective.h File Reference

CUDA schedule for injective operations. More...

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

Include dependency graph for injective.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

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

Functions
Schedule	tvm::topi::cuda::schedule_injective_from_existing (Schedule sch, const Tensor &out)
	Updates an existing schedule for the given injective ops. More...
Schedule	tvm::topi::cuda::schedule_injective (const Target &target, const Array< Tensor > &outs)
	Create a CUDA schedule for the given output tensors. More...

Detailed Description

CUDA schedule for injective operations.