experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typo: BufferFromPyal -> BufferFromPyval.

Browse Source 
PiperOrigin-RevId: 321346825
Change-Id: Ica5e622513464f42c2e44d65bd63efe9a97d585e
This commit is contained in:
Tom Hennigan 2020-07-15 06:12:51 -07:00 committed by TensorFlower Gardener
parent bc9a9a35bc
commit b7cbbb454d
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/python/py_client.cc
View File

@ -83,7 +83,7 @@ PyClient::GetDefaultDeviceAssignment1D(int num_replicas) {
return result;
}
StatusOr<std::unique_ptr<PyBuffer>> PyClient::BufferFromPyal(
StatusOr<std::unique_ptr<PyBuffer>> PyClient::BufferFromPyval(
const pybind11::object& argument, Device* device, bool force_copy,
PjRtBuffer::HostBufferSemantics host_buffer_semantics) {
if (device == nullptr) {

2
tensorflow/compiler/xla/python/py_client.h
View File

@ -120,7 +120,7 @@ class PyClient : public std::enable_shared_from_this<PyClient> {
return pjrt_client_->client()->CreateHostToDeviceChannelHandle();
}
StatusOr<std::unique_ptr<PyBuffer>> BufferFromPyal(
StatusOr<std::unique_ptr<PyBuffer>> BufferFromPyval(
const pybind11::object& argument, Device* device, bool force_copy,
PjRtBuffer::HostBufferSemantics host_buffer_semantics);

2
tensorflow/compiler/xla/python/xla.cc
View File

@ -539,7 +539,7 @@ PYBIND11_MODULE(xla_extension, m) {
&PyClient::CreateDeviceToHostChannelHandle)
.def("create_host_to_device_channel_handle",
&PyClient::CreateHostToDeviceChannelHandle)
.def("buffer_from_pyval", &PyClient::BufferFromPyal, py::arg("argument"),
.def("buffer_from_pyval", &PyClient::BufferFromPyval, py::arg("argument"),
py::arg("device") = nullptr, py::arg("force_copy") = false,
py::arg("host_buffer_semantics") =
PjRtBuffer::HostBufferSemantics::kZeroCopy)