HDK
|
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) |
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) |
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.
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.
using Ort::ShapeInferContext::Shape = std::vector<SymbolicInteger> |
Definition at line 2201 of file onnxruntime_cxx_api.h.
using Ort::ShapeInferContext::Strings = std::vector<std::string> |
Definition at line 2223 of file onnxruntime_cxx_api.h.
Ort::ShapeInferContext::ShapeInferContext | ( | const OrtApi * | ort_api, |
OrtShapeInferContext * | ctx | ||
) |
|
inline |
Definition at line 2005 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 2013 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1981 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 1989 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 2029 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 2044 of file onnxruntime_cxx_inline.h.
|
inline |
Definition at line 2207 of file onnxruntime_cxx_api.h.
|
inline |
Definition at line 2205 of file onnxruntime_cxx_api.h.
Definition at line 1949 of file onnxruntime_cxx_inline.h.