Searched defs:AxisEnv (Results 1 – 1 of 1) sorted by relevance
492 class AxisEnv(NamedTuple): class498 def extend_axis_env(env: AxisEnv, name, size: int):507 def axis_groups(axis_env: AxisEnv, name):