tvm::s_tir::meta_schedule::CostModel Class Reference

Managed reference to CostModelObj. More...

#include <cost_model.h>

Inheritance diagram for tvm::s_tir::meta_schedule::CostModel:

Collaboration diagram for tvm::s_tir::meta_schedule::CostModel:

Public Member Functions
	CostModel (ffi::UnsafeInit tag, ffi::ObjectPtr< CostModelObj > data)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (CostModel, ffi::ObjectRef, CostModelObj)

Static Public Member Functions
static CostModel	PyCostModel (PyCostModelObj::FLoad f_load, PyCostModelObj::FSave f_save, PyCostModelObj::FUpdate f_update, PyCostModelObj::FPredict f_predict)
	Create a cost model with customized methods on the python-side.

Detailed Description

Managed reference to CostModelObj.

See also

CostModelObj

Constructor & Destructor Documentation

CostModel()

tvm::s_tir::meta_schedule::CostModel::CostModel ( ffi::UnsafeInit  tag, ffi::ObjectPtr< CostModelObj >  data  )

inlineexplicit

Member Function Documentation

PyCostModel()

static CostModel tvm::s_tir::meta_schedule::CostModel::PyCostModel ( PyCostModelObj::FLoad  f_load, PyCostModelObj::FSave  f_save, PyCostModelObj::FUpdate  f_update, PyCostModelObj::FPredict  f_predict  )

static

Create a cost model with customized methods on the python-side.

Parameters

f_load	The packed function of Load.
f_save	The packed function of Save.
f_update	The packed function of Update.
f_predict	The packed function of Predict.

Returns

The cost model created.

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::s_tir::meta_schedule::CostModel::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE	(	CostModel	,
		ffi::ObjectRef	,
		CostModelObj
	)

---

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

• include/tvm/s_tir/meta_schedule/cost_model.h