Searched refs:SparseSegmentReductionOperation (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | segment_reduction_ops_impl.h | 1033 SparseSegmentReductionOperation operation,
1051 (operation == SparseSegmentReductionOperation::kSum ? 0 : num_segments),
1053 if (operation != SparseSegmentReductionOperation::kSum) {
1064 case SparseSegmentReductionOperation::kSum: {
1071 case SparseSegmentReductionOperation::kMean: {
1075 case SparseSegmentReductionOperation::kSqrtN: {
1099 const T scale = (operation == SparseSegmentReductionOperation::kSum
1138 SparseSegmentReductionOperation operation)
1176 const SparseSegmentReductionOperation operation_;
1185 context, SparseSegmentReductionOperation::kSum) {}
[all …]
|
| H A D | segment_reduction_ops.h | 32 enum class SparseSegmentReductionOperation { kSum, kMean, kSqrtN }; enum
150 SparseSegmentReductionOperation operation,
|
| H A D | segment_reduction_ops_gpu.cu.h | 649 SegmentId nsegments, SparseSegmentReductionOperation operation,
655 if (operation == SparseSegmentReductionOperation::kMean) {
657 } else if (operation == SparseSegmentReductionOperation::kSqrtN) {
666 SparseSegmentReductionOperation operation,
937 SparseSegmentReductionOperation operation,
952 if (operation != SparseSegmentReductionOperation::kSum) {
|