Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Static Public Member Functions | Static Public Attributes
caffe2::AbstractSparseSortedSegmentDef< T, SIndex, Context, ReducerDef > Struct Template Reference

Public Types

using OpDef = ReducerDef
using Reducer = typename ReducerDef::template Reducer< T, Context >
using ReducerGradient = typename ReducerDef::template ReducerGradient< T, Context >
using ForwardOp = AbstractSortedSegmentOp< T, SIndex, Context, Reducer >
using BackwardOp = AbstractSortedSegmentGradientOp< T, SIndex, Context, ReducerGradient >
using GetGradient = SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, true, true >

Static Public Member Functions

static void PopulateSchema (OpSchema &schema)

Static Public Attributes

static constexpr const char * basename = "SparseSortedSegment"
static constexpr const char * doc

Detailed Description

template<typename T, typename SIndex, typename Context, typename ReducerDef>
struct caffe2::AbstractSparseSortedSegmentDef< T, SIndex, Context, ReducerDef >

Definition at line 893 of file segment_reduction_op.h.

Field Documentation

template<typename T , typename SIndex , typename Context , typename ReducerDef >
constexpr const char* caffe2::AbstractSparseSortedSegmentDef< T, SIndex, Context, ReducerDef >::doc
Initial value:
= R"DOC(
Pulls in slices of the input tensor, groups them into segments and applies
'{op}' to each segment. Segments need to be sorted and contiguous. See also
SparseUnsortedSegment{op} that doesn't have this requirement.
This op is basically Gather and SortedSegment{op} fused together.
INDICES should contain integers in range 0..N-1 where N is the first dimension
of DATA. INDICES represent which slices of DATA need to be pulled in.
SEGMENT_IDS is a vector that maps each referenced slice of the DATA to a
particular group (segment). Values belonging to the same segment are aggregated
together. SEGMENT_IDS should have the same dimension as INDICES.
The first dimension of the output is equal to the number of input segments,
i.e. `SEGMENT_IDS[-1]+1`. Other dimensions are inherited from the input tensor.

Definition at line 896 of file segment_reduction_op.h.

The documentation for this struct was generated from the following file: