1 #ifndef CAFFE2_OPENGL_OPERATORS_ACTIVATION_OPS_H_ 2 #define CAFFE2_OPENGL_OPERATORS_ACTIVATION_OPS_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 15 bool RunOnDevice()
override;
17 arm_compute::GCActivationLayer sigmoid_layer_;
18 bool first_run_ =
true, second_run_ =
true;
19 GLContext::deleted_unique_ptr<const GLTensor<T>> X_;
28 bool RunOnDevice()
override;
30 arm_compute::GCActivationLayer relu_layer_;
31 bool first_run_ =
true, second_run_ =
true;
32 GLContext::deleted_unique_ptr<const GLTensor<T>> X_;
38 #endif // CAFFE2_OPENGL_OPERATORS_ACTIVATION_OPS_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 ...