Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h75 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()
Dtransfer_manager.cc48 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()
Dgeneric_transfer_manager.cc59 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/
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
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/
Dfine_audio_buffer.cc23 FineAudioBuffer::FineAudioBuffer(AudioDeviceBuffer* device_buffer, in FineAudioBuffer()