saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix onehot code arm build break.

Browse source
This commit is contained in:
Lei Zhang 2019-04-19 15:28:54 -07:00 committed by Zhang Lei
parent 2879ee8bd2
commit 2947e1f9d4
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/cpu/tensor/onehot.cc
View file

@ -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);