Skip to content

Commit f884d78

Browse files
committed
use tf.shape instead of .shape for dynamic axes
1 parent 0110e0e commit f884d78

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnx_tf/handlers/backend/instance_normalization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def _common(cls, node, **kwargs):
3131
beta = tensor_dict[node.inputs[2]]
3232

3333
inputs = tensor_dict[node.inputs[0]]
34-
inputs_shape = inputs.shape
34+
inputs_shape = tf.shape(inputs)
3535
inputs_rank = inputs.shape.ndims
3636

3737
moments_axes = list(range(inputs_rank))[2:]

0 commit comments

Comments
 (0)