auto &factory = OperationFactory::instance();
OperationFactory::Param param{inputCount, inputs, outputCount, outputs};
- switch (type)
- {
- case ANEURALNETWORKS_DEPTHWISE_CONV_2D:
- case ANEURALNETWORKS_MAX_POOL_2D:
- case ANEURALNETWORKS_AVERAGE_POOL_2D:
- case ANEURALNETWORKS_CONCATENATION:
- case ANEURALNETWORKS_RESHAPE:
- case ANEURALNETWORKS_FULLY_CONNECTED:
- case ANEURALNETWORKS_SOFTMAX:
- case ANEURALNETWORKS_ADD:
- case ANEURALNETWORKS_SUB:
- case ANEURALNETWORKS_CONV_2D:
- case ANEURALNETWORKS_MUL:
- case ANEURALNETWORKS_TANH:
- case ANEURALNETWORKS_STRIDED_SLICE:
- case ANEURALNETWORKS_LOGISTIC:
- {
- auto node = factory.create(type, param);
- _model->addOperation(std::unique_ptr<neurun::model::operation::Node>{node});
-
- break;
- }
-
- default:
- throw std::runtime_error{"Not supported operation"};
- };
+ auto node = factory.create(type, param);
+ _model->addOperation(std::unique_ptr<neurun::model::operation::Node>{node});
}
catch (const std::exception &e)
{