HDK
|
#include <onnxruntime_lite_custom_op.h>
Public Types | |
using | strings = std::vector< std::string > |
Public Member Functions | |
Tensor (OrtKernelContext *ctx, size_t indice, bool is_input) | |
const strings & | Data () const |
const void * | DataRaw () const override |
size_t | SizeInBytes () const override |
void | SetStringOutput (const strings &ss, const std::vector< int64_t > &dims) |
const Span< std::string > & | AsSpan () |
const std::string & | AsScalar () |
![]() | |
TensorBase (OrtKernelContext *ctx, size_t indice, bool is_input) | |
operator bool () const | |
const std::vector< int64_t > & | Shape () const |
ONNXTensorElementDataType | Type () const |
int64_t | NumberOfElement () const |
std::string | Shape2Str () const |
bool | IsCpuTensor () const |
![]() | |
ArgBase (OrtKernelContext *ctx, size_t indice, bool is_input) | |
virtual | ~ArgBase () |
Additional Inherited Members | |
![]() | |
std::optional< std::vector < int64_t > > | shape_ |
ONNXTensorElementDataType | type_ = ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED |
const char * | mem_type_ = "Cpu" |
![]() | |
struct KernelContext | ctx_ |
size_t | indice_ |
bool | is_input_ |
Definition at line 169 of file onnxruntime_lite_custom_op.h.
using Ort::Custom::Tensor< std::string >::strings = std::vector<std::string> |
Definition at line 171 of file onnxruntime_lite_custom_op.h.
|
inline |
Definition at line 173 of file onnxruntime_lite_custom_op.h.
|
inline |
Definition at line 230 of file onnxruntime_lite_custom_op.h.
|
inline |
Definition at line 227 of file onnxruntime_lite_custom_op.h.
|
inline |
Definition at line 202 of file onnxruntime_lite_custom_op.h.
|
inlineoverridevirtual |
Implements Ort::Custom::TensorBase.
Definition at line 205 of file onnxruntime_lite_custom_op.h.
|
inline |
Definition at line 217 of file onnxruntime_lite_custom_op.h.
|
inlineoverridevirtual |
Implements Ort::Custom::TensorBase.
Definition at line 211 of file onnxruntime_lite_custom_op.h.