ONNX
模块:polygraphy.backend.onnx
- 加载器
GsFromOnnx
gs_from_onnx()
OnnxFromPath
onnx_from_path()
OnnxFromTfGraph
onnx_from_tf_graph()
ModifyOutputs
modify_outputs()
ConvertToFp16
convert_to_fp16()
FoldConstants
fold_constants()
SetUpperBound
set_upper_bound()
InferShapes
infer_shapes()
ExtractSubgraph
extract_subgraph()
SaveOnnx
save_onnx()
BytesFromOnnx
bytes_from_onnx()
OnnxFromBytes
onnx_from_bytes()