tvm::meta_schedule::FeatureExtractor Class Reference

Managed reference to FeatureExtractorNode. More...

#include <feature_extractor.h>

Inheritance diagram for tvm::meta_schedule::FeatureExtractor:

Collaboration diagram for tvm::meta_schedule::FeatureExtractor:

Public Member Functions
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (FeatureExtractor, ObjectRef, FeatureExtractorNode)

Static Public Member Functions
static FeatureExtractor	PerStoreFeature (int buffers_per_store=5, int arith_intensity_curve_num_samples=10, int cache_line_bytes=64, bool extract_workload=false)
	Create a feature extractor that extracts features from each BufferStore. More...
static FeatureExtractor	PyFeatureExtractor (PyFeatureExtractorNode::FExtractFrom f_extract_from, PyFeatureExtractorNode::FAsString f_as_string)
	Create a feature extractor with customized methods on the python-side. More...

Detailed Description

Managed reference to FeatureExtractorNode.

See also

FeatureExtractorNode

Member Function Documentation

PerStoreFeature()

static FeatureExtractor tvm::meta_schedule::FeatureExtractor::PerStoreFeature ( int  buffers_per_store = 5, int  arith_intensity_curve_num_samples = 10, int  cache_line_bytes = 64, bool  extract_workload = false  )

static

Create a feature extractor that extracts features from each BufferStore.

Parameters

buffers_per_store	The number of buffers in each BufferStore; Pad or truncate if necessary.
arith_intensity_curve_num_samples	The number of samples used in the arithmetic intensity curve.
cache_line_bytes	The number of bytes in a cache line.
extract_workload	Whether to extract features in the workload in tuning context or not.

Returns

The feature extractor created.

PyFeatureExtractor()

static FeatureExtractor tvm::meta_schedule::FeatureExtractor::PyFeatureExtractor ( PyFeatureExtractorNode::FExtractFrom  f_extract_from, PyFeatureExtractorNode::FAsString  f_as_string  )

static

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

Parameters

f_extract_from	The packed function of ExtractFrom.
f_as_string	The packed function of AsString.

Returns

The feature extractor created.

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::meta_schedule::FeatureExtractor::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	FeatureExtractor	,
		ObjectRef	,
		FeatureExtractorNode
	)

---

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

• include/tvm/meta_schedule/feature_extractor.h