hub / github.com/apple/axlearn / test_extend_step

Route test_extend_step

axlearn/common/decoder_test.py:None–None · view source on GitHub ↗

parameterized.parameters(
        dict(  # Baseline
            use_cross_attention=False,
            stack_cfg=StackedTransformerLayer.default_config(),
            custom_attention_mask_cfg=None,

Source from the content-addressed store, hash-verified

1	# Copyright © 2023 Apple Inc.
2
3	"""Tests decoder layers."""
4

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected