cost_model.h File Reference

Cost models that estimate the performance of programs. More...

#include <tvm/auto_scheduler/compute_dag.h>
#include <tvm/auto_scheduler/measure.h>
#include <tvm/node/node.h>
#include <tvm/runtime/packed_func.h>
#include <vector>

Include dependency graph for cost_model.h:

Go to the source code of this file.

Classes
class	tvm::auto_scheduler::CostModelNode
	The base class for cost model. More...
class	tvm::auto_scheduler::CostModel
	Managed reference to CostModelNode. More...
class	tvm::auto_scheduler::RandomModelNode
	The cost model returning random value for all predictions. More...
class	tvm::auto_scheduler::RandomModel
	Managed reference to RandomModelNode. More...
class	tvm::auto_scheduler::PythonBasedModelNode
	A wrapper for cost model defined by python code This class will call functions defined in the python. More...
class	tvm::auto_scheduler::PythonBasedModel
	Managed reference to PythonBasedModelNode. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::auto_scheduler

Detailed Description

Cost models that estimate the performance of programs.