Home
last modified time | relevance | path

Searched refs:id_to_buffer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1248 absl::flat_hash_map<LogicalBuffer::Id, const LogicalBuffer*> id_to_buffer; in ComputePeakMemoryLogicalBuffers() local
1253 id_to_buffer[buffer->id()] = buffer; in ComputePeakMemoryLogicalBuffers()
1259 auto memory_delta = [&id_to_buffer, &buffer_sizes]( in ComputePeakMemoryLogicalBuffers()
1261 const LogicalBuffer* buffer = id_to_buffer.at(event.buffer_id()); in ComputePeakMemoryLogicalBuffers()
1291 const LogicalBuffer* buffer = id_to_buffer.at(event.buffer_id()); in ComputePeakMemoryLogicalBuffers()
/external/perfetto/src/tracing/core/
Dtracing_service_impl.cc2066 for (const auto& id_to_buffer : buffers_) { in UpdateMemoryGuardrail() local
2067 total_buffer_bytes += id_to_buffer.second->size(); in UpdateMemoryGuardrail()