Skip to main content

M_getDeviceTypeFromSpec

Function M_getDeviceTypeFromSpec 

Source
pub unsafe extern "C" fn M_getDeviceTypeFromSpec(
    spec: *const M_TensorSpec,
) -> M_DeviceType
Expand description

Gets the device type from a tensor specification.

@param spec The tensor spec.

@return The device type (CPU or GPU).