Home
last modified time | relevance | path

Searched defs:c_device_buffer (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_transfer_manager.cc78 XLA_ShapedBuffer c_device_buffer; in TransferLiteralToDeviceAsync() local
199 XLA_ShapedBuffer c_device_buffer; in TransferLiteralFromDevice() local
249 XLA_ShapedBuffer c_device_buffer; in CanShapedBufferBeAccessedNow() local
263 SE_DeviceMemoryBase c_device_buffer{const_cast<void*>(device_buffer.opaque()), in CanBufferBeAccessedNow() local
331 XLA_ShapedBuffer c_device_buffer; in ReadDynamicShapes() local
H A Dc_api_conversions.cc381 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer) { in ToC()