Home
last modified time | relevance | path

Searched refs:PjRtExecute (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.h63 StatusOr<std::vector<std::unique_ptr<PyBuffer>>> PjRtExecute(
Dpy_executable.cc73 StatusOr<std::vector<std::unique_ptr<PyBuffer>>> PyExecutable::PjRtExecute( in PjRtExecute() function in xla::PyExecutable
Djax_jit.cc1181 ValueOrThrow(cache_entry->executable->PjRtExecute(arguments.arg_buffers)); in Call()