Halide
12.0.1
Halide compiler and libraries
|
Functions | |
std::vector< int > | get_dims (const at::Tensor tensor) |
template<class scalar_t > | |
void | check_type (at::Tensor &tensor) |
AT_FORALL_SCALAR_TYPES_WITH_COMPLEX (HL_PT_DEFINE_TYPECHECK) | |
template<class scalar_t > | |
Buffer< scalar_t > | wrap (at::Tensor &tensor) |
|
inline |
Definition at line 34 of file HalidePyTorchHelpers.h.
Referenced by wrap().
|
inline |
Definition at line 45 of file HalidePyTorchHelpers.h.
Halide::PyTorch::AT_FORALL_SCALAR_TYPES_WITH_COMPLEX | ( | HL_PT_DEFINE_TYPECHECK | ) |
|
inline |
Definition at line 88 of file HalidePyTorchHelpers.h.
References get_dims(), and halide_cuda_device_interface().