Searched refs:ExecuteShardedOnLocalDevices (Results 1 – 6 of 6) sorted by relevance
74 ExecuteShardedOnLocalDevices(absl::Span<const std::vector<PyBuffer*>> args);
140 PyExecutable::ExecuteShardedOnLocalDevices( in ExecuteShardedOnLocalDevices() function in xla::PyExecutable
360 &PyExecutable::ExecuteShardedOnLocalDevices, py::arg("arguments")) in PYBIND11_MODULE()
201 &PyTpuExecutable::ExecuteShardedOnLocalDevices, in PYBIND11_MODULE()
349 ExecuteShardedOnLocalDevices(
755 PyTpuExecutable::ExecuteShardedOnLocalDevices( in ExecuteShardedOnLocalDevices() function in xla::PyTpuExecutable