diff options
| author | Sergio Guadarrama <sguada@gmail.com> | 2014-05-15 16:09:07 -0700 |
|---|---|---|
| committer | Sergey Karayev <sergeykarayev@gmail.com> | 2014-05-20 21:25:24 -0700 |
| commit | d19c18060134b7b409291cf7c9ffe815b1a9b0df (patch) | |
| tree | 00e136e315f652063f8c84b16007ab46cf528557 /src/caffe/layers/argmax_layer.cpp | |
| parent | bdcd75e48746be0aab0d4c8b606eeb2aac6c9ae8 (diff) | |
| download | caffeonacl-d19c18060134b7b409291cf7c9ffe815b1a9b0df.tar.gz caffeonacl-d19c18060134b7b409291cf7c9ffe815b1a9b0df.tar.bz2 caffeonacl-d19c18060134b7b409291cf7c9ffe815b1a9b0df.zip | |
Added FLT_MAX to argmax layer
Diffstat (limited to 'src/caffe/layers/argmax_layer.cpp')
| -rw-r--r-- | src/caffe/layers/argmax_layer.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/caffe/layers/argmax_layer.cpp b/src/caffe/layers/argmax_layer.cpp index b09fd788..33ec1d34 100644 --- a/src/caffe/layers/argmax_layer.cpp +++ b/src/caffe/layers/argmax_layer.cpp @@ -1,12 +1,11 @@ // Copyright 2014 BVLC and contributors. -#include <algorithm> #include <vector> +#include <cfloat> #include "caffe/layer.hpp" #include "caffe/vision_layers.hpp" -using std::max; namespace caffe { |