this will happen in one of the last steps before emitting the flatbuffer model
    in order to enable optimizations that rely on a fixed order of operations (such
    as rematerialization.)
    The flatbuffer exporter will unwrap the wrapped region and annotate the generated
    model with metadata such that any runtime reorderings will respect the order
    given by the control dependencies.
  }];