![]() |
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) |