code: def _lsh_self_attention_fn(self):
      return functools.partial(
          tl.LSHSelfAttention,
          attention_dropout=0.0,
          chunk_len=64,
          n_buckets=[32, 32],
          n_chunks_after=0,
          n_chunks_before=1,
          n_hashes=1,
          n_parallel_heads=1,
          predict_drop_len=128,
          predict_mem_len=1024,
      )
  
file_path: trax/trax/models/research/terraformer_oom_test.py
imports: import functools
  import operator
  from absl.testing import absltest
  import gin
  import numpy
  from trax import fastmath
  from trax import layers
  from trax import shapes
  from trax.models.research import terraformer
label_desc: Use image node_id 2 for calling the TerraformerOOMTest obj's underlying member method code with example usage: obj._lsh_self_attention_fn() and returns: functools