diff --git a/backends/cadence/aot/replace_ops.py b/backends/cadence/aot/replace_ops.py index 2104764cd14..24390da5e16 100644 --- a/backends/cadence/aot/replace_ops.py +++ b/backends/cadence/aot/replace_ops.py @@ -1590,7 +1590,7 @@ def call_operator(self, op, args, kwargs, meta): updated_args = list(args) for op_arg_index in args_to_be_replaced: arg = args[op_arg_index] - if not arg.is_tensor(): + if not isinstance(arg, ProxyValue) or not arg.is_tensor(): return super().call_operator(op, args, kwargs, meta) if not isinstance(arg.node.target, EdgeOpOverload):