Skip to content
 
ml-switcheroo ml_switcheroo.plugins.topk
Type to start searching
    ml-switcheroo
    • API Reference
    • ml_switcheroo
    • ml_switcheroo.plugins
    ml-switcheroo
    • Documentation
    • ml-switcheroo 🔄🦘
    • Architecture
    • Extending
    • Maintenance
    • Ideas
    • Reference
    • API Reference
      • injector_fw
      • ml_switcheroo
      • injector_spec
      • injector_plugin
    • ml_switcheroo.plugins.topk
      • Functions
      • Module Contents
        • transform_topk()
    • Show Source

    ml_switcheroo.plugins.topk¶

    Plugin for TopK Output Adaptation.

    Functions¶

    transform_topk(→ libcst.CSTNode)

    Module Contents¶

    ml_switcheroo.plugins.topk.transform_topk(node: libcst.Call, ctx: ml_switcheroo.core.hooks.HookContext) → libcst.CSTNode¶
    "Previous" ml_switcheroo.plugins.tf_data_loader
    "Next" ml_switcheroo.semantics
    © Copyright 2025, Samuel Marks.
    Created using Sphinx 8.2.3. and Material for Sphinx