tvm
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Namespaces | Functions
injective.h File Reference

x86 schedule for injective ops More...

#include <tvm/target/generic_func.h>
#include <tvm/te/operation.h>
#include <tvm/topi/detail/fuse.h>
#include <tvm/topi/tags.h>
Include dependency graph for injective.h:

Go to the source code of this file.

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::topi
 
 tvm::topi::x86
 

Functions

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

Detailed Description

x86 schedule for injective ops