device: "/job:localhost/replica:0/task:0/device:GPU:0"
  attr {
    key: "T"
    value {
      type: DT_FLOAT
    }
  }
}
node {
  name: "_retval_loss_1/add_0_0"
  op: "_Retval"
  input: "loss_1/add"
  device: "/job:localhost/replica:0/task:0/device:CPU:0"
  attr {
    key: "T"
    value {
      type: DT_FLOAT
    }
  }
  attr {
    key: "index"

    }
  }
  attr {
    key: "is_training"
    value {
      b: true
    }
  }
}
node {
  name: "metrics_2_sparse_categorical_accuracy_identity_RetVal"
  op: "_Retval"
  input: "metrics_2/sparse_categorical_accuracy/div_no_nan"
  device: "/job:localhost/replica:0/task:0/device:GPU:0"
  attr {
    key: "T"
    value {
      type: DT_FLOAT
    }
  }