Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_values.cc279 py::object pxla_module = py::module::import("jax.interpreters.pxla"); in DevicePut() local
440 py::object pxla_module = py::module::import("jax.interpreters.pxla"); in PyArgSignatureOfValue() local
Dpmap_lib.cc288 py::object pxla_module = py::module::import("jax").attr("config"); in UpdateArgsSignature() local