Searched refs:XLA_ShapedBuffer (Results 1 – 5 of 5) sorted by relevance
71 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer);72 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer);73 void Free(XLA_ShapedBuffer* c_buffer);86 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer);87 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer);88 void Free(XLA_ShapedBuffer* c_buffer);
229 typedef struct XLA_ShapedBuffer { struct235 } XLA_ShapedBuffer; argument263 XLA_ShapedBuffer result;
76 XLA_ShapedBuffer c_device_buffer; in TransferLiteralToDeviceAsync()197 XLA_ShapedBuffer c_device_buffer; in TransferLiteralFromDevice()247 XLA_ShapedBuffer c_device_buffer; in CanShapedBufferBeAccessedNow()
184 XLA_ShapedBuffer* device_buffer, TF_Status* status);187 XLA_ShapedBuffer* device_buffer, XLA_Literal* literal,196 XLA_ShapedBuffer* device_buffer);
25 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer) { in FromC()351 void ToC(const xla::ShapedBuffer& buffer, XLA_ShapedBuffer* c_device_buffer) { in ToC()375 void Free(XLA_ShapedBuffer* c_buffer) { in Free()