code: def resigter_custom_layers(
      custom_objects: Dict[str, Any]
  ) -> Dict[str, Any]:
      custom_objects[
          "KConditionalRandomField"
      ] = KConditionalRandomField
      custom_objects["BahdanauAttention"] = BahdanauAttention
      return custom_objects
  
imports: from typing import Dict, Any
  from tensorflow import keras
  from .conditional_random_field import KConditionalRandomField
  from .behdanau_attention import BahdanauAttention
label_desc: Use image node_id 1 for calling a global function with example usage: resigter_custom_layers(custom_objects) and returns: custom_objects