Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_values.cc282 py::object pxla_module = py::module::import("jax.interpreters.pxla"); in DevicePut() local
443 py::object pxla_module = py::module::import("jax.interpreters.pxla"); in PyArgSignatureOfValue() local