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... | |