@@ -19,44 +19,48 @@ void ReadDaqImpl(const uint8_t *buf, ModelBuilder &builder);
1919
2020std::string layer_type_to_str (DNN::LayerType type) {
2121 switch (type) {
22+ // DaqReader auto generated layer_type_to_str start
23+ case DNN::LayerType::Conv2D:
24+ return " Conv2D" ;
25+ case DNN::LayerType::AvePool:
26+ return " AvePool" ;
27+ case DNN::LayerType::MaxPool:
28+ return " MaxPool" ;
29+ case DNN::LayerType::Relu:
30+ return " Relu" ;
31+ case DNN::LayerType::Softmax:
32+ return " Softmax" ;
2233 case DNN::LayerType::FC:
23- return " fc " ;
34+ return " FC " ;
2435 case DNN::LayerType::Add:
2536 return " Add" ;
26- case DNN::LayerType::Relu:
27- return " relu" ;
28- case DNN::LayerType::Conv2D:
29- return " conv" ;
3037 case DNN::LayerType::Concat:
31- return " concat" ;
32- case DNN::LayerType::MaxPool:
33- return " maxpool" ;
34- case DNN::LayerType::AvePool:
35- return " avepool" ;
36- case DNN::LayerType::Softmax:
37- return " softmax" ;
38+ return " Concat" ;
3839 case DNN::LayerType::DepthwiseConv2D:
39- return " depthwsie " ;
40+ return " DepthwiseConv2D " ;
4041 case DNN::LayerType::BatchToSpace:
41- return " batch2space " ;
42+ return " BatchToSpace " ;
4243 case DNN::LayerType::SpaceToBatch:
43- return " space2batch " ;
44+ return " SpaceToBatch " ;
4445 case DNN::LayerType::StridedSlice:
45- return " stridedslice " ;
46+ return " StridedSlice " ;
4647 case DNN::LayerType::Mul:
47- return " mul" ;
48- case DNN::LayerType::MulScalar:
49- return " mulscalar" ;
48+ return " Mul" ;
5049 case DNN::LayerType::AddScalar:
51- return " addscalar" ;
50+ return " AddScalar" ;
51+ case DNN::LayerType::MulScalar:
52+ return " MulScalar" ;
5253 case DNN::LayerType::Dequantize:
53- return " dequantize " ;
54+ return " Dequantize " ;
5455 case DNN::LayerType::LRN:
5556 return " LRN" ;
5657 case DNN::LayerType::Tanh:
57- return " tanh " ;
58+ return " Tanh " ;
5859 case DNN::LayerType::Floor:
59- return " floor" ;
60+ return " Floor" ;
61+ case DNN::LayerType::Logistic:
62+ return " Logistic" ;
63+ // DaqReader auto generated layer_type_to_str end
6064 }
6165}
6266
@@ -275,29 +279,19 @@ void AddLayers(const DNN::Model &model, ModelBuilder &builder) {
275279 break ;
276280 }
277281 case DNN::LayerType::BatchToSpace: {
278- #if __ANDROID_API__ >= __ANDROID_API_P__
279282 ADD_LAYER (batch_to_space, BatchToSpaceND, input, block_sizes,
280283 output);
281284 break ;
282- #endif
283285 }
284286 case DNN::LayerType::SpaceToBatch: {
285- #if __ANDROID_API__ >= __ANDROID_API_P__
286287 ADD_LAYER (space_to_batch, SpaceToBatchND, input, block_sizes,
287288 pads, output);
288289 break ;
289- #endif
290290 }
291291 case DNN::LayerType::StridedSlice: {
292- #if __ANDROID_API__ >= __ANDROID_API_P__
293292 ADD_LAYER (strided_slice, StridedSlice, input, starts, ends,
294293 strides, begin_mask, end_mask, shrink_axis_mask,
295294 output);
296- #else
297- throw std::invalid_argument (" Unsupported layer " +
298- layer_type_to_str (layer->type ()) +
299- " in API 28" );
300- #endif
301295 break ;
302296 }
303297 case DNN::LayerType::LRN: {
@@ -312,6 +306,10 @@ void AddLayers(const DNN::Model &model, ModelBuilder &builder) {
312306 ADD_LAYER (floor, Floor, input, output);
313307 break ;
314308 }
309+ case DNN::LayerType::Logistic: {
310+ ADD_LAYER (logistic, Logistic, input, output);
311+ break ;
312+ }
315313 }
316314 }
317315}
0 commit comments