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

#include <genetic_autotuner.h>

Collaboration diagram for tc::autotune::detail::GeneticAutotuner:

Public Member Functions

 GeneticAutotuner (const std::string &tc)
 
void storeCaches (const std::string &filename)
 
std::vector< MappingOptionsload (const std::string &cacheFileName, const std::string &tcName, const std::vector< const DLTensor * > &inputs, const size_t numCandidates)
 
llvm::Optional< MappingOptionstune (const std::string &cacheFileName, const std::string &tcName, const std::unordered_map< size_t, std::vector< const DLTensor * >> &inputs, std::unordered_map< size_t, std::vector< DLTensor * >> &outputs, MappingOptions baseMapping, std::vector< MappingOptions > startingPoints, const TuningParameterFixer &fixedParams)
 

Private Attributes

std::string tc_
 
std::map< std::string,
lang::TreeRef
tcNameMap_
 

Constructor & Destructor Documentation

tc::autotune::detail::GeneticAutotuner::GeneticAutotuner ( const std::string &  tc)
explicit

Member Function Documentation

std::vector<MappingOptions> tc::autotune::detail::GeneticAutotuner::load ( const std::string &  cacheFileName,
const std::string &  tcName,
const std::vector< const DLTensor * > &  inputs,
const size_t  numCandidates 
)
void tc::autotune::detail::GeneticAutotuner::storeCaches ( const std::string &  filename)
llvm::Optional<MappingOptions> tc::autotune::detail::GeneticAutotuner::tune ( const std::string &  cacheFileName,
const std::string &  tcName,
const std::unordered_map< size_t, std::vector< const DLTensor * >> &  inputs,
std::unordered_map< size_t, std::vector< DLTensor * >> &  outputs,
MappingOptions  baseMapping,
std::vector< MappingOptions startingPoints,
const TuningParameterFixer fixedParams 
)

Member Data Documentation

std::string tc::autotune::detail::GeneticAutotuner::tc_
private
std::map<std::string, lang::TreeRef> tc::autotune::detail::GeneticAutotuner::tcNameMap_
private

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