| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 81 se::Stream* stream, const ShapedBuffer& device_buffer) { in TransferLiteralFromDevice()
89 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
111 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
135 const ShapedBuffer& device_buffer) { in TransferLiteralToDevice()
161 const ShapedBuffer& device_buffer) { in TransferLiteralToDeviceAsync()
269 se::StreamExecutor* executor, const ShapedBuffer& device_buffer) const { in CanShapedBufferBeAccessedNow()
276 const se::DeviceMemoryBase& device_buffer) const { in CanBufferBeAccessedNow()
|
| D | transfer_manager.cc | 51 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
78 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
100 const ShapedBuffer& device_buffer, in TransferLiteralToDevice()
203 ShapedBuffer* device_buffer, in ReadDynamicShapes()
288 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTables()
294 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTablesAsync()
323 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteRootTupleIndexTable()
|
| D | generic_transfer_manager.cc | 62 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
94 const ShapedBuffer& device_buffer, in TransferLiteralToDeviceAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 74 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 …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.h | 49 const ShapedBuffer& device_buffer) const override { in CanShapedBufferBeAccessedNow()
55 const se::DeviceMemoryBase& device_buffer) const override { in CanBufferBeAccessedNow()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 328 void RecordUsage(PjRtStreamExecutorBuffer::ScopedHold device_buffer, in RecordUsage()
481 PjRtStreamExecutorBuffer::ScopedHold device_buffer, in AddDestinationBufferSynchronization()
710 auto device_buffer = std::make_shared<TrackedDeviceBuffer>( in BufferFromHostBuffer() local
726 PjRtStreamExecutorBuffer::ScopedHold device_buffer( in BufferFromHostBuffer() local
771 PjRtStreamExecutorBuffer::ScopedHold device_buffer(movable_device_buffer); in BufferFromHostBuffer() local
879 PjRtStreamExecutorBuffer::ScopedHold device_buffer( in BufferFromHostLiteral() local
893 PjRtStreamExecutorBuffer::ScopedHold device_buffer(movable_device_buffer); in BufferFromHostLiteral() local
963 auto device_buffer = std::make_shared<TrackedDeviceBuffer>( in CreateViewOfDeviceBuffer() local
998 Shape on_device_shape, std::shared_ptr<TrackedDeviceBuffer> device_buffer, in PjRtStreamExecutorBuffer()
1042 std::shared_ptr<TrackedDeviceBuffer> device_buffer; in Release() local
[all …]
|
| D | tracked_device_buffer_test.cc | 105 std::shared_ptr<TrackedDeviceBuffer> device_buffer = in TEST() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.cc | 69 const xla::ShapedBuffer& device_buffer, in TransferLiteralToDeviceAsync()
191 stream_executor::Stream* stream, const xla::ShapedBuffer& device_buffer, in TransferLiteralFromDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 266 Shape on_host_shape, std::shared_ptr<TpuSharedBuffer> device_buffer, in PyTpuBuffer()
430 std::shared_ptr<TpuSharedBuffer> device_buffer = DeviceBuffer(); in BlockHostUntilReady() local
492 auto device_buffer = std::make_shared<TpuSharedBuffer>( in CreateBuffer() local
|
| /external/deqp-deps/amber/src/dawn/ |
| D | engine_dawn.cc | 369 auto& device_buffer = compute_pipeline.buffers[i]; in MapDeviceBufferToHostBuffer() local
|