Struct CupyUtil

struct CupyUtil

Structure that encapsulates cupy utilities.

Public Static Functions

static pybind11::module_ get_cp()

TODO(Documentation)

static pybind11::object tensor_to_cupy(const TensorObject &tensor)

TODO(Documentation)

static TensorObject cupy_to_tensor(pybind11::object cupy_array)

TODO(Documentation)

Public Static Attributes

static pybind11::object cp_module = pybind11::none()

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.