Ort::Custom::TensorArray Struct Reference

#include <onnxruntime_lite_custom_op.h>

Inheritance diagram for Ort::Custom::TensorArray:

Public Member Functions
	TensorArray (OrtKernelContext *ctx, size_t start_indice, bool is_input)
template<typename T >
T *	AllocateOutput (size_t ith_output, const std::vector< int64_t > &shape)
Tensor< std::string > &	AllocateStringTensor (size_t ith_output)
size_t	Size () const
const TensorPtr &	operator[] (size_t ith_input) const
Public Member Functions inherited from Ort::Custom::ArgBase
	ArgBase (OrtKernelContext *ctx, size_t indice, bool is_input)
virtual	~ArgBase ()

Additional Inherited Members
Protected Attributes inherited from Ort::Custom::ArgBase
struct KernelContext	ctx_
size_t	indice_
bool	is_input_

Detailed Description

Definition at line 313 of file onnxruntime_lite_custom_op.h.

Constructor & Destructor Documentation

Ort::Custom::TensorArray::TensorArray ( OrtKernelContext *  ctx, size_t  start_indice, bool  is_input  )

inline

Definition at line 314 of file onnxruntime_lite_custom_op.h.

Member Function Documentation

template<typename T >

T* Ort::Custom::TensorArray::AllocateOutput ( size_t  ith_output, const std::vector< int64_t > &  shape  )

inline

Definition at line 372 of file onnxruntime_lite_custom_op.h.

Tensor<std::string>& Ort::Custom::TensorArray::AllocateStringTensor ( size_t  ith_output )

inline

Definition at line 380 of file onnxruntime_lite_custom_op.h.

const TensorPtr& Ort::Custom::TensorArray::operator[] ( size_t  ith_input ) const

inline

Definition at line 391 of file onnxruntime_lite_custom_op.h.

size_t Ort::Custom::TensorArray::Size ( ) const

inline

Definition at line 388 of file onnxruntime_lite_custom_op.h.

---

The documentation for this struct was generated from the following file:

• onnxruntime/core/session/onnxruntime_lite_custom_op.h