from tensorflow.contrib.eager.python.examples.revnet import revnet

    with tfe.execution_mode(execution_mode):
      device, data_format = device_and_format
      model = revnet.RevNet(config=config)
      if defun:
