Function GetFlatBufferDataLayout

src/armnnSerializer/SerializerUtils.cpp:94–108

}

armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout)
{
    switch (dataLayout)
    {
        case armnn::DataLayout::NHWC:
            return armnnSerializer::DataLayout::DataLayout_NHWC;
        case armnn::DataLayout::NDHWC:
            return armnnSerializer::DataLayout::DataLayout_NDHWC;
        case armnn::DataLayout::NCDHW:
            return armnnSerializer::DataLayout::DataLayout_NCDHW;
        case armnn::DataLayout::NCHW:
        default:
            return armnnSerializer::DataLayout::DataLayout_NCHW;
    }
}

armnnSerializer::BinaryOperation GetFlatBufferBinaryOperation(armnn::BinaryOperation binaryOperation)
{