| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 74 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
78 device_buffer)); in XLA_TEST_F()
81 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
90 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
94 device_buffer)); in XLA_TEST_F()
97 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
108 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
112 device_buffer)); in XLA_TEST_F()
115 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
124 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 59 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() argument
67 done(TransferLiteralFromDeviceInternal(stream->parent(), device_buffer, in TransferLiteralFromDevice()
72 se::StreamExecutor* executor, const ShapedBuffer& device_buffer, in TransferLiteralFromDeviceInternal() argument
75 << executor->device_ordinal() << "; device buffer: " << device_buffer; in TransferLiteralFromDeviceInternal()
76 TF_RET_CHECK(executor->device_ordinal() == device_buffer.device_ordinal()); in TransferLiteralFromDeviceInternal()
80 TF_RET_CHECK(ShapeUtil::Equal(device_buffer.on_device_shape(), in TransferLiteralFromDeviceInternal()
81 device_buffer.on_host_shape())); in TransferLiteralFromDeviceInternal()
84 device_buffer.on_host_shape(), in TransferLiteralFromDeviceInternal()
88 /*source=*/device_buffer.buffer(index), in TransferLiteralFromDeviceInternal()
101 const ShapedBuffer& device_buffer, in TransferLiteralToDeviceAsync() argument
[all …]
|
| D | transfer_manager.h | 72 se::Stream* stream, const ShapedBuffer& device_buffer,
75 se::Stream* stream, const ShapedBuffer& device_buffer) { in TransferLiteralFromDevice() argument
76 return TransferLiteralFromDevice(stream, device_buffer, nullptr); in TransferLiteralFromDevice()
79 se::Stream* stream, const ShapedBuffer& device_buffer,
83 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() argument
85 return TransferLiteralFromDevice(stream, device_buffer, literal, nullptr); in TransferLiteralFromDevice()
101 se::Stream* stream, const ShapedBuffer& device_buffer,
105 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() argument
108 return TransferLiteralFromDevice(stream, device_buffer, literal, done, in TransferLiteralFromDevice()
125 const ShapedBuffer& device_buffer,
[all …]
|
| D | transfer_manager.cc | 48 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() argument
59 Literal literal(device_buffer.on_host_shape()); in TransferLiteralFromDevice()
61 substream, device_buffer, literal, in TransferLiteralFromDevice()
75 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() argument
85 substream, device_buffer, literal, in TransferLiteralFromDevice()
97 const ShapedBuffer& device_buffer, in TransferLiteralToDevice() argument
107 substream, literal, device_buffer, transfer_metadata)); in TransferLiteralToDevice()
234 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTables() argument
235 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in WriteTupleIndexTables()
240 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTablesAsync() argument
[all …]
|
| D | generic_transfer_manager.h | 44 se::Stream* stream, const ShapedBuffer& device_buffer,
50 const ShapedBuffer& device_buffer,
70 const ShapedBuffer& device_buffer,
|
| /external/webrtc/webrtc/modules/audio_device/ |
| D | fine_audio_buffer.cc | 23 FineAudioBuffer::FineAudioBuffer(AudioDeviceBuffer* device_buffer, in FineAudioBuffer() argument
26 : device_buffer_(device_buffer), in FineAudioBuffer()
|
| D | fine_audio_buffer.h | 38 FineAudioBuffer(AudioDeviceBuffer* device_buffer,
|