Home
last modified time | relevance | path

Searched refs:c_device_buffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.cc76 XLA_ShapedBuffer c_device_buffer; in TransferLiteralToDeviceAsync() local
77 ApiConverter::ToC(device_buffer, &c_device_buffer); in TransferLiteralToDeviceAsync()
83 &c_literal, &c_device_buffer, status.c_status); in TransferLiteralToDeviceAsync()
84 ApiConverter::Free(&c_device_buffer); in TransferLiteralToDeviceAsync()
197 XLA_ShapedBuffer c_device_buffer; in TransferLiteralFromDevice() local
198 ApiConverter::ToC(device_buffer, &c_device_buffer); in TransferLiteralFromDevice()
206 &c_device_buffer, &c_literal, TransferLiteralFromDeviceTrampoline, state); in TransferLiteralFromDevice()
207 ApiConverter::Free(&c_device_buffer); in TransferLiteralFromDevice()
247 XLA_ShapedBuffer c_device_buffer; in CanShapedBufferBeAccessedNow() local
248 ApiConverter::ToC(device_buffer, &c_device_buffer); in CanShapedBufferBeAccessedNow()
[all …]
Dc_api_conversions.h71 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer);
86 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer);
Dc_api_conversions.cc351 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer) { in ToC() argument
353 &c_device_buffer->on_device_shape); in ToC()
354 c_device_buffer->device_ordinal = buffer.device_ordinal(); in ToC()
359 c_device_buffer->count = bases.size(); in ToC()
360 c_device_buffer->bases = new SE_DeviceMemoryBase[bases.size()]; in ToC()
362 c_device_buffer->bases[i] = bases[i]; in ToC()