if \'constant\' is in the name of the graph input then have it define…

…d as a relay constant
tenstorrent · Jul 27, 2024 · 731ac2a · 731ac2a
1 parent c748d64
commit 731ac2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -7203,7 +7203,7 @@ def _parse_graph_initializers(self, graph):
             if self._freeze_params:
                 self._nodes[init_tensor.name] = _expr.const(array)
             else:
-                if (
+                if "constant" in init_tensor.name.lower() or (
                     "weight" not in init_tensor.name
                     and "bias" not in init_tensor.name
                     and ("int" in array.dtype or "bool" in array.dtype)