Home
last modified time | relevance | path

Searched refs:flatten_shape (Results 1 – 1 of 1) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py599 def flatten_shape(s: XlaShape) -> Sequence[Tuple[Sequence[int], XlaShape]]: function
740 for param_index, element in flatten_shape(c.GetShape(arg)):
752 for output_index, element in flatten_shape(c.GetShape(out_tuple)):