megatron.schedules.get\_tensor\_shapes
======================================

.. currentmodule:: megatron.schedules

.. autofunction:: get_tensor_shapes