diff options
Diffstat (limited to 'tensorflow/core/kernels/segment_reduction_ops.h')
-rw-r--r-- | tensorflow/core/kernels/segment_reduction_ops.h | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/tensorflow/core/kernels/segment_reduction_ops.h b/tensorflow/core/kernels/segment_reduction_ops.h index 412c1d601d..b10bea72ba 100644 --- a/tensorflow/core/kernels/segment_reduction_ops.h +++ b/tensorflow/core/kernels/segment_reduction_ops.h @@ -30,14 +30,14 @@ namespace functor { #ifdef GOOGLE_CUDA typedef Eigen::GpuDevice GPUDevice; // Functor for SegmentSumGPUOp. -// 'output_rows': the number of output segments (unique segment ids in +// output_rows: the number of output segments (unique segment ids in // 'segment_ids'). -// 'segment_ids_shape': shape of 'segment_ids' tensor. -// 'segment_ids': unsorted map from input to output segment ids at which to +// segment_ids_shape: shape of 'segment_ids' tensor. +// segment_ids: unsorted map from input to output segment ids at which to // perform segment sum operation. -// 'data_size': size of input data tensor. -// 'data': input data tensor. -// 'output': output reshaped to {output_rows, output.size/output_rows} +// data_size: size of input data tensor. +// data: input data tensor. +// output: output reshaped to {output_rows, output.size/output_rows} template <typename T, typename Index> struct SegmentSumFunctor { void operator()(OpKernelContext* ctx, const GPUDevice& d, @@ -61,14 +61,14 @@ struct UnsortedSegmentBaseFunctor{ }; // Functor for UnsortedSegmentSumOp. -// 'output_rows': the number of output segments (unique segment ids in +// output_rows: the number of output segments (unique segment ids in // 'segment_ids'). -// 'segment_ids_shape': shape of 'segment_ids' tensor. -// 'segment_ids': unsorted map from input to output segment ids at which to +// segment_ids_shape: shape of 'segment_ids' tensor. +// segment_ids: unsorted map from input to output segment ids at which to // perform segment sum operation. -// 'data_size': size of input data tensor. -// 'data': input data tensor. -// 'output': output reshaped to {output_rows, output.size/output_rows} +// data_size: size of input data tensor. +// data: input data tensor. +// output: output reshaped to {output_rows, output.size/output_rows} template <typename Device, typename T, typename Index> struct UnsortedSegmentSumFunctor: public UnsortedSegmentBaseFunctor<Device, T, Index> { void operator()(OpKernelContext* ctx, const Device& d, @@ -79,14 +79,14 @@ struct UnsortedSegmentSumFunctor: public UnsortedSegmentBaseFunctor<Device, T, I }; // Functor for UnsortedSegmentMaxOp. -// 'output_rows': the number of output segments (unique segment ids in +// output_rows: the number of output segments (unique segment ids in // 'segment_ids'). -// 'segment_ids_shape': shape of 'segment_ids' tensor. -// 'segment_ids': unsorted map from input to output segment ids at which to +// segment_ids_shape: shape of 'segment_ids' tensor. +// segment_ids: unsorted map from input to output segment ids at which to // perform segment sum operation. -// 'data_size': size of input data tensor. -// 'data': input data tensor. -// 'output': output reshaped to {output_rows, output.size/output_rows} +// data_size: size of input data tensor. +// data: input data tensor. +// output: output reshaped to {output_rows, output.size/output_rows} template <typename Device, typename T, typename Index> struct UnsortedSegmentMaxFunctor: public UnsortedSegmentBaseFunctor<Device, T, Index> { void operator()(OpKernelContext* ctx, const Device& d, |