class nbla::DlpackArrayRegistry

class DlpackArrayRegistry

DlpackArrayCreator class this is never be instantiated.

This creator class is used only for DlpackArray.

Public Static Functions

static Context create_context(const DLTensor &dlp)

Interface to create a context for DlpackArray.

static DLDeviceType array_to_device_type(const string &array_class)

Interface to convert an array class to DLDeviceType.

static void add_map(const DLDeviceType device_type, const string &backend, const string &array_class)

Register the map from DLDeviceType to Array class and backend name.

static void add_map(const string &array_class, const DLDeviceType device_tyep)

Register the map from Array class to DLDeviceType.