code: def _get_linear_decay_lr(spec):
      return learning_rate_schedules.linear_decay_learning_rate(
          spec.step,
          spec.total_train_steps,
          spec.initial_lr,
          spec.offset,
      )
  
file_path: mesh/mesh_tensorflow/transformer/learning_rate_schedules_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import dataclasses
  from mesh_tensorflow.transformer import learning_rate_schedules
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling a global function with example usage: _get_linear_decay_lr(spec) and returns: learning_rate_schedules