HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Ort::ShapeInferContext Struct Reference

Provide access to per-node attributes and input shapes, so one could compute and set output shapes. More...

#include <onnxruntime_cxx_api.h>

Classes

struct  SymbolicInteger
 

Public Types

using Shape = std::vector< SymbolicInteger >
 
using Ints = std::vector< int64_t >
 
using Floats = std::vector< float >
 
using Strings = std::vector< std::string >
 

Public Member Functions

 ShapeInferContext (const OrtApi *ort_api, OrtShapeInferContext *ctx)
 
const ShapeGetInputShape (size_t indice) const
 
size_t GetInputCount () const
 
Status SetOutputShape (size_t indice, const Shape &shape)
 
int64_t GetAttrInt (const char *attr_name)
 
Ints GetAttrInts (const char *attr_name)
 
float GetAttrFloat (const char *attr_name)
 
Floats GetAttrFloats (const char *attr_name)
 
std::string GetAttrString (const char *attr_name)
 
Strings GetAttrStrings (const char *attr_name)
 

Detailed Description

Provide access to per-node attributes and input shapes, so one could compute and set output shapes.

Definition at line 2166 of file onnxruntime_cxx_api.h.

Member Typedef Documentation

using Ort::ShapeInferContext::Floats = std::vector<float>

Definition at line 2218 of file onnxruntime_cxx_api.h.

using Ort::ShapeInferContext::Ints = std::vector<int64_t>

Definition at line 2213 of file onnxruntime_cxx_api.h.

Definition at line 2201 of file onnxruntime_cxx_api.h.

Definition at line 2223 of file onnxruntime_cxx_api.h.

Constructor & Destructor Documentation

Ort::ShapeInferContext::ShapeInferContext ( const OrtApi *  ort_api,
OrtShapeInferContext *  ctx 
)

Member Function Documentation

float Ort::ShapeInferContext::GetAttrFloat ( const char *  attr_name)
inline

Definition at line 2005 of file onnxruntime_cxx_inline.h.

ShapeInferContext::Floats Ort::ShapeInferContext::GetAttrFloats ( const char *  attr_name)
inline

Definition at line 2013 of file onnxruntime_cxx_inline.h.

int64_t Ort::ShapeInferContext::GetAttrInt ( const char *  attr_name)
inline

Definition at line 1981 of file onnxruntime_cxx_inline.h.

ShapeInferContext::Ints Ort::ShapeInferContext::GetAttrInts ( const char *  attr_name)
inline

Definition at line 1989 of file onnxruntime_cxx_inline.h.

std::string Ort::ShapeInferContext::GetAttrString ( const char *  attr_name)
inline

Definition at line 2029 of file onnxruntime_cxx_inline.h.

ShapeInferContext::Strings Ort::ShapeInferContext::GetAttrStrings ( const char *  attr_name)
inline

Definition at line 2044 of file onnxruntime_cxx_inline.h.

size_t Ort::ShapeInferContext::GetInputCount ( ) const
inline

Definition at line 2207 of file onnxruntime_cxx_api.h.

const Shape& Ort::ShapeInferContext::GetInputShape ( size_t  indice) const
inline

Definition at line 2205 of file onnxruntime_cxx_api.h.

Status Ort::ShapeInferContext::SetOutputShape ( size_t  indice,
const Shape shape 
)
inline

Definition at line 1949 of file onnxruntime_cxx_inline.h.


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