We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3068bf5 commit 2794413Copy full SHA for 2794413
Deeploy/DeeployTypes.py
@@ -3429,8 +3429,9 @@ def _removeIdentityNodes(self):
3429
self.graph.deleteNode(node)
3430
3431
def _assertTensorsHaveShape(self) -> None:
3432
- for tensor in self.graph.tensors().values():
3433
- assert tensor.shape is not None, f"Tensors are expected to have shape. Shape inference not supported."
+ missingShapes = [name for name, tensor in self.graph.tensors().items() if tensor.shape is None]
+ assert len(missingShapes) == 0, \
3434
+ f"Shape inference is not supported.\nFound tensors with missing shape annotation: {missingShapes}"
3435
3436
def frontEnd(self):
3437
"""API hook to prepare the graph to be deployed and build the initial NetworkContext
0 commit comments