Searched defs:device_buffer (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | transfer_manager.h | 75 se::Stream* stream, const ShapedBuffer& device_buffer) { in TransferLiteralFromDevice() 83 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() 105 const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() 129 const ShapedBuffer& device_buffer) { in TransferLiteralToDevice() 155 const ShapedBuffer& device_buffer) { in TransferLiteralToDeviceAsync() 242 se::StreamExecutor* executor, const ShapedBuffer& device_buffer) const { in CanShapedBufferBeAccessedNow()
|
D | transfer_manager.cc | 48 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() 75 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() 97 const ShapedBuffer& device_buffer, in TransferLiteralToDevice() 234 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTables() 240 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteTupleIndexTablesAsync() 268 se::Stream* stream, const ShapedBuffer& device_buffer) { in WriteRootTupleIndexTable()
|
D | generic_transfer_manager.cc | 59 se::Stream* stream, const ShapedBuffer& device_buffer, in TransferLiteralFromDevice() 72 se::StreamExecutor* executor, const ShapedBuffer& device_buffer, in TransferLiteralFromDeviceInternal() 101 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 108 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local 124 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local 140 auto device_buffer = AllocateDeviceBuffer(shape); in XLA_TEST_F() local 159 auto device_buffer = AllocateDeviceBuffer(ondevice_shape); in XLA_TEST_F() local 180 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local 194 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local 213 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local 228 auto device_buffer = AllocateDeviceBuffer(literal.shape()); in XLA_TEST_F() local [all …]
|
/external/webrtc/webrtc/modules/audio_device/ |
D | fine_audio_buffer.cc | 23 FineAudioBuffer::FineAudioBuffer(AudioDeviceBuffer* device_buffer, in FineAudioBuffer()
|