11#pragma once
22#include < algorithm>
33#include < cmath>
4+ #include < cstdint>
45
56#include " layers/Layer.hpp"
67
@@ -76,10 +77,8 @@ class InputLayer : public Layer {
7677 }
7778 }
7879 }
79- Shape sh1 ({static_cast <unsigned long long >(n),
80- static_cast <unsigned long long >(h),
81- static_cast <unsigned long long >(w),
82- static_cast <unsigned long long >(c)});
80+ Shape sh1 ({static_cast <uint64_t >(n), static_cast <uint64_t >(h),
81+ static_cast <uint64_t >(w), static_cast <uint64_t >(c)});
8382 output[0 ] = make_tensor<int >(res, sh1);
8483 break ;
8584 }
@@ -103,10 +102,8 @@ class InputLayer : public Layer {
103102 }
104103 }
105104 }
106- Shape sh1 ({static_cast <unsigned long long >(n),
107- static_cast <unsigned long long >(c),
108- static_cast <unsigned long long >(h),
109- static_cast <unsigned long long >(w)});
105+ Shape sh1 ({static_cast <uint64_t >(n), static_cast <uint64_t >(c),
106+ static_cast <uint64_t >(h), static_cast <uint64_t >(w)});
110107 output[0 ] = make_tensor<int >(res, sh1);
111108 break ;
112109 }
@@ -144,10 +141,8 @@ class InputLayer : public Layer {
144141 }
145142 }
146143 }
147- Shape sh1 ({static_cast <unsigned long long >(n),
148- static_cast <unsigned long long >(h),
149- static_cast <unsigned long long >(w),
150- static_cast <unsigned long long >(c)});
144+ Shape sh1 ({static_cast <uint64_t >(n), static_cast <uint64_t >(h),
145+ static_cast <uint64_t >(w), static_cast <uint64_t >(c)});
151146 output[0 ] = make_tensor<float >(res, sh1);
152147 break ;
153148 }
@@ -171,10 +166,8 @@ class InputLayer : public Layer {
171166 }
172167 }
173168 }
174- Shape sh1 ({static_cast <unsigned long long >(n),
175- static_cast <unsigned long long >(c),
176- static_cast <unsigned long long >(h),
177- static_cast <unsigned long long >(w)});
169+ Shape sh1 ({static_cast <uint64_t >(n), static_cast <uint64_t >(c),
170+ static_cast <uint64_t >(h), static_cast <uint64_t >(w)});
178171 output[0 ] = make_tensor<float >(res, sh1);
179172 break ;
180173 }
0 commit comments