platform: msm_shared: Change to enable collection for dtbo stats

Change to enable collection of dtbo boot stats while booting.

Change-Id: Ie76b243c17af6be23d854367ac76144534e8eaeb
diff --git a/platform/msm_shared/boot_stats.c b/platform/msm_shared/boot_stats.c
index 20635f2..05b0ed4 100644
--- a/platform/msm_shared/boot_stats.c
+++ b/platform/msm_shared/boot_stats.c
@@ -55,6 +55,16 @@
 				writel(clk_count - kernel_load_start,
 					bs_imem + (sizeof(uint32_t) * BS_KERNEL_LOAD_TIME));
 			}
+			return;
+		}
+		if(bs_id == BS_DTB_OVERLAY_START){
+			clk_count = platform_get_sclk_count();
+			dprintf(INFO, "DTBO Overlay count start: %u\n", clk_count);
+			return;
+		}
+		if(bs_id == BS_DTB_OVERLAY_END){
+			clk_count = platform_get_sclk_count();
+			dprintf(INFO, "DTBO Overlay count done: %u\n", clk_count);
 		}
 		else{
 			clk_count = platform_get_sclk_count();