code: def mlp_in():
      return [
          Layer(
              weight="feed_forward.dense.dense.weight",
              bias="feed_forward.dense.dense.bias",
          ),
      ]
  
file_path: parallelformers/parallelformers/policies/reformer.py
imports: from transformers.models.reformer.modeling_reformer import ReformerLayer
  from parallelformers.policies.base import Layer, Policy
  from parallelformers.utils.dist_utils import AllReduceLinear
label_desc: Use image node_id 4 for calling the ReformerPolicy obj's underlying member method code with example usage: obj.mlp_in() and returns: list