MCPcopy Create free account
hub / github.com/apple/axlearn / ParamInitSpec

Class ParamInitSpec

axlearn/common/test_utils.py:371–374  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

369# Use dataclass so that jax.tree.map does not expand it.
370@dataclasses.dataclass
371class ParamInitSpec:
372 shape: Optional[Sequence[int]]
373 initializer: Initializer
374 fan_axes: Optional[FanAxes]
375
376
377# initialize is intentionally left unimplemented.

Callers 4

param_init_debug_stringFunction · 0.90
check_initFunction · 0.90
test_delegatesMethod · 0.90
patched_initFunction · 0.85

Calls

no outgoing calls

Tested by 3

param_init_debug_stringFunction · 0.72
test_delegatesMethod · 0.72
patched_initFunction · 0.68