1 #ifndef CAFFE2_OPERATORS_CONV_TRANSPOSE_OP_H_ 2 #define CAFFE2_OPERATORS_CONV_TRANSPOSE_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 6 #include "caffe2/operators/conv_transpose_unpool_op_base.h" 10 template <
typename T,
class Context>
13 USE_CONV_TRANSPOSE_UNPOOL_BASE_FUNCTIONS(Context);
17 bool RunOnDeviceWithOrderNCHW()
override;
18 bool RunOnDeviceWithOrderNHWC()
override;
25 INPUT_TAGS(INPUT, FILTER, BIAS);
28 template <
typename T,
class Context>
31 USE_CONV_TRANSPOSE_UNPOOL_BASE_FUNCTIONS(Context);
34 no_bias_(OperatorBase::GetSingleArgument<bool>(
"no_bias",
false)) {
36 !(no_bias_ && OutputSize() == 3),
37 "If bias is not present, you should not have 3 grad output.");
40 bool RunOnDeviceWithOrderNCHW()
override;
41 bool RunOnDeviceWithOrderNHWC()
override;
49 INPUT_TAGS(INPUT, FILTER, OUTPUT_GRAD);
50 OUTPUT_TAGS(FILTER_GRAD, BIAS_OR_INPUT_GRAD, INPUT_GRAD);
55 #endif // CAFFE2_OPERATORS_CONV_TRANSPOSE_OP_H_
Tensor is the basic class in Caffe2 that stores a contiguous memory with its shape information...
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 ...