Function consume_common_graph_parameters

utils/CommonGraphOptions.cpp:187–220

}

CommonGraphParams consume_common_graph_parameters(CommonGraphOptions &options)
{
    FastMathHint fast_math_hint_value =
        options.fast_math_hint->value() ? FastMathHint::Enabled : FastMathHint::Disabled;
    auto validation_range = parse_validation_range(options.validation_range->value());

    CommonGraphParams common_params;
    common_params.help      = options.help->is_set() ? options.help->value() : false;
    common_params.threads   = options.threads->value();
    common_params.batches   = options.batches->value();
    common_params.target    = options.target->value();
    common_params.data_type = options.data_type->value();
    if (options.data_layout->is_set())
    {
        common_params.data_layout = options.data_layout->value();
    }
    common_params.enable_tuner    = options.enable_tuner->is_set() ? options.enable_tuner->value() : false;
    common_params.enable_cl_cache = common_params.target == arm_compute::graph::Target::NEON
                                        ? false
                                        : (options.enable_cl_cache->is_set() ? options.enable_cl_cache->value() : true);
    common_params.tuner_mode      = options.tuner_mode->value();
    common_params.fast_math_hint  = options.fast_math_hint->is_set() ? fast_math_hint_value : FastMathHint::Disabled;
    common_params.data_path       = options.data_path->value();
    common_params.image           = options.image->value();
    common_params.labels          = options.labels->value();
    common_params.validation_file = options.validation_file->value();
    common_params.validation_path = options.validation_path->value();
    common_params.validation_range_start = validation_range.first;
    common_params.validation_range_end   = validation_range.second;
    common_params.tuner_file             = options.tuner_file->value();
    common_params.mlgo_file              = options.mlgo_file->value();

    return common_params;
}
} // namespace utils
} // namespace arm_compute