| 607 | } |
| 608 | |
| 609 | void MILPFormulator::addLayerToModel( GurobiWrapper &gurobi, |
| 610 | const Layer *layer, |
| 611 | LayerOwner *layerOwner ) |
| 612 | { |
| 613 | switch ( layer->getLayerType() ) |
| 614 | { |
| 615 | case Layer::INPUT: |
| 616 | case Layer::WEIGHTED_SUM: |
| 617 | break; |
| 618 | |
| 619 | case Layer::RELU: |
| 620 | addReluLayerToMILPFormulation( gurobi, layer, layerOwner ); |
| 621 | break; |
| 622 | |
| 623 | default: |
| 624 | throw NLRError( NLRError::LAYER_TYPE_NOT_SUPPORTED, "MILPFormulator" ); |
| 625 | break; |
| 626 | } |
| 627 | } |
| 628 | |
| 629 | void MILPFormulator::addNeuronToModel( GurobiWrapper &gurobi, |
| 630 | const Layer *layer, |
no test coverage detected