Searched refs:buffers_defined (Results 1 – 2 of 2) sorted by relevance
145 BufferIdList buffers_defined; member600 for (auto buffer_id : item->buffers_defined) { in AllocatedSize()849 item->buffers_defined.push_back(buffer->id); in MemoryUsageTracker()881 for (BufferId buffer_id : item->buffers_defined) { in BeginInstruction()919 for (BufferId buffer_id : in_progress_item_->buffers_defined) { in EndInstruction()981 for (BufferId buffer_id : item->buffers_defined) { in MemoryReducedIfRematerialized()1057 compressed_item->buffers_defined.push_back(compressed_buffer.id); in AddCompressInstructions()1066 uncompressed_item->buffers_defined = {uncompressed_buffer.id}; in AddCompressInstructions()1114 for (BufferId old_buffer_id : original_item->buffers_defined) { in AddRematerializedInstruction()1130 CHECK(user.user->buffers_defined.empty()) in AddRematerializedInstruction()[all …]
172 std::vector<std::vector<const HloValue*>> buffers_defined( in RunComputation() local218 buffers_defined[live_range.start].push_back(value); in RunComputation()233 for (const HloValue* value : buffers_defined[i]) { in RunComputation()