Home
last modified time | relevance | path

Searched refs:ExecuteOnLocalDevices (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.h68 ExecuteOnLocalDevices(absl::Span<const std::vector<PyBuffer*>> args);
Dpy_executable.cc112 PyExecutable::ExecuteOnLocalDevices( in ExecuteOnLocalDevices() function in xla::PyExecutable
Dxla.cc357 .def("execute_on_local_devices", &PyExecutable::ExecuteOnLocalDevices, in PYBIND11_MODULE()
Dpmap_lib.cc327 ValueOrThrow(cache_entry->executable->ExecuteOnLocalDevices(arg_buffers)); in Call()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc193 .def("ExecuteOnLocalDevices", &PyTpuExecutable::ExecuteOnLocalDevices, in PYBIND11_MODULE()
198 .def("execute_on_local_devices", &PyTpuExecutable::ExecuteOnLocalDevices, in PYBIND11_MODULE()
Dtpu_client.h345 ExecuteOnLocalDevices(
Dtpu_client.cc667 PyTpuExecutable::ExecuteOnLocalDevices( in ExecuteOnLocalDevices() function in xla::PyTpuExecutable
780 TF_ASSIGN_OR_RETURN(output_buffers, ExecuteOnLocalDevices(arg_buffers)); in ExecuteShardedOnLocalDevices()