Searched defs:OutputBuffers (Results 1 – 3 of 3) 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()
1037 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in emitDebugSectionImpl()