Home
last modified time | relevance | path

Searched defs:xla_module (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.cc268 py::object xla_module = py::module::import("jax.interpreters.xla"); in DevicePut() local
429 py::object xla_module = py::module::import("jax.interpreters.xla"); in PyArgSignatureOfValue() local
H A Djax_jit.cc641 py::module xla_module(py::module::import("jax.interpreters.xla")); in GetJitArgumentStickyDevice() local