Method createDeepPolyElement

src/nlr/DeepPolyAnalysis.cpp:200–239

}

DeepPolyElement *DeepPolyAnalysis::createDeepPolyElement( Layer *layer )
{
    Layer::Type type = layer->getLayerType();
    DeepPolyElement *deepPolyElement;
    if ( type == Layer::INPUT )
        deepPolyElement = new DeepPolyInputElement( layer );
    else if ( type == Layer::WEIGHTED_SUM )
    {
        deepPolyElement = new DeepPolyWeightedSumElement( layer );
        // Weighted sum layers need working memory for back substitution
        deepPolyElement->setWorkingMemory( _work1SymbolicLb,
                                           _work1SymbolicUb,
                                           _work2SymbolicLb,
                                           _work2SymbolicUb,
                                           _workSymbolicLowerBias,
                                           _workSymbolicUpperBias );
    }
    else if ( type == Layer::RELU )
        deepPolyElement = new DeepPolyReLUElement( layer );
    else if ( type == Layer::ROUND )
        deepPolyElement = new DeepPolyRoundElement( layer );
    else if ( type == Layer::LEAKY_RELU )
        deepPolyElement = new DeepPolyLeakyReLUElement( layer );
    else if ( type == Layer::SIGN )
        deepPolyElement = new DeepPolySignElement( layer );
    else if ( type == Layer::ABSOLUTE_VALUE )
        deepPolyElement = new DeepPolyAbsoluteValueElement( layer );
    else if ( type == Layer::MAX )
        deepPolyElement = new DeepPolyMaxPoolElement( layer );
    else if ( type == Layer::SIGMOID )
        deepPolyElement = new DeepPolySigmoidElement( layer );
    else if ( type == Layer::SOFTMAX )
        deepPolyElement = new DeepPolySoftmaxElement( layer, _maxLayerSize );
    else if ( type == Layer::BILINEAR )
        deepPolyElement = new DeepPolyBilinearElement( layer );
    else
        throw NLRError( NLRError::LAYER_TYPE_NOT_SUPPORTED,
                        Stringf( "Layer %u not yet supported", layer->getLayerType() ).ascii() );
    return deepPolyElement;
}

void DeepPolyAnalysis::log( const String &message )
{