test

microsoft · Oct 26, 2023 · 06d0ff3 · 06d0ff3
1 parent 379903a
commit 06d0ff3
Show file tree

Hide file tree

Showing 2 changed files with 331 additions and 329 deletions.
diff --git a/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py b/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py
@@ -220,6 +220,8 @@ def _create_graph_for_step(self, graph: onnx.GraphProto, onnx_opset: int):
         def build_output_declare():
             output_base = []
             for out in self.output_names:
+                print("xxxxxxxxxxxxxxxxxx")
+                print(out)
                 output_base.append(f"int64[{output_shape_str}] {out}")
 
             return ",".join(output_base)