![]() |
Tensor Comprehensions
|
#include <chrono>#include <string>#include <unordered_set>#include <vector>#include "tc/core/tc2halide.h"#include "tc/core/utils/cuda_info.h"#include "tc/core/utils/dlpack.h"

Go to the source code of this file.
Namespaces | |
| tc | |
Functions | |
| std::map< std::string, int > | tc::computeParamValueMap (const tc2halide::HalideComponents &components, const std::vector< const DLTensor * > &inputsDLT) |
| std::vector < dlutils::DLTensorUPtr > | tc::inferOutputTensorInfo (const tc2halide::HalideComponents &halide, const std::vector< const DLTensor * > &inputsDLT) |
| std::string | tc::halideCodegenC (const Halide::Internal::Stmt &s) |
| Just generates a C function body from a Halide stmt. Exposed for testing. More... | |