tvm
Public Member Functions | Static Public Member Functions | List of all members
tvm::meta_schedule::CostModel Class Reference

Managed reference to CostModelNode. More...

#include <cost_model.h>

Inheritance diagram for tvm::meta_schedule::CostModel:
Collaboration diagram for tvm::meta_schedule::CostModel:

Public Member Functions

 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (CostModel, ObjectRef, CostModelNode)
 

Static Public Member Functions

static CostModel PyCostModel (PyCostModelNode::FLoad f_load, PyCostModelNode::FSave f_save, PyCostModelNode::FUpdate f_update, PyCostModelNode::FPredict f_predict, PyCostModelNode::FAsString f_as_string)
 Create a feature extractor with customized methods on the python-side. More...
 

Detailed Description

Managed reference to CostModelNode.

See also
CostModelNode

Member Function Documentation

◆ PyCostModel()

static CostModel tvm::meta_schedule::CostModel::PyCostModel ( PyCostModelNode::FLoad  f_load,
PyCostModelNode::FSave  f_save,
PyCostModelNode::FUpdate  f_update,
PyCostModelNode::FPredict  f_predict,
PyCostModelNode::FAsString  f_as_string 
)
static

Create a feature extractor with customized methods on the python-side.

Parameters
f_loadThe packed function of Load.
f_saveThe packed function of Save.
f_updateThe packed function of Update.
f_predictThe packed function of Predict.
f_as_stringThe packed function of AsString.
Returns
The feature extractor created.

◆ TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS()

tvm::meta_schedule::CostModel::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS ( CostModel  ,
ObjectRef  ,
CostModelNode   
)

The documentation for this class was generated from the following file: