gva_tensor_meta.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright (C) 2018-2020 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 ******************************************************************************/
 
#ifndef __GVA_TENSOR_META_H__
#define __GVA_TENSOR_META_H__
 
#include <gst/gst.h>
 
#define GVA_TENSOR_META_API_NAME "GstGVATensorMetaAPI"
#define GVA_TENSOR_META_IMPL_NAME "GstGVATensorMeta"
#define GVA_TENSOR_META_TAG "gva_tensor_meta"
 
#define GVA_TENSOR_MAX_RANK 8
 
G_BEGIN_DECLS
 
typedef enum {
    GVA_PRECISION_UNSPECIFIED = 255, 
    GVA_PRECISION_FP32 = 10,         
    GVA_PRECISION_U8 = 40            
} GVAPrecision;
 
typedef enum {
    GVA_LAYOUT_ANY = 0,  
    GVA_LAYOUT_NCHW = 1, 
    GVA_LAYOUT_NHWC = 2, 
    GVA_LAYOUT_NC = 193  
} GVALayout;
 
inline const void *gva_get_tensor_data(GstStructure *s, gsize *nbytes) {
    const GValue *f = gst_structure_get_value(s, "data_buffer");
    if (!f)
        return NULL;
    GVariant *v = g_value_get_variant(f);
    return g_variant_get_fixed_array(v, nbytes, 1);
}
 
typedef struct _GstGVATensorMeta GstGVATensorMeta;
 
struct _GstGVATensorMeta {
    GstMeta meta;       
    GstStructure *data; 
};
 
const GstMetaInfo *gst_gva_tensor_meta_get_info(void);
 
GType gst_gva_tensor_meta_api_get_type(void);
 
#define GST_GVA_TENSOR_META_INFO (gst_gva_tensor_meta_get_info())
 
#define GST_GVA_TENSOR_META_GET(buf) ((GstGVATensorMeta *)gst_buffer_get_meta(buf, gst_gva_tensor_meta_api_get_type()))
 
#define GST_GVA_TENSOR_META_ITERATE(buf, state)                                                                        \
    ((GstGVATensorMeta *)gst_buffer_iterate_meta_filtered(buf, state, gst_gva_tensor_meta_api_get_type()))
 
#define GST_GVA_TENSOR_META_ADD(buf)                                                                                   \
    ((GstGVATensorMeta *)gst_buffer_add_meta(buf, gst_gva_tensor_meta_get_info(), NULL))
 
#define GST_GVA_TENSOR_META_COUNT(buf) (gst_buffer_get_n_meta(buf, gst_gva_tensor_meta_api_get_type()))
 
GstGVATensorMeta *find_tensor_meta(GstBuffer *buffer, const char *model_name, const char *output_layer);
 
GstGVATensorMeta *find_tensor_meta_ext(GstBuffer *buffer, const char *model_name, const char *output_layer,
                                       const char *element_id);
 
G_END_DECLS
 
#endif /* __GVA_TENSOR_META_H__ */