From 2947e1f9d43d5087080ed8afa623046ce950a4f1 Mon Sep 17 00:00:00 2001
From: Lei Zhang <zhang.huanning@hotmail.com>
Date: Fri, 19 Apr 2019 15:28:54 -0700
Subject: [PATCH] Fix onehot code arm build break.

---
 onnxruntime/core/providers/cpu/tensor/onehot.cc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/onnxruntime/core/providers/cpu/tensor/onehot.cc b/onnxruntime/core/providers/cpu/tensor/onehot.cc
index 16e213b2a1..f76c6efc8c 100644
--- a/onnxruntime/core/providers/cpu/tensor/onehot.cc
+++ b/onnxruntime/core/providers/cpu/tensor/onehot.cc
@@ -134,13 +134,15 @@ Status OneHotOp<in_type, out_type, depth_type>::Compute(OpKernelContext* p_op_ke
   const int64_t suffix_dim_size = indices_shape.Size() / prefix_dim_size;
 
   // Split indices into matrix of size prefix_dim_size x suffix_dim_size
-  Eigen::array<Eigen::DenseIndex, 2> indices_dims_e = {{prefix_dim_size, suffix_dim_size}};
+  Eigen::array<Eigen::DenseIndex, 2> indices_dims_e = {
+    {static_cast<Eigen::DenseIndex>(prefix_dim_size), static_cast<Eigen::DenseIndex>(suffix_dim_size)}};
   const auto* indices_data = indices->Data<in_type>();
   typename EigenTensorTypes<in_type, 2>::ConstEigenTensorMap indices_tensor_e(indices_data, indices_dims_e);
 
   // Split output into 3-Tensor of size:
   //   prefix_dim_size x depth x suffix_dim_size.
-  Eigen::array<Eigen::DenseIndex, 3> output_dims_e = {{prefix_dim_size, depth_val, suffix_dim_size}};
+  Eigen::array<Eigen::DenseIndex, 3> output_dims_e = {
+    {static_cast<Eigen::DenseIndex>(prefix_dim_size), static_cast<Eigen::DenseIndex>(depth_val), static_cast<Eigen::DenseIndex>(suffix_dim_size)}};
   auto* output_data = output->MutableData<out_type>();
   typename EigenTensorTypes<out_type, 3>::EigenTensorMap output_tensor_e(output_data, output_dims_e);