Render¶
- 
systemds.onnx_systemds.render.gen_graph_functions(env: jinja2.environment.Environment, main_graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto) -> ([<class 'str'>], <class 'str'>, [<class 'str'>])¶
- Traverses the node tree of the onnx-graph structure and generates a script string for each node, as well as a string for the required imports together with all functions of sub-graphs. The resulting lists are correctly ordered for inserting them in the dml script. - Parameters
- env – Jinja environment to load the template files 
- main_graph – the onnx graph for which the script shall be generated 
 
- Returns
- Tuple (imports, main function, sub-graph functions) 
 
- 
systemds.onnx_systemds.render.gen_model_header(env: jinja2.environment.Environment, model: onnx.onnx_ONNX_REL_1_7_ml_pb2.ModelProto) → str¶
- Generates the header of the script for the given model - Parameters
- env – Jinja environment to load the template files 
- model – the onnx model for which the header shall be generated 
 
- Returns
- the generated header 
 
- 
systemds.onnx_systemds.render.gen_node_script(env: jinja2.environment.Environment, graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto, node: onnx.onnx_ONNX_REL_1_7_ml_pb2.NodeProto) → systemds.onnx_systemds.operator_gen.GeneratedScriptPart¶
- Generates a dml script snippet, the required imports and sub-graphs for the given node - Parameters
- env – Jinja environment to load the template files 
- graph – the onnx graph for which the script shall be generated 
- node – the node for which the dml snippet shall be generated 
 
- Returns
- The generated script-part 
 
- 
systemds.onnx_systemds.render.gen_script(model: onnx.onnx_ONNX_REL_1_7_ml_pb2.ModelProto, output_file: str = None) → str¶
- Generate the dml script for the given model and return it. If an output_file is given, the script is also written to a file. - Parameters
- model – the model for which the dml script shall be generated 
- output_file – (optional) the file to which the script shall be written 
 
- Returns
- the generated dml-script 
 
- 
systemds.onnx_systemds.render.render_function(env: jinja2.environment.Environment, graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto, generated_node_scripts: [<class 'str'>]) → str¶
- Generates the dml function for the given graph and inserts the ‘generated_node_scripts’ in the function-body. - Parameters
- env – Jinja environment to load the template files 
- graph – the graph for which the function shall be generated 
- generated_node_scripts – the node scripts in correct order for the function-body 
 
- Returns
- the generated function