Managed reference to FeatureExtractorNode.
More...
#include <feature_extractor.h>
◆ 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()
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_DEFINE_MUTABLE_OBJECT_REF_METHODS()
The documentation for this class was generated from the following file: