Tensor Comprehensions
|
#include <cstdint>
#include <memory>
#include <mutex>
#include <stdexcept>
#include <string>
#include <vector>
#include <dlpack/dlpack.h>
#include <compcache.pb.h>
#include "tc/core/mapping_options.h"
#include "tc/core/utils/cuda_info.h"
#include "tc/core/compilation_cache-inl.h"
Go to the source code of this file.
Classes | |
struct | tc::detail::TensorInfo |
class | tc::Cache< CC > |
class | tc::CacheEntrySameKeyDifferentValue |
class | tc::CudaCache |
struct | tc::CudaCache::RetrievalResult |
struct | tc::CudaCache::CachedEntry |
struct | tc::CudaCache::CachedEntry::Key |
struct | tc::CudaCache::CachedEntry::Values |
class | tc::OptionsCache |
struct | tc::OptionsCache::CachedEntry |
struct | tc::OptionsCache::CachedEntry::Key |
struct | tc::OptionsCache::CachedEntry::Values |
struct | tc::OptionsCache::RetrievalResult |
class | tc::ManualCudaCache |
struct | tc::ManualCudaCache::CachedEntry |
struct | tc::ManualCudaCache::CachedEntry::Key |
struct | tc::ManualCudaCache::CachedEntry::Values |
Namespaces | |
tc | |
tc::detail | |
Functions | |
void | tc::removeFromCudaCacheEntriesNotInOptionsCache (CudaCache &cc, const OptionsCache &oc) |
bool | tc::operator== (const std::vector< const DLTensor * > &inputsTensor, const std::vector< detail::TensorInfo > &inputsInfo) |
std::string | tc::makeOptionsFilename (const std::string &filename) |
std::string | tc::makeCudaFilename (const std::string &filename) |