OMTensorList.h

Go to the documentation of this file.

/*
 * SPDX-License-Identifier: Apache-2.0
 */
 
//===-------- OMTensorList.h - OMTensorList Declaration header-------------===//
//
// Copyright 2019-2023 The IBM Research Authors.
//
// =============================================================================
//
// This file contains declaration of OMTensorList data structures and
// API functions.
//
//===----------------------------------------------------------------------===//
 
#ifndef ONNX_MLIR_OMTENSORLIST_H
#define ONNX_MLIR_OMTENSORLIST_H
 
#include "onnx-mlir/Runtime/OMTensor.h"
 
struct OMTensorList;
 
#ifndef __cplusplus
typedef struct OMTensorList OMTensorList;
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
OM_EXTERNAL_VISIBILITY OMTensorList *omTensorListCreate(
    OMTensor **tensors, int64_t n);
 
OM_EXTERNAL_VISIBILITY void omTensorListDestroy(OMTensorList *list);
 
OM_EXTERNAL_VISIBILITY void omTensorListDestroyShallow(OMTensorList *list);
 
OM_EXTERNAL_VISIBILITY OMTensor **omTensorListGetOmtArray(const OMTensorList *list);
 
OM_EXTERNAL_VISIBILITY int64_t omTensorListGetSize(const OMTensorList *list);
 
OM_EXTERNAL_VISIBILITY OMTensor *omTensorListGetOmtByIndex(
    const OMTensorList *list, int64_t index);
 
#ifdef __cplusplus
}
#endif
 
#endif // ONNX_MLIR_OMTENSORLIST_H