Package | Description |
---|---|
org.bytedeco.onnx | |
org.bytedeco.onnx.global |
Modifier and Type | Class and Description |
---|---|
class |
OpSchemaRegistry
\brief A registry to hold all the operator schemas.
|
Modifier and Type | Method and Description |
---|---|
ISchemaRegistry |
CheckerContext.get_schema_registry() |
Modifier and Type | Method and Description |
---|---|
void |
CheckerContext.set_schema_registry(ISchemaRegistry schema_registry) |
Modifier and Type | Method and Description |
---|---|
static void |
onnx.InferShapeForFunctionNode(FunctionProto func,
ISchemaRegistry schema_registry,
InferenceContext ctx) |
static void |
onnx.InferShapeForFunctionNode(FunctionProto func,
ISchemaRegistry schema_registry,
InferenceContext ctx,
ShapeInferenceOptions options,
SizeTStringMap model_local_functions_map,
SymbolTable symbolTable,
StringTensorShapeProtoMap generated_shape_data_by_name)
ModelLocalFunctionsMap is a map of function id -> model local function proto
All the ONNX helper utilities expect the function id ==
|
static void |
onnx.InferShapeForFunctionNode(FunctionProto func_proto,
StringIntMap func_opset_imports,
ISchemaRegistry schema_registry,
InferenceContext ctx) |
static void |
onnx.InferShapeForFunctionNode(FunctionProto func_proto,
StringIntMap func_opset_imports,
ISchemaRegistry schema_registry,
InferenceContext ctx,
ShapeInferenceOptions options,
SizeTStringMap model_local_functions_map,
SymbolTable symbolTable,
StringTensorShapeProtoMap generated_shape_data_by_name)
ModelLocalFunctionsMap is a map of function id -> model local function proto
All the ONNX helper utilities expect the function id ==
|
static void |
onnx.InferShapes(BytePointer model_path,
BytePointer save_path,
ISchemaRegistry schema_registry,
ShapeInferenceOptions options,
StringTensorShapeProtoMap generated_shape_data_by_name) |
static void |
onnx.InferShapes(GraphProto g,
StringIntMap opset_imports,
ISchemaRegistry schema_registry,
ShapeInferenceOptions options,
SizeTStringMap in_model_functions)
ModelLocalFunctionsMap is a map of function id -> model local function proto
All the ONNX helper utilities expect the function id ==
|
static void |
onnx.InferShapes(ModelProto m,
ISchemaRegistry schema_registry,
ShapeInferenceOptions options,
StringTensorShapeProtoMap generated_shape_data_by_name) |
static void |
onnx.InferShapes(String model_path,
String save_path,
ISchemaRegistry schema_registry,
ShapeInferenceOptions options,
StringTensorShapeProtoMap generated_shape_data_by_name) |
Copyright © 2024. All rights reserved.