The feature extractor with customized methods on the python-side. More...

#include <feature_extractor.h>

Inheritance diagram for tvm::meta_schedule::PyFeatureExtractorNode:

Collaboration diagram for tvm::meta_schedule::PyFeatureExtractorNode:

Public Types
using	FExtractFrom = ffi::TypedFunction< Array< tvm::runtime::NDArray >(const TuneContext &context, const Array< MeasureCandidate > &candidates)>
	Extract features from the given measure candidate. More...

using	FAsString = ffi::TypedFunction< String()>
	Get the feature extractor as string with name. More...

Public Member Functions
Array< tvm::runtime::NDArray >	ExtractFrom (const TuneContext &context, const Array< MeasureCandidate > &candidates) final
	Extract features from the given measure candidate. More...

	TVM_DECLARE_FINAL_OBJECT_INFO (PyFeatureExtractorNode, FeatureExtractorNode)

Public Member Functions inherited from tvm::meta_schedule::FeatureExtractorNode
virtual	~FeatureExtractorNode ()=default
	Virtual destructor. More...

	TVM_DECLARE_BASE_OBJECT_INFO (FeatureExtractorNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::meta_schedule::FeatureExtractorNode
static void	RegisterReflection ()

Public Attributes
FExtractFrom	f_extract_from
	The packed function to the `ExtractFrom` function. More...

FAsString	f_as_string
	The packed function to the `AsString` function. More...

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

Static Public Attributes inherited from tvm::meta_schedule::FeatureExtractorNode
static constexpr const char *	_type_key = "meta_schedule.FeatureExtractor"

Detailed Description

The feature extractor with customized methods on the python-side.

Member Typedef Documentation

◆ FAsString

using tvm::meta_schedule::PyFeatureExtractorNode::FAsString = ffi::TypedFunction<String()>

Get the feature extractor as string with name.

Returns: The string of the feature extractor.

◆ FExtractFrom

using tvm::meta_schedule::PyFeatureExtractorNode::FExtractFrom = ffi::TypedFunction<Array<tvm::runtime::NDArray>( const TuneContext& context, const Array<MeasureCandidate>& candidates)>

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.

Member Function Documentation

◆ ExtractFrom()

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

finalvirtual

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.

Implements tvm::meta_schedule::FeatureExtractorNode.

◆ RegisterReflection()

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

inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::PyFeatureExtractorNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PyFeatureExtractorNode	,
		FeatureExtractorNode
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::meta_schedule::PyFeatureExtractorNode::_type_key = "meta_schedule.PyFeatureExtractor"

staticconstexpr

◆ f_as_string

FAsString tvm::meta_schedule::PyFeatureExtractorNode::f_as_string

The packed function to the AsString function.

◆ f_extract_from

FExtractFrom tvm::meta_schedule::PyFeatureExtractorNode::f_extract_from

The packed function to the ExtractFrom function.

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

include/tvm/meta_schedule/feature_extractor.h

Public Types

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes