Searched refs:logical_buffers (Results 1 – 5 of 5) sorted by relevance
203 if (!result->logical_buffers().empty()) { in Convert()204 std::string common_shape = result->logical_buffers(0).shape(); in Convert()206 if (result->logical_buffers(i).shape() != common_shape) { in Convert()274 hlo_proto.buffer_assignment().logical_buffers()) { in ConvertHloProtoToPreprocessResult()310 std::vector<int64_t> logical_buffers; in ConvertHloProtoToPreprocessResult() local351 logical_buffers.push_back(event.buffer_id()); in ConvertHloProtoToPreprocessResult()365 peak_logical_buffers = logical_buffers; in ConvertHloProtoToPreprocessResult()368 logical_buffers.erase( in ConvertHloProtoToPreprocessResult()369 std::remove(logical_buffers.begin(), logical_buffers.end(), in ConvertHloProtoToPreprocessResult()371 logical_buffers.end()); in ConvertHloProtoToPreprocessResult()[all …]
42 const std::vector<std::unique_ptr<LogicalBuffer>>& logical_buffers() const { in logical_buffers() function
686 for (const auto& b : logical_buffer_analysis_->logical_buffers()) { in ToString()
622 repeated LogicalBufferProto logical_buffers = 1; field
46 repeated LogicalBuffer logical_buffers = 4; field