code: def single_arg_test_fn(x):
      return ops.concatenate([(x + 1) ** 2, x], axis=-1)
  
file_path: keras/keras/backend/tests/compute_output_spec_test.py
imports: import unittest
  import pytest
  from keras import backend
  from keras import ops
  from keras.backend.common.keras_tensor import KerasTensor
label_desc: Use image node_id 1 for calling a global function with example usage: single_arg_test_fn(x) and returns: ops