Function ExtractPrecisionParameters

software/xtract/caffe_network_parser.cpp:3228–3311

Extract FP Precision Parameters

// Extract FP Precision Parameters
bool ExtractPrecisionParameters(const caffe::LayerParameter& src, XLayer& layer, bool isParamsPrecisionMandatory) {
    bool parsedSuccessfully = true;

    // Check if it is a single precision layer
    if(SWLayerSet.find(layer.type) != SWLayerSet.end()) {
	layer.quantization_scheme = "SinglePrecision";
	return true;
    }

    // precision_param() or quantization_param() ?
    string field_type = "";

    if(src.has_precision_param()) {
        // src_params = &(src.precision_param());
	layer.quantization_scheme = src.precision_param().quant_type();
	field_type = "precision"; 
    }
    else if(src.has_quantization_param()) {
	// src_params = &(src.quantization_param());
	layer.quantization_scheme = src.quantization_param().quant_type();
	field_type = "quantization"; 
    }
    else {
	parsedSuccessfully = false;
    }

    layer.quantization_scheme = layer.quantization_scheme.empty() ? "DynamicFixed" : layer.quantization_scheme; 

    // quant_type should be either "DynamicFixed" or "Xilinx" (if empty, by default it is Dynamic Fixed)
    ASSERT((layer.quantization_scheme == "DynamicFixed" || layer.quantization_scheme == "Xilinx"), EP501,
    		"Quantization Scheme should be either \"DynamicFixed\" or \"Xilinx\". "
    		"Quantization Scheme of layer " << layer.name << " is " << "\"" << layer.quantization_scheme << "\""
			<< " which is not supported" )

    // INFO : Special case : Don't run precision extraction for Maxpool DynamicFixed format
    // TODO : Maxpool has prec param only in Offline mode, not in DynamicFixed mode. Better keep same pattern everywhere.
    if(layer.quantization_scheme == "DynamicFixed") {
	if ((layer.type == "Pooling" && layer.pool_params->PoolType == MAX) ||
		(layer.type == "Concat") || (layer.type == "Softmax"))
	{
	    return true;
	}
    }

    // Parse based on the quantization_scheme
    if(layer.quantization_scheme == "DynamicFixed") {
	if(field_type == "precision") {
	    parsedSuccessfully = ExtractDynamicFixedPointParameters(src.precision_param(), layer, isParamsPrecisionMandatory, parsedSuccessfully);
	}
	else { // quantization
	    parsedSuccessfully = ExtractDynamicFixedPointParameters(src.quantization_param(), layer, isParamsPrecisionMandatory, parsedSuccessfully);
	}
    }
    else if(layer.quantization_scheme == "Xilinx") {
	if(field_type == "precision") {
	    parsedSuccessfully = ExtractOfflineQuantizationParameters(src.precision_param(), layer, isParamsPrecisionMandatory, parsedSuccessfully);
	}
	else { // quantization