Home
last modified time | relevance | path

Searched defs:device_buffer (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtransfer_manager_test.cc74 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
90 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
112 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
130 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
148 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
166 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
182 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
201 auto device_buffer = AllocateDeviceBuffer(ondevice_shape); in XLA_TEST_F() local
222 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local
236 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.h79 se::Stream* stream, const ShapedBuffer& device_buffer) { in TransferLiteralFromDevice()
87 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
109 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
133 const ShapedBuffer& device_buffer) { in TransferLiteralToDevice()
159 const ShapedBuffer& device_buffer) { in TransferLiteralToDeviceAsync()
275 se::StreamExecutor* executor, const ShapedBuffer& device_buffer) const { in CanShapedBufferBeAccessedNow()
282 const se::DeviceMemoryBase& device_buffer) const { in CanBufferBeAccessedNow()
H A Dtransfer_manager.cc51 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
77 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
98 const ShapedBuffer& device_buffer, in TransferLiteralToDevice()
198 ShapedBuffer* device_buffer, in ReadDynamicShapes()
282 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTables()
288 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTablesAsync()
317 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteRootTupleIndexTable()
H A Dgeneric_transfer_manager.cc62 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
98 const ShapedBuffer& device_buffer, in TransferLiteralToDeviceAsync()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc331 void RecordUsage(PjRtStreamExecutorBuffer::ScopedHold device_buffer, in RecordUsage()
491 PjRtStreamExecutorBuffer::ScopedHold device_buffer, in AddDestinationBufferSynchronization()
597 ScopedHold device_buffer(this, ScopedHold::kUsage); in logical_on_device_shape() local
766 auto device_buffer = std::make_shared<TrackedDeviceBuffer>( in BufferFromHostBuffer() local
782 PjRtStreamExecutorBuffer::ScopedHold device_buffer( in BufferFromHostBuffer() local
842 PjRtStreamExecutorBuffer::ScopedHold device_buffer( in BufferFromHostBuffer() local
958 PjRtStreamExecutorBuffer::ScopedHold device_buffer( in BufferFromHostLiteral() local
972 PjRtStreamExecutorBuffer::ScopedHold device_buffer(movable_device_buffer); in BufferFromHostLiteral() local
1079 auto device_buffer = std::make_shared<TrackedDeviceBuffer>( in CreateViewOfDeviceBuffer() local
1115 Shape on_device_shape, std::shared_ptr<TrackedDeviceBuffer> device_buffer, in PjRtStreamExecutorBuffer()
[all …]
H A Dtfrt_cpu_pjrt_client.cc516 auto device_buffer = std::make_shared<MaybeOwningCpuMemory>( in BufferFromHostBuffer() local
612 auto* device_buffer = output_buffer->AcquireUsage(std::move(usage_event)); in BufferFromHostLiteral() local
739 std::unique_ptr<TrackedTfrtCpuDeviceBuffer> device_buffer) { in AbortDonation()
748 auto device_buffer = ReleaseBufferLocked(); in Delete() local
788 std::unique_ptr<TrackedTfrtCpuDeviceBuffer> device_buffer = in Release() local
866 auto* device_buffer = AcquireUsage(usage_event); in logical_on_device_shape() local
925 auto* device_buffer = AcquireUsage(usage_event); in ToLiteral() local
H A Dtfrt_cpu_pjrt_client.h366 std::unique_ptr<TrackedTfrtCpuDeviceBuffer> device_buffer) in DonationTransaction()
390 TrackedTfrtCpuDeviceBuffer* device_buffer() const { in device_buffer() function
H A Dtracked_device_buffer_test.cc105 std::shared_ptr<TrackedDeviceBuffer> device_buffer = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_transfer_manager.h47 const ShapedBuffer& device_buffer) const override { in CanShapedBufferBeAccessedNow()
53 const se::DeviceMemoryBase& device_buffer) const override { in CanBufferBeAccessedNow()
H A Dcpu_transfer_manager.cc56 ShapedBuffer* device_buffer, in ReadDynamicShapes()
H A Dcpu_xfeed.cc277 ShapedBuffer* device_buffer, Shape* device_shape, in ReadDynamicShapesOnCpu()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_transfer_manager.cc71 const xla::ShapedBuffer& device_buffer, in TransferLiteralToDeviceAsync()
193 stream_executor::Stream* stream, const xla::ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
329 xla::ShapedBuffer* device_buffer, in ReadDynamicShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc277 Shape on_host_shape, std::shared_ptr<TpuSharedBuffer> device_buffer, in PyTpuBuffer()
441 std::shared_ptr<TpuSharedBuffer> device_buffer = DeviceBuffer(); in BlockHostUntilReady() local
503 auto device_buffer = std::make_shared<TpuSharedBuffer>( in CreateBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_transfer_manager.cc86 ShapedBuffer* device_buffer, in ReadDynamicShapes()
/aosp_15_r20/hardware/libhardware/modules/camera/3_4/
H A Dv4l2_wrapper.cpp609 v4l2_buffer device_buffer; in EnqueueRequest() local
/aosp_15_r20/external/deqp-deps/amber/src/dawn/
Dengine_dawn.cc369 auto& device_buffer = compute_pipeline.buffers[i]; in MapDeviceBufferToHostBuffer() local