|
|
@ -61,7 +61,7 @@ public: |
|
|
|
return backendId == DNN_BACKEND_OPENCV || |
|
|
|
return backendId == DNN_BACKEND_OPENCV || |
|
|
|
backendId == DNN_BACKEND_CUDA || |
|
|
|
backendId == DNN_BACKEND_CUDA || |
|
|
|
backendId == DNN_BACKEND_HALIDE || |
|
|
|
backendId == DNN_BACKEND_HALIDE || |
|
|
|
(backendId == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && axis == 1) || |
|
|
|
(backendId == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && axis == 1 && !blobs.empty()) || |
|
|
|
(backendId == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && axis > 0); |
|
|
|
(backendId == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && axis > 0); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|