1 #ifndef CAFFE2_OPERATORS_MAX_POOL_WITH_INDEX_H_ 2 #define CAFFE2_OPERATORS_MAX_POOL_WITH_INDEX_H_ 5 #include "caffe2/core/context.h" 6 #include "caffe2/core/context_gpu.h" 7 #include "caffe2/core/logging.h" 8 #include "caffe2/core/operator.h" 9 #include "caffe2/operators/conv_pool_op_base.h" 10 #include "caffe2/operators/pool_op.h" 11 #include "caffe2/utils/math.h" 25 bool RunOnDevice()
override;
41 bool RunOnDevice()
override;
49 #endif // CAFFE2_OPERATORS_MAX_POOL_WITH_INDEX_H_
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...