Searched defs:OutputBuffers (Results 1 – 2 of 2) sorted by relevance
360 struct OutputBuffers { struct361 OutputBuffers(xla::ScopedShapedBuffer b, se::DeviceMemoryAllocator* allocator) in OutputBuffers() argument366 ~OutputBuffers() { in ~OutputBuffers()380 xla::ShapeTree<bool> owned_buffers;382 xla::ShapedBuffer buffers;384 se::DeviceMemoryAllocator* const memory_allocator;
300 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in EmitDebugSectionImpl()