File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -137,13 +137,13 @@ static std::vector<float> compute_tensor_averages(const Stats & tstats) {
137137 std::vector<float > vec;
138138 vec.reserve (len);
139139
140- if (! tstats.activations .empty ()) {
140+ if (tstats.activations .empty ()) {
141141 for (size_t m = 0 ; m < n_mat; ++m) {
142142 const float c = (float )tstats.counts [m];
143143 if (c <= 0 ) return {};
144144 const size_t off = m * row;
145145 for (size_t j = 0 ; j < row; ++j) {
146- vec.push_back (tstats.activations [off + j] / c);
146+ vec.push_back (tstats.values [off + j] / c);
147147 }
148148 }
149149 } else {
@@ -152,7 +152,7 @@ static std::vector<float> compute_tensor_averages(const Stats & tstats) {
152152 if (c <= 0 ) return {};
153153 const size_t off = m * row;
154154 for (size_t j = 0 ; j < row; ++j) {
155- vec.push_back (tstats.values [off + j] / c);
155+ vec.push_back (tstats.activations [off + j] / c);
156156 }
157157 }
158158 }
You can’t perform that action at this time.
0 commit comments