Searched refs:buffers_defined (Results 1 – 2 of 2) sorted by relevance
141 BufferIdList buffers_defined; member591 for (auto buffer_id : item->buffers_defined) { in AllocatedSize()838 item->buffers_defined.push_back(buffer->id); in MemoryUsageTracker()870 for (BufferId buffer_id : item->buffers_defined) { in BeginInstruction()908 for (BufferId buffer_id : in_progress_item_->buffers_defined) { in EndInstruction()970 for (BufferId buffer_id : item->buffers_defined) { in MemoryReducedIfRematerialized()1046 compressed_item->buffers_defined.push_back(compressed_buffer.id); in AddCompressInstructions()1055 uncompressed_item->buffers_defined = {uncompressed_buffer.id}; in AddCompressInstructions()1103 for (BufferId old_buffer_id : original_item->buffers_defined) { in AddRematerializedInstruction()1119 CHECK(user.user->buffers_defined.empty()) in AddRematerializedInstruction()[all …]
169 std::vector<std::vector<const HloValue*>> buffers_defined( in RunComputation() local215 buffers_defined[live_range.start].push_back(value); in RunComputation()230 for (const HloValue* value : buffers_defined[i]) { in RunComputation()