Searched refs:SparseSegmentReductionOperation (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | segment_reduction_ops_impl.h | 952 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 …]
|
D | segment_reduction_ops.h | 38 enum class SparseSegmentReductionOperation { kSum, kMean, kSqrtN }; enum 188 SparseSegmentReductionOperation operation,
|
D | segment_reduction_ops_gpu.cu.h | 590 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) {
|