Tensor Comprehensions
|
#include <compilation_cache.h>
Classes | |
struct | Key |
struct | Values |
Public Member Functions | |
CachedEntry (const std::string &id, const std::vector< const DLTensor * > &inputs, const std::vector< const DLTensor * > &outputs, const std::string &deviceStr, const MappingOptions &options, Duration runtime) | |
CachedEntry (const OptionsCacheEntryProto &buf) | |
OptionsCacheEntryProto | toProtobuf () const |
Public Attributes | |
Key | key |
std::vector< Values > | values |
An OptionsCache holds multiple CachedEntry's. Each CachedEntry is split to two conceptual parts the key and the values. The key is: the kernel/op's unique id (string), the specialized input dimensions, the target architecture (string), tc's version (string), The values are a vector of: the isl options used when the kernel was optimized, profiling information
tc::OptionsCache::CachedEntry::CachedEntry | ( | const std::string & | id, |
const std::vector< const DLTensor * > & | inputs, | ||
const std::vector< const DLTensor * > & | outputs, | ||
const std::string & | deviceStr, | ||
const MappingOptions & | options, | ||
Duration | runtime | ||
) |
tc::OptionsCache::CachedEntry::CachedEntry | ( | const OptionsCacheEntryProto & | buf | ) |
OptionsCacheEntryProto tc::OptionsCache::CachedEntry::toProtobuf | ( | ) | const |
Key tc::OptionsCache::CachedEntry::key |
std::vector<Values> tc::OptionsCache::CachedEntry::values |