Searched defs:logical_buffer (Results 1 – 4 of 4) sorted by relevance
76 std::string GetInstructionName(const LogicalBufferProto* logical_buffer) { in GetInstructionName()273 for (const auto& logical_buffer : in ConvertHloProtoToPreprocessResult() local296 const LogicalBufferProto* logical_buffer = in ConvertHloProtoToPreprocessResult() local340 const auto* logical_buffer = id_to_logical_buffer.at(event.buffer_id()); in ConvertHloProtoToPreprocessResult() local411 const BufferAllocationProto* buffer_allocation) { in ConvertHloProtoToPreprocessResult()438 for (const LogicalBufferProto* logical_buffer : all_logical_buffers) { in ConvertHloProtoToPreprocessResult() local443 for (const LogicalBufferProto* logical_buffer : unseen) { in ConvertHloProtoToPreprocessResult() local496 const auto* logical_buffer = id_to_logical_buffer.at(logical_buffer_id); in ConvertHloProtoToPreprocessResult() local571 for (const auto& logical_buffer : in GetHeapSimulatorTraceIdFromEvents() local
450 const LogicalBuffer* logical_buffer, in GetUsers()665 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer()812 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker() local862 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker() local
221 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer() function
1937 for (const HloValue* logical_buffer : peak_buffers) { in TEST_F() local