code: def input_types(self) -> Optional[Dict[str, NeuralType]]:
      return {
          "input_ids": NeuralType(("B", "T"), ChannelType()),
          "attention_mask": NeuralType(
              ("B", "T"), MaskType(), optional=True
          ),
          "token_type_ids": NeuralType(
              ("B", "T"), ChannelType(), optional=True
          ),
      }
  
file_path: NeMo/nemo/collections/nlp/modules/common/bert_module.py
imports: import os
  import re
  from typing import Dict, Optional
  import torch
  from nemo.core.classes import NeuralModule
  from nemo.core.classes.exportable import Exportable
  from nemo.core.neural_types import ChannelType, MaskType, NeuralType
  from nemo.utils import logging
label_desc: Use image node_id 1 for calling the BertModule obj's underlying member method code with example usage: obj.input_types() and returns: dict