OMTensorList.h

Go to the documentation of this file.

//===-------- OMTensorList.h - OMTensorList Declaration header-------------===//
//
// Copyright 2019-2020 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

OMTensorList *omTensorListCreate(OMTensor **tensors, int n);

void omTensorListDestroy(OMTensorList *list);

OMTensor **omTensorListGetOmtArray(OMTensorList *list);

int omTensorListGetSize(OMTensorList *list);

OMTensor *omTensorListGetOmtByIndex(OMTensorList *list, size_t index);

#ifdef __cplusplus
}
#endif

#endif // ONNX_MLIR_OMTENSORLIST_H