1 #include "caffe2/mobile/contrib/arm-compute/core/context.h" 2 #include "caffe2/mobile/contrib/arm-compute/core/operator.h" 3 #include "caffe2/operators/utility_ops.h" 13 bool RunOnDevice()
override;
15 arm_compute::GCArithmeticAddition add_layer_;
16 bool first_run_ =
true, second_run_ =
true;
17 GLContext::deleted_unique_ptr<const GLTensor<T>> A_, B_;
24 auto *Ablob = OperatorBase::Inputs()[0];
25 auto *Bblob = OperatorBase::Inputs()[1];
28 A_ = GLContext::getGLTensor<T>(Ablob);
29 B_ = GLContext::getGLTensor<T>(Bblob);
33 OperatorBase::Outputs()[0]->template GetMutable<GLTensor<T>>();
37 add_layer_.configure(A_->get_underlying(), B_->get_underlying(), Y->get_underlying(), arm_compute::ConvertPolicy::SATURATE);
39 A_->lazy_allocate(Ablob, second_run_,
true);
40 B_->lazy_allocate(Bblob, second_run_,
true);
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 ...