TIM-VX/include/tim/utils/nbg_parser/nbg_parser_impl.h

92 lines
3.1 KiB
C

/****************************************************************************
*
* Copyright (c) 2020-2023 Vivante Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*
*****************************************************************************/
#ifndef _NBG_PARSER_IMPL_H
#define _NBG_PARSER_IMPL_H
#if defined(__cplusplus)
extern "C"{
#endif
#include "gc_vip_nbg_format.h"
/* NBG format version */
#define NBG_FORMAT_VERSION 0x00010014
enum nbg_nn_command_size_e
{
NBG_NN_COMMAND_SIZE_128 = 0,
NBG_NN_COMMAND_SIZE_192 = 1,
};
typedef struct _nbg_reader
{
vip_uint32_t offset;
vip_uint32_t total_size;
vip_uint8_t *data;
vip_uint8_t *current_data;
} nbg_reader_t;
typedef struct _nbg_parser_data
{
/* Fixed part of the bin. */
gcvip_bin_fixed_t fixed;
/* Dynamic data part of the bin. */
gcvip_bin_inout_entry_t *inputs;
gcvip_bin_inout_entry_t *outputs;
gcvip_bin_layer_t *orig_layers; /* original layers info, loading from binary graph */
gcvip_bin_operation_t *operations;
gcvip_bin_entry_t *LCDT;
gcvip_bin_sh_operation_t *sh_ops;
void *nn_ops;
gcvip_bin_tp_operation_t *tp_ops;
gcvip_bin_patch_data_entry_t *pd_entries;
gcvip_bin_hw_init_operation_info_entry_t *hw_init_ops;
gcvip_bin_entry_t *ICDT;
void *LCD;
vip_uint32_t n_inputs;
vip_uint32_t n_outputs;
vip_uint32_t n_orig_layers; /* the number of original layers */
vip_uint32_t n_operations;
vip_uint32_t n_LCDT;
vip_uint32_t n_nn_ops;
vip_uint32_t n_tp_ops;
vip_uint32_t n_sh_ops;
vip_uint32_t n_pd_entries;
vip_uint32_t n_hw_init_ops;
vip_uint32_t n_ICDT;
nbg_reader_t reader;
} nbg_parser_data_t;
#if defined(__cplusplus)
}
#endif
#endif