|
using | OpDef = ReducerDef |
|
using | ReducerGradient = typename ReducerDef::template ReducerGradient< T, Context > |
|
using | ForwardOp = AbstractReduceFrontOrBackOp< T, Context, typename ReducerDef::template Reducer< T, Context >, true > |
|
using | BackwardOp = AbstractReduceFrontOrBackGradientOp< T, Context, ReducerGradient, true > |
|
|
static void | PopulateSchema (OpSchema &schema) |
|
|
static constexpr const char * | basename = "ReduceFront" |
|
static constexpr const char * | doc |
|
template<typename T, typename Context, typename ReducerDef>
struct caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >
Definition at line 431 of file segment_reduction_op.h.
template<typename T , typename Context , typename ReducerDef >
Initial value:= R"DOC(
Reduces the input tensor along the first dimension of the input tensor by
applying '{op}'. This op acts in a similar way to SortedSegment{op} and
UnsortedSegment{op} but as if all input slices belong to a single segment.
{op_doc}
)DOC"
Definition at line 434 of file segment_reduction_op.h.
The documentation for this struct was generated from the following file: