Home
last modified time | relevance | path

Searched refs:SparseSegmentReductionOperation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_impl.h952 SparseSegmentReductionOperation operation,
970 (operation == SparseSegmentReductionOperation::kSum ? 0 : num_segments),
972 if (operation != SparseSegmentReductionOperation::kSum) {
983 case SparseSegmentReductionOperation::kSum: {
990 case SparseSegmentReductionOperation::kMean: {
994 case SparseSegmentReductionOperation::kSqrtN: {
1018 const T scale = (operation == SparseSegmentReductionOperation::kSum
1057 SparseSegmentReductionOperation operation)
1095 const SparseSegmentReductionOperation operation_;
1104 context, SparseSegmentReductionOperation::kSum) {}
[all …]
Dsegment_reduction_ops.h38 enum class SparseSegmentReductionOperation { kSum, kMean, kSqrtN }; enum
188 SparseSegmentReductionOperation operation,
Dsegment_reduction_ops_gpu.cu.h590 SegmentId nsegments, SparseSegmentReductionOperation operation,
596 if (operation == SparseSegmentReductionOperation::kMean) {
598 } else if (operation == SparseSegmentReductionOperation::kSqrtN) {
607 SparseSegmentReductionOperation operation,
757 SparseSegmentReductionOperation operation,
772 if (operation != SparseSegmentReductionOperation::kSum) {