Cost model.
More...
#include <cost_model.h>
|
static constexpr const char * | _type_key = "meta_schedule.CostModel" |
|
◆ ~CostModelNode()
virtual tvm::meta_schedule::CostModelNode::~CostModelNode |
( |
| ) |
|
|
virtualdefault |
◆ Load()
virtual void tvm::meta_schedule::CostModelNode::Load |
( |
const String & |
path | ) |
|
|
pure virtual |
◆ Predict()
virtual std::vector<double> tvm::meta_schedule::CostModelNode::Predict |
( |
const TuneContext & |
context, |
|
|
const Array< MeasureCandidate > & |
candidates |
|
) |
| |
|
pure virtual |
Predict the normalized score (the larger the better) of given measure candidates.
- Parameters
-
context | The tuning context. |
candidates | The measure candidates. |
- Returns
- The predicted normalized score.
Implemented in tvm::meta_schedule::PyCostModelNode.
◆ Save()
virtual void tvm::meta_schedule::CostModelNode::Save |
( |
const String & |
path | ) |
|
|
pure virtual |
◆ TVM_DECLARE_BASE_OBJECT_INFO()
tvm::meta_schedule::CostModelNode::TVM_DECLARE_BASE_OBJECT_INFO |
( |
CostModelNode |
, |
|
|
Object |
|
|
) |
| |
◆ Update()
Update the cost model given running results.
- Parameters
-
context | The tuning context. |
candidates | The measure candidates. |
results | The running results of the measure candidates. |
Implemented in tvm::meta_schedule::PyCostModelNode.
◆ _type_key
constexpr const char* tvm::meta_schedule::CostModelNode::_type_key = "meta_schedule.CostModel" |
|
staticconstexpr |
The documentation for this class was generated from the following file: