Skip to content

Instantly share code, notes, and snippets.

@edgarriba

edgarriba/tiny-dnn-v2-device.puml

Last active August 4, 2016 11:16
Show Gist options
  • Save edgarriba/e416cc43a092ec53109be6a02777508b to your computer and use it in GitHub Desktop.
Save edgarriba/e416cc43a092ec53109be6a02777508b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment