| 7 | namespace caffe { namespace db { |
| 8 | |
| 9 | DB* GetDB(DataParameter::DB backend) { |
| 10 | switch (backend) { |
| 11 | #ifdef USE_LEVELDB |
| 12 | case DataParameter_DB_LEVELDB: |
| 13 | return new LevelDB(); |
| 14 | #endif // USE_LEVELDB |
| 15 | #ifdef USE_LMDB |
| 16 | case DataParameter_DB_LMDB: |
| 17 | return new LMDB(); |
| 18 | #endif // USE_LMDB |
| 19 | default: |
| 20 | LOG(FATAL) << "Unknown database backend"; |
| 21 | return NULL; |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | DB* GetDB(const string& backend) { |
| 26 | #ifdef USE_LEVELDB |
no outgoing calls