Home
last modified time | relevance | path

Searched refs:heap_trace (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h242 void AddHeapTrace(const HeapSimulatorTrace& heap_trace) { in AddHeapTrace() argument
243 heap_traces_.push_back(heap_trace); in AddHeapTrace()
Dbuffer_assignment.cc901 for (const HeapSimulatorTrace& heap_trace : allocation.HeapTraces()) { in ToProto() local
902 *proto.add_heap_simulator_traces() = heap_trace; in ToProto()
1505 const BufferAllocation& allocation, const HeapSimulatorTrace& heap_trace) { in ComputePeakMemoryLogicalBuffers() argument
1536 for (const auto& event : heap_trace.events()) { in ComputePeakMemoryLogicalBuffers()
1553 for (const auto& event : heap_trace.events()) { in ComputePeakMemoryLogicalBuffers()
Dmemory_space_assignment.cc3429 HeapSimulatorTrace* heap_trace = in VerifyAndExportHeapSimulatorTrace() local
3444 HeapSimulatorTrace::Event* heap_trace_event = heap_trace->add_events(); in VerifyAndExportHeapSimulatorTrace()