Searched defs:xla_module (Results 1 – 1 of 1) sorted by relevance
387 const auto xla_module = py::module::import("jax.interpreters.xla"); in ArgSignatureOfValue() local751 const auto xla_module = py::module::import("jax.interpreters.xla"); in DevicePut() local952 static const auto* xla_module = in ConvertArgsToBuffers() local1195 static const auto* xla_module = in Call() local