Home
last modified time | relevance | path

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

/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.cc1483 absl::Span<const PjRtStreamExecutorBuffer::ScopedHold> device_buffers, in MakeTupleHelper()
1638 absl::Span<const PjRtStreamExecutorBuffer::ScopedHold> device_buffers, in MakeExecutionInputsAndWaitForEvents()
1685 std::vector<PjRtStreamExecutorBuffer::ScopedHold>* device_buffers, in EnqueueExecution()
1887 std::vector<PjRtStreamExecutorBuffer::ScopedHold> device_buffers; in ExecuteHelper() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.h163 pybind11::list device_buffers) in ShardedDeviceArray()