Tensor Comprehensions
|
#include <genetic_autotuner_aten.h>
Public Member Functions | |
GeneticAutotunerATen (const std::string tc) | |
std::vector< MappingOptions > | load (const std::string &cacheFileName, const std::string &tcName, const std::vector< at::Tensor > inputs, const size_t numCandidates) |
llvm::Optional< MappingOptions > | tune (const std::string &cacheFileName, const std::string &tcName, const std::vector< at::Tensor > &inputs, MappingOptions baseMapping, std::vector< MappingOptions > startingPoints, const TuningParameterFixer &fixedParams={}) |
Private Attributes | |
std::string | tc_ |
std::unique_ptr < detail::GeneticAutotuner > | geneticAutotuner_ |
tc::autotune::GeneticAutotunerATen::GeneticAutotunerATen | ( | const std::string | tc | ) |
std::vector<MappingOptions> tc::autotune::GeneticAutotunerATen::load | ( | const std::string & | cacheFileName, |
const std::string & | tcName, | ||
const std::vector< at::Tensor > | inputs, | ||
const size_t | numCandidates | ||
) |
llvm::Optional<MappingOptions> tc::autotune::GeneticAutotunerATen::tune | ( | const std::string & | cacheFileName, |
const std::string & | tcName, | ||
const std::vector< at::Tensor > & | inputs, | ||
MappingOptions | baseMapping, | ||
std::vector< MappingOptions > | startingPoints, | ||
const TuningParameterFixer & | fixedParams = {} |
||
) |
|
private |
|
private |