Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
halide_utils.h File Reference
#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"
Include dependency graph for halide_utils.h:
This graph shows which files directly or indirectly include this file:

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