Searched defs:memory_color (Results 1 – 1 of 1) sorted by relevance
259 int64_t heap_simulator_trace_id, int64_t memory_color) { in ConvertHloProtoToPreprocessResult()568 int64_t memory_color) { in GetHeapSimulatorTraceIdFromEvents()603 int64_t memory_color) { in GetHeapSimulatorTraceIdFromBufferAllocationIndex()634 int64_t GetHeapSimulatorTraceId(const HloProto& proto, int64_t memory_color) { in GetHeapSimulatorTraceId()