ONNX Runtime
|
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) |
Provide access to per-node attributes and input shapes, so one could compute and set output shapes.
using Ort::ShapeInferContext::Floats = std::vector<float> |
using Ort::ShapeInferContext::Ints = std::vector<int64_t> |
using Ort::ShapeInferContext::Shape = std::vector<SymbolicInteger> |
using Ort::ShapeInferContext::Strings = std::vector<std::string> |
Ort::ShapeInferContext::ShapeInferContext | ( | const OrtApi * | ort_api, |
OrtShapeInferContext * | ctx | ||
) |
float Ort::ShapeInferContext::GetAttrFloat | ( | const char * | attr_name | ) |
Floats Ort::ShapeInferContext::GetAttrFloats | ( | const char * | attr_name | ) |
int64_t Ort::ShapeInferContext::GetAttrInt | ( | const char * | attr_name | ) |
Ints Ort::ShapeInferContext::GetAttrInts | ( | const char * | attr_name | ) |
std::string Ort::ShapeInferContext::GetAttrString | ( | const char * | attr_name | ) |
Strings Ort::ShapeInferContext::GetAttrStrings | ( | const char * | attr_name | ) |
|
inline |
|
inline |
Status Ort::ShapeInferContext::SetOutputShape | ( | size_t | indice, |
const Shape & | shape, | ||
ONNXTensorElementDataType | type = ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT |
||
) |