Home
last modified time | relevance | path

Searched refs:host_buffer_semantics (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.cc184 PjRtClient::HostBufferSemantics host_buffer_semantics = in HandleNumpyArray() local
193 host_buffer_semantics = PjRtClient::HostBufferSemantics::kZeroCopy; in HandleNumpyArray()
201 data, squashed_type, dims, byte_strides, host_buffer_semantics, in HandleNumpyArray()
H A Dpy_client.h153 PjRtClient::HostBufferSemantics host_buffer_semantics);
H A Dpy_client.cc216 PjRtClient::HostBufferSemantics host_buffer_semantics) { in BufferFromPyval() argument
258 (host_buffer_semantics == PjRtClient::HostBufferSemantics::kZeroCopy)); in BufferFromPyval()
H A Dxla_client_test.py2395 x, host_buffer_semantics=xla_client.HostBufferSemantics.ZERO_COPY)
2406 x, host_buffer_semantics=during_call)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc693 HostBufferSemantics host_buffer_semantics, in BufferFromHostBuffer() argument
734 host_buffer_semantics == HostBufferSemantics::kZeroCopy && in BufferFromHostBuffer()
738 (host_buffer_semantics == in BufferFromHostBuffer()
789 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall || in BufferFromHostBuffer()
816 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall) { in BufferFromHostBuffer()
841 host_buffer_semantics, transpose{std::move(transpose)}]() { in BufferFromHostBuffer()
857 if (host_buffer_semantics != in BufferFromHostBuffer()
H A Dpjrt_c_api_client.h198 HostBufferSemantics host_buffer_semantics, in BufferFromHostBuffer() argument
203 data, type, dims, byte_strides, host_buffer_semantics, in BufferFromHostBuffer()
H A Dtfrt_cpu_pjrt_client.cc494 HostBufferSemantics host_buffer_semantics, in BufferFromHostBuffer() argument
508 host_buffer_semantics == HostBufferSemantics::kZeroCopy && in BufferFromHostBuffer()
548 host_buffer_semantics == in BufferFromHostBuffer()
H A Dtfrt_cpu_pjrt_client.h186 HostBufferSemantics host_buffer_semantics,
H A Dpjrt_client.h687 HostBufferSemantics host_buffer_semantics,
H A Dpjrt_stream_executor_client.h236 HostBufferSemantics host_buffer_semantics,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A D__init__.pyi376 host_buffer_semantics: HostBufferSemantics = ...) -> Buffer: ...