Merge pull request #558 from robertknight/ort-infer-bool

Support generating inputs for bool tensors in ort-infer.py
robertknight · Jan 29, 2025 · cfb3acc · cfb3acc
2 parents 0f5f37e + 7fa3182
commit cfb3acc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/ort-infer.py b/tools/ort-infer.py
@@ -90,6 +90,7 @@ def run_model(
             "tensor(float16)": np.float16,
             "tensor(int64)": np.int64,
             "tensor(int32)": np.int32,
+            "tensor(bool)": np.bool_,
         }
 
         resolved_shape = []