Searched refs:OutputBuffers (Results 1 – 3 of 3) sorted by relevance
361 struct OutputBuffers { struct362 OutputBuffers(xla::ScopedShapedBuffer b, se::DeviceMemoryAllocator* allocator) in OutputBuffers() argument367 ~OutputBuffers() { in ~OutputBuffers() argument394 xla::StatusOr<std::unique_ptr<OutputBuffers>> AllocateOutputTensors( in AllocateOutputTensors()440 absl::make_unique<OutputBuffers>(std::move(scoped_buffers), allocator); in AllocateOutputTensors()765 std::unique_ptr<OutputBuffers> output_buffers, in DoWork()
300 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in EmitDebugSectionImpl() argument306 OutputBuffers[Sec] = MemoryBuffer::getMemBufferCopy(Data); in EmitDebugSectionImpl()
1037 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in emitDebugSectionImpl() argument1047 OutputBuffers[Sec] = MemoryBuffer::getMemBufferCopy(Data); in emitDebugSectionImpl()