3 #include "caffe2/core/context.h" 4 #include "caffe2/core/logging.h" 5 #include "caffe2/core/operator.h" 6 #include "caffe2/utils/math.h" 10 template <
typename T,
class Context>
15 order_(StringToStorageOrder(
16 OperatorBase::GetSingleArgument<string>(
"order",
"NCHW"))) {}
18 USE_OPERATOR_CONTEXT_FUNCTIONS;
20 bool RunOnDevice()
override;
26 template <
typename T,
class Context>
31 order_(StringToStorageOrder(
32 OperatorBase::GetSingleArgument<string>(
"order",
"NCHW"))) {}
33 USE_OPERATOR_CONTEXT_FUNCTIONS;
35 bool RunOnDevice()
override;
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 ...