Searched defs:OutputBuffers (Results 1 – 5 of 5) 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;
1015 : OutputBuffers(componentName, name) { } in OutputBuffers() function1088 : OutputBuffers(componentName, name), in OutputBuffers() function
321 OutputBuffers::OutputBuffers(const char *componentName, const char *name) in OutputBuffers() function in android::OutputBuffers
300 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in EmitDebugSectionImpl()
1037 StringMap<std::unique_ptr<MemoryBuffer>> &OutputBuffers) { in emitDebugSectionImpl()