Searched defs:XlaShape (Results 1 – 2 of 2) sorted by relevance
379 XlaShape = Any # xla_client.Shape variable
57 XlaShape = Any # xla_client.Shape variable599 def flatten_shape(s: XlaShape) -> Sequence[Tuple[Sequence[int], XlaShape]]: