Searched refs:FromScopedShapedBuffer (Results 1 – 4 of 4) sorted by relevance
95 TEST(TrackedDeviceBufferTest, FromScopedShapedBuffer) { in TEST() argument106 TrackedDeviceBuffer::FromScopedShapedBuffer(&shaped_buffer, {}); in TEST()
135 static std::shared_ptr<TrackedDeviceBuffer> FromScopedShapedBuffer(
97 TrackedDeviceBuffer::FromScopedShapedBuffer( in FromScopedShapedBuffer() function in xla::TrackedDeviceBuffer
453 TrackedDeviceBuffer::FromScopedShapedBuffer(&dst_buffer, in AllocateDestinationBuffer()1555 TrackedDeviceBuffer::FromScopedShapedBuffer(result_buffer, in OutputBufferHelper()