Searched refs:XlaTensorBuffer (Results 1 – 3 of 3) sorted by relevance
198 class XlaTensorBuffer : public TensorBuffer {200 XlaTensorBuffer(const void* ptr, size_t expected_size, size_t actual_size, in XlaTensorBuffer() function207 ~XlaTensorBuffer() override { in ~XlaTensorBuffer()224 auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, in MakeTensor()
370 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs()434 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs()
617 friend class XlaTensorBuffer; // For access to the private constructor taking variable