![]() |
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 |