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 Shape &	GetInputShape (size_t indice) const
size_t	GetInputCount () const
Status	SetOutputShape (size_t indice, const Shape &shape, ONNXTensorElementDataType type=ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT)
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 2224 of file onnxruntime_cxx_api.h.

Member Typedef Documentation

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

Definition at line 2276 of file onnxruntime_cxx_api.h.

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

Definition at line 2271 of file onnxruntime_cxx_api.h.

using Ort::ShapeInferContext::Shape = std::vector<SymbolicInteger>

Definition at line 2259 of file onnxruntime_cxx_api.h.

using Ort::ShapeInferContext::Strings = std::vector<std::string>

Definition at line 2281 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 2088 of file onnxruntime_cxx_inline.h.

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

inline

Definition at line 2096 of file onnxruntime_cxx_inline.h.

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

inline

Definition at line 2058 of file onnxruntime_cxx_inline.h.

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

inline

Definition at line 2066 of file onnxruntime_cxx_inline.h.

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

inline

Definition at line 2118 of file onnxruntime_cxx_inline.h.

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

inline

Definition at line 2133 of file onnxruntime_cxx_inline.h.

size_t Ort::ShapeInferContext::GetInputCount ( ) const

inline

Definition at line 2265 of file onnxruntime_cxx_api.h.

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

inline

Definition at line 2263 of file onnxruntime_cxx_api.h.

Status Ort::ShapeInferContext::SetOutputShape ( size_t  indice, const Shape &  shape, ONNXTensorElementDataType  type = ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT  )

inline

Definition at line 2025 of file onnxruntime_cxx_inline.h.

---

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

• onnxruntime/core/session/onnxruntime_cxx_api.h
• onnxruntime/core/session/onnxruntime_cxx_inline.h