File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -1180,7 +1180,7 @@ def _cmd_sbs(argv: List[Any]):
11801180 import pandas
11811181 import torch
11821182 from .helpers import string_type
1183- from .torch_onnx .sbs import run_aligned , post_process_run_aligned_obs
1183+ from .torch_onnx .sbs import run_aligned
11841184 from .reference import OnnxruntimeEvaluator
11851185
11861186 parser = get_parser_sbs ()
@@ -1241,9 +1241,12 @@ def _size(name):
12411241 use_tensor = True ,
12421242 exc = False ,
12431243 ):
1244- pobs = post_process_run_aligned_obs (obs )
1245- data .append (pobs )
1246- if "initializer" not in pobs and "placeholder" not in pobs and len (data ) % ratio == 0 :
1244+ data .append (obs )
1245+ if (
1246+ obs .onnx_op_type != "initializer"
1247+ and onnx .ep_target != "placeholder"
1248+ and len (data ) % ratio == 0
1249+ ):
12471250 df = pandas .DataFrame (data ).apply (
12481251 lambda col : col .fillna ("" ) if col .dtype == "object" else col
12491252 )
You can’t perform that action at this time.
0 commit comments