Searched refs:XlaOp (Results 1 – 2 of 2) sorted by relevance
56 XlaOp = Any # xla_extension.XlaOp variable335 assert isinstance(ans, xe.XlaOp)466 assert isinstance(ans, xe.XlaOp)
378 XlaOp = Any # xla_extension.XlaOp variable784 comp: XlaComputationBuilder, *args_op: XlaOp, **params): argument