tvm::meta_schedule::FeatureExtractorNode Class Reference

Extractor for features from measure candidates for use in cost model. More...

#include <feature_extractor.h>

Inheritance diagram for tvm::meta_schedule::FeatureExtractorNode:

Collaboration diagram for tvm::meta_schedule::FeatureExtractorNode:

Public Member Functions
virtual	~FeatureExtractorNode ()=default
	Virtual destructor. More...
virtual Array< tvm::runtime::NDArray >	ExtractFrom (const TuneContext &context, const Array< MeasureCandidate > &candidates)=0
	Extract features from the given measure candidate. More...
	TVM_DECLARE_BASE_OBJECT_INFO (FeatureExtractorNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.FeatureExtractor"

Detailed Description

Extractor for features from measure candidates for use in cost model.

Constructor & Destructor Documentation

~FeatureExtractorNode()

virtual tvm::meta_schedule::FeatureExtractorNode::~FeatureExtractorNode ( )

virtualdefault

Virtual destructor.

Member Function Documentation

ExtractFrom()

virtual Array<tvm::runtime::NDArray> tvm::meta_schedule::FeatureExtractorNode::ExtractFrom ( const TuneContext &  context, const Array< MeasureCandidate > &  candidates  )

pure virtual

Extract features from the given measure candidate.

Parameters

context	The tuning context for feature extraction.
candidates	The measure candidates to extract features from.

Returns

The feature ndarray extracted.

Implemented in tvm::meta_schedule::PyFeatureExtractorNode.

RegisterReflection()

static void tvm::meta_schedule::FeatureExtractorNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_BASE_OBJECT_INFO()

tvm::meta_schedule::FeatureExtractorNode::TVM_DECLARE_BASE_OBJECT_INFO	(	FeatureExtractorNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::FeatureExtractorNode::_type_key = "meta_schedule.FeatureExtractor"

staticconstexpr

---

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

• include/tvm/meta_schedule/feature_extractor.h