Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/experimental/
H A Dhost_callback.py380 XlaComputationBuilder = Any # xla_bridge._JaxComputationBuilder variable
784 comp: XlaComputationBuilder, *args_op: XlaOp, **params): argument
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py58 XlaComputationBuilder = Any # xla_bridge._JaxComputationBuilder variable