switch

Auto-generated from flax_nnx_code_defs

Keras

API: keras.ops.switch
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: keras.ops.switch
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.transforms.transforms.switch
Strategy: Direct Mapping
Official Docs ↗