Searched defs:OutputBuffers (Results 1 – 2 of 2) sorted by relevance
361 struct OutputBuffers { struct362 OutputBuffers(xla::ScopedShapedBuffer b, se::DeviceMemoryAllocator* allocator) in OutputBuffers() function367 ~OutputBuffers() { in ~OutputBuffers()381 xla::ShapeTree<bool> owned_buffers;383 xla::ShapedBuffer buffers;385 se::DeviceMemoryAllocator* const memory_allocator;
300 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in EmitDebugSectionImpl()