ml_switcheroo.plugins.tf_data_loader

Plugin for Native TensorFlow Data Pipeline Generation.

Functions

transform_tf_dataloader(→ Union[libcst.Call, ...)

Module Contents

ml_switcheroo.plugins.tf_data_loader.transform_tf_dataloader(node: libcst.Call, ctx: ml_switcheroo.core.hooks.HookContext) → libcst.Call | libcst.FlattenSentinel