http://uml.mvnsearch.org/gist/e416cc43a092ec53109be6a02777508b

tiny-dnn-v2-device.puml

@startuml
enum DeviceType {
 CPU
 GPU
}

class Layer {
}

class Context {
 - bool is_opencl
 - CLCudaAPI::Context device_context
}

class Program {
 - Device device
 - Layer op
 - CLCuda::Program op_program
}

class Device {
 - Context context
 + Device(DeviceType device_type)
 + Device(DeviceType device_type, int platform_id, int device_id)
}

class OpenCLProgramManager {
 + registerOp(Device device, Layer op)
}

OpenCLProgramManager *-- Program
@enduml