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< ffi::Array< tvm::runtime::Tensor >(const TuneContext &context, const ffi::Array< MeasureCandidate > &candidates)>
	Extract features from the given measure candidate. More...

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

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

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("meta_schedule.PyFeatureExtractor", PyFeatureExtractorNode, FeatureExtractorNode)

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

	TVM_FFI_DECLARE_OBJECT_INFO ("meta_schedule.FeatureExtractor", 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...

Detailed Description

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

Member Typedef Documentation

◆ FAsString

using tvm::meta_schedule::PyFeatureExtractorNode::FAsString = ffi::TypedFunction<ffi::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<ffi::Array<tvm::runtime::Tensor>( const TuneContext& context, const ffi::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 tensor extracted.

Member Function Documentation

◆ ExtractFrom()

ffi::Array<tvm::runtime::Tensor> tvm::meta_schedule::PyFeatureExtractorNode::ExtractFrom	(	const TuneContext &	context,
		const ffi::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 tensor extracted.

Implements tvm::meta_schedule::FeatureExtractorNode.

◆ RegisterReflection()

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

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::meta_schedule::PyFeatureExtractorNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"meta_schedule.PyFeatureExtractor"	,
		PyFeatureExtractorNode	,
		FeatureExtractorNode
	)

Member Data Documentation

◆ 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