#include <DefaultCostModel.h>
Inherits Halide::CostModel.
Definition at line 10 of file DefaultCostModel.h.
◆ DefaultCostModel()
Halide::DefaultCostModel::DefaultCostModel |
( |
const std::string & |
weights_in_path, |
|
|
const std::string & |
weights_out_path, |
|
|
bool |
randomize_weights |
|
) |
| |
|
inline |
◆ ~DefaultCostModel()
Halide::DefaultCostModel::~DefaultCostModel |
( |
| ) |
|
|
overridedefault |
◆ set_pipeline_features() [1/2]
◆ set_pipeline_features() [2/2]
void Halide::DefaultCostModel::set_pipeline_features |
( |
const Runtime::Buffer< float > & |
, |
|
|
int |
n |
|
) |
| |
◆ enqueue() [1/2]
◆ enqueue() [2/2]
void Halide::DefaultCostModel::enqueue |
( |
int |
ns, |
|
|
Runtime::Buffer< float > * |
schedule_feats, |
|
|
double * |
cost_ptr |
|
) |
| |
◆ evaluate_costs()
void Halide::DefaultCostModel::evaluate_costs |
( |
| ) |
|
|
overridevirtual |
◆ reset()
void Halide::DefaultCostModel::reset |
( |
| ) |
|
|
overridevirtual |
◆ backprop()
float Halide::DefaultCostModel::backprop |
( |
const Runtime::Buffer< const float > & |
true_runtimes, |
|
|
float |
learning_rate |
|
) |
| |
◆ save_weights()
void Halide::DefaultCostModel::save_weights |
( |
| ) |
|
◆ load_weights()
void Halide::DefaultCostModel::load_weights |
( |
| ) |
|
The documentation for this class was generated from the following file: