CaffeParser.cpp File Reference

#include "CaffeParser.hpp"
#include "RecordByRecordCaffeParser.hpp"
#include "armnn/Descriptors.hpp"
#include "armnn/INetwork.hpp"
#include "armnn/Utils.hpp"
#include "armnn/Exceptions.hpp"
#include "GraphTopologicalSort.hpp"
#include "VerificationHelpers.hpp"
#include <armnn/utility/Assert.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <boost/format.hpp>
#include "caffe/proto/caffe.pb.h"
#include <google/protobuf/io/coded_stream.h>
#include <google/protobuf/io/zero_copy_stream.h>
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/text_format.h>
#include <google/protobuf/stubs/common.h>
#include <google/protobuf/stubs/once.h>
#include <google/protobuf/descriptor.h>
#include <google/protobuf/generated_message_reflection.h>
#include <google/protobuf/reflection_ops.h>
#include <google/protobuf/wire_format.h>
#include <cmath>
#include <sstream>
#include <queue>
#include <fcntl.h>

Go to the source code of this file.

Namespaces
	armnnCaffeParser
	Caffe networks are loaded from protobuf files (binary or text) using the protobuf library and the generated code from caffe.pb.h.

Macros
#define	GET_OPTIONAL_WITH_VECTOR_FALLBACK(PARAM, PARAM_TYPE, OPTIONAL_VALUE, FALLBACK_VECTOR, VALUE_TYPE, DEFAULT_VALUE)
#define	GET_OPTIONAL_WITH_FALLBACK(PARAM, PARAM_TYPE, OPTIONAL_VALUE, FALLBACK_VALUE, VALUE_TYPE, DEFAULT_VALUE)

Functions
BlobShape	TensorDescToBlobShape (const TensorInfo &desc)

Macro Definition Documentation

GET_OPTIONAL_WITH_FALLBACK

#define GET_OPTIONAL_WITH_FALLBACK	(		PARAM,
			PARAM_TYPE,
			OPTIONAL_VALUE,
			FALLBACK_VALUE,
			VALUE_TYPE,
			DEFAULT_VALUE
	)

Value:

GetOptionalWithFallback( \
        PARAM, \
        [](const PARAM_TYPE & param, VALUE_TYPE defaultValue) \
        { \
            if (param.has_##OPTIONAL_VALUE ()) \
            { \
                return std::make_pair(true, param.OPTIONAL_VALUE ()); \
            } \
            else \
            { \
                return std::make_pair(false, defaultValue); \
            } \
        }, \
        [](const PARAM_TYPE & param, VALUE_TYPE defaultValue) \
        { \
            if (param.has_##FALLBACK_VALUE ()) \
            { \
                return std::make_pair(true, param.FALLBACK_VALUE ()); \
            } \
            else \
            { \
                return std::make_pair(false, defaultValue); \
            } \
        }, \
        DEFAULT_VALUE)

Definition at line 209 of file CaffeParser.cpp.

Referenced by CaffeParserBase::ParsePoolingLayer().

GET_OPTIONAL_WITH_VECTOR_FALLBACK

#define GET_OPTIONAL_WITH_VECTOR_FALLBACK	(		PARAM,
			PARAM_TYPE,
			OPTIONAL_VALUE,
			FALLBACK_VECTOR,
			VALUE_TYPE,
			DEFAULT_VALUE
	)

Value:

GetOptionalWithFallback( \
        PARAM, \
        [](const PARAM_TYPE & param, VALUE_TYPE defaultValue) \
        { \
            if (param.has_##OPTIONAL_VALUE ()) \
            { \
                return std::make_pair(true, param.OPTIONAL_VALUE ()); \
            } \
            else \
            { \
                return std::make_pair(false, defaultValue); \
            } \
        }, \
        [](const PARAM_TYPE & param, VALUE_TYPE defaultValue) \
        { \
            if (param.FALLBACK_VECTOR##_size() > 0) \
            { \
                return std::make_pair(true, (param.FALLBACK_VECTOR ()).Get(0)); \
            } \
            else \
            { \
                return std::make_pair(false, defaultValue); \
            } \
        }, \
        DEFAULT_VALUE)

Definition at line 177 of file CaffeParser.cpp.

Referenced by CaffeParserBase::ParseConvLayer().