Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
tc::autotune::GeneticAutotunerATen Class Reference

#include <genetic_autotuner_aten.h>

Collaboration diagram for tc::autotune::GeneticAutotunerATen:

Public Member Functions

 GeneticAutotunerATen (const std::string tc)
 
std::vector< MappingOptionsload (const std::string &cacheFileName, const std::string &tcName, const std::vector< at::Tensor > inputs, const size_t numCandidates)
 
llvm::Optional< MappingOptionstune (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_
 

Constructor & Destructor Documentation

tc::autotune::GeneticAutotunerATen::GeneticAutotunerATen ( const std::string  tc)

Member Function Documentation

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 = {} 
)

Member Data Documentation

std::unique_ptr<detail::GeneticAutotuner> tc::autotune::GeneticAutotunerATen::geneticAutotuner_
private
std::string tc::autotune::GeneticAutotunerATen::tc_
private

The documentation for this class was generated from the following file: