(x, x2, y2)  # pylint: disable=pointless-statement
  return

#--- test fn for scf control flow ---


@composite.Composite('TestTwoInputsOp')
def _tfr_control_flow_if(x, y, pred):
  if pred:
    return x
  else:
    return y


@composite.Composite('TestThreeInputsOp')
def _tfr_control_flow_nested_if(x, y, z, select):
  if select == 'x':
    return x