Searched refs:XlaTensorBuffer (Results 1 – 2 of 2) sorted by relevance
191 class XlaTensorBuffer : public TensorBuffer {193 XlaTensorBuffer(const void* ptr, size_t expected_size, size_t actual_size, in XlaTensorBuffer() function200 ~XlaTensorBuffer() override { in ~XlaTensorBuffer()
293 auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, in MakeTensor()