Home
last modified time | relevance | path

Searched defs:device_buffers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.h222 pybind11::list device_buffers, pybind11::handle indices) in ShardedDeviceArray()
231 absl::optional<pybind11::list> device_buffers() const { in device_buffers() function
234 void set_device_buffers(absl::optional<pybind11::list> device_buffers) { in set_device_buffers()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc32 std::vector<stream_executor::DeviceMemoryBase> device_buffers; in MakeArray() local
Dpjrt_stream_executor_client.cc1632 absl::Span<const PjRtStreamExecutorBuffer::ScopedHold> device_buffers, in MakeTupleHelper()
1800 absl::Span<const PjRtStreamExecutorBuffer::ScopedHold> device_buffers, in MakeExecutionInputsAndWaitForEvents()
1857 std::vector<PjRtStreamExecutorBuffer::ScopedHold>* device_buffers, in EnqueueExecution()
2078 std::vector<PjRtStreamExecutorBuffer::ScopedHold> device_buffers; in ExecuteHelper() local
Dtfrt_cpu_pjrt_client.cc1380 absl::InlinedVector<TfrtCpuBuffer::ScopedHold, 4> device_buffers; in ExecuteHelper() local