tvm
Classes | Namespaces | Functions
extracted_task.h File Reference
#include <tvm/ir/module.h>
#include <tvm/node/reflection.h>
#include <tvm/runtime/container/array.h>
#include <tvm/runtime/container/string.h>
#include <tvm/runtime/object.h>
#include <tvm/target/target.h>
Include dependency graph for extracted_task.h:

Go to the source code of this file.

Classes

class  tvm::meta_schedule::ExtractedTaskNode
 A tuning task extracted from the high-level IR. More...
 
class  tvm::meta_schedule::ExtractedTask
 Managed reference to ExtractedTaskNode. More...
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::tir
 
 tvm::te
 Tensor expression language DSL.
 
 tvm::meta_schedule
 

Functions

Optional< tvm::tir::PrimFunctvm::meta_schedule::DefaultTaskFilter (const Array< tvm::te::Tensor, void > &args)
 The default TE task filter. More...
 
Optional< tir::PrimFunc > tvm::meta_schedule::DefaultTaskFilterAllowExtern (const Array< tvm::te::Tensor, void > &args)
 The default TE task filter, with te.extern allowed. More...