using caffe::Layer;
using caffe::shared_ptr;
using caffe::Timer;
-using caffe::string;
using caffe::vector;
// A simple registry for caffe commands.
typedef int (*BrewFunction)();
-typedef std::map<string, BrewFunction> BrewMap;
+typedef std::map<caffe::string, BrewFunction> BrewMap;
BrewMap g_brew_map;
#define RegisterBrewFunction(func) \
__Registerer_##func g_registerer_##func; \
}
-static BrewFunction GetBrewFunction(const string& name) {
+static BrewFunction GetBrewFunction(const caffe::string& name) {
if (g_brew_map.count(name)) {
return g_brew_map[name];
} else {
forward_timer.Start();
Timer timer;
for (int i = 0; i < layers.size(); ++i) {
- const string& layername = layers[i]->layer_param().name();
+ const caffe::string& layername = layers[i]->layer_param().name();
timer.Start();
for (int j = 0; j < FLAGS_run_iterations; ++j) {
layers[i]->Forward(bottom_vecs[i], &top_vecs[i]);
Timer backward_timer;
backward_timer.Start();
for (int i = layers.size() - 1; i >= 0; --i) {
- const string& layername = layers[i]->layer_param().name();
+ const caffe::string& layername = layers[i]->layer_param().name();
timer.Start();
for (int j = 0; j < FLAGS_run_iterations; ++j) {
layers[i]->Backward(top_vecs[i], bottom_need_backward[i],
int main(int argc, char** argv) {
caffe::GlobalInit(&argc, &argv);
CHECK_EQ(argc, 2);
- return GetBrewFunction(string(argv[1]))();
+ return GetBrewFunction(caffe::string(argv[1]))();
}