|
Halide 20.0.0
Halide compiler and libraries
|
Classes | |
| struct | UserContext |
Typedefs | |
| typedef struct Halide::PyTorch::UserContext | UserContext |
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) |
| template<class scalar_t > | |
| Buffer< scalar_t > | wrap_cuda (at::Tensor &tensor) |
| typedef struct Halide::PyTorch::UserContext Halide::PyTorch::UserContext |
|
inline |
Definition at line 30 of file HalidePyTorchHelpers.h.
Referenced by wrap(), and wrap_cuda().
|
inline |
Definition at line 41 of file HalidePyTorchHelpers.h.
| Halide::PyTorch::AT_FORALL_SCALAR_TYPES_WITH_COMPLEX | ( | HL_PT_DEFINE_TYPECHECK | ) |
|
inline |
Definition at line 84 of file HalidePyTorchHelpers.h.
References get_dims().
|
inline |