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