Tensor Comprehensions
|
#include <vector>
#include <ATen/ATen.h>
#include "tc/core/cuda.h"
#include "tc/core/mapping_options.h"
#include "tc/core/utils/dlpack.h"
#include <llvm/ADT/Optional.h>
#include "tc/autotuner/utils/utils-inl.h"
Go to the source code of this file.
Classes | |
struct | tc::autotune::OptionsWithMedianTime |
Namespaces | |
tc | |
tc::autotune | |
Functions | |
std::vector< std::size_t > | tc::autotune::powers2andCeilDivisors (std::size_t val) |
template<typename Vector , typename... Vectors> | |
Vector | tc::autotune::mergeVectors (Vector &&v, Vectors &&...vs) |
std::vector < OptionsWithMedianTime > | tc::autotune::getOptionsAndMedianRuntimes (const std::string &id, const std::vector< const DLTensor * > &inputs) |
std::vector< MappingOptions > | tc::autotune::restoreCandidates (const std::string &id, const std::vector< const DLTensor * > &inputs, const std::vector< const DLTensor * > &outputs) |
llvm::Optional< MappingOptions > | tc::autotune::getBestOptions (const std::string &id, const std::vector< const DLTensor * > &inputs, const std::vector< const DLTensor * > &outputs) |