| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 39 static absl::optional<ReductionKind> MatchReductionComputation( in MatchReductionComputation()
67 return ReductionKind::MIN; in MatchReductionComputation()
69 return ReductionKind::MAX; in MatchReductionComputation()
76 return ReductionKind::SUM; in MatchReductionComputation()
78 return ReductionKind::PRODUCT; in MatchReductionComputation()
80 return ReductionKind::MAX; in MatchReductionComputation()
82 return ReductionKind::MIN; in MatchReductionComputation()
|
| D | nccl_utils.cc | 36 ncclRedOp_t ToNcclReduction(ReductionKind kind) { in ToNcclReduction()
38 case ReductionKind::SUM: in ToNcclReduction()
40 case ReductionKind::PRODUCT: in ToNcclReduction()
42 case ReductionKind::MIN: in ToNcclReduction()
44 case ReductionKind::MAX: in ToNcclReduction()
|
| D | nccl_all_reduce_thunk.h | 32 ReductionKind reduction_kind;
|
| D | nccl_utils.h | 50 ncclRedOp_t ToNcclReduction(ReductionKind kind);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 22 absl::optional<ReductionKind> MatchReductionComputation( in MatchReductionComputation()
39 return ReductionKind::SUM; in MatchReductionComputation()
41 return ReductionKind::PRODUCT; in MatchReductionComputation()
44 return ReductionKind::MIN; in MatchReductionComputation()
47 return ReductionKind::MAX; in MatchReductionComputation()
|
| D | collective_ops_utils.h | 34 enum class ReductionKind { SUM, PRODUCT, MIN, MAX }; enum
37 absl::optional<ReductionKind> MatchReductionComputation(
176 ReductionKind reduction_kind;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 477 xla::ReductionKind reduction_kind = participant.reduction_kind; in DoAllReduce()
538 T GetInitialValue(xla::ReductionKind reduction_kind) { in GetInitialValue()
540 case xla::ReductionKind::SUM: in GetInitialValue()
542 case xla::ReductionKind::PRODUCT: in GetInitialValue()
544 case xla::ReductionKind::MIN: in GetInitialValue()
546 case xla::ReductionKind::MAX: in GetInitialValue()
552 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep()
554 case xla::ReductionKind::SUM: in PerformReductionStep()
556 case xla::ReductionKind::PRODUCT: in PerformReductionStep()
558 case xla::ReductionKind::MIN: in PerformReductionStep()
[all …]
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Analysis/ |
| D | TargetTransformInfo.cpp | 881 enum ReductionKind { enum
890 ReductionData(ReductionKind Kind, unsigned Opcode, Value *LHS, Value *RHS) in ReductionData()
897 ReductionKind Kind = RK_None;
927 static ReductionKind matchPairwiseReductionAtLevel(Instruction *I, in matchPairwiseReductionAtLevel()
1010 static ReductionKind matchPairwiseReduction(const ExtractElementInst *ReduxRoot, in matchPairwiseReduction()
1074 static ReductionKind
|
| /external/llvm-project/llvm/lib/Analysis/ |
| D | TargetTransformInfo.cpp | 1102 static TTI::ReductionKind matchPairwiseReductionAtLevel(Instruction *I, in matchPairwiseReductionAtLevel()
1187 TTI::ReductionKind TTI::matchPairwiseReduction( in matchPairwiseReduction()
1251 TTI::ReductionKind TTI::matchVectorSplittingReduction( in matchVectorSplittingReduction()
1331 TTI::ReductionKind
1334 TTI::ReductionKind RdxKind = matchVectorSplittingReduction(Root, Opcode, Ty); in matchVectorReduction()
1335 if (RdxKind != TTI::ReductionKind::RK_None) { in matchVectorReduction()
|
| /external/llvm-project/llvm/include/llvm/Analysis/ |
| D | TargetTransformInfo.h | 856 enum ReductionKind { enum
866 ReductionData(ReductionKind Kind, unsigned Opcode, Value *LHS, Value *RHS) in ReductionData()
873 ReductionKind Kind = RK_None;
879 static ReductionKind matchPairwiseReduction(
882 static ReductionKind matchVectorSplittingReduction(
885 static ReductionKind matchVectorReduction(const ExtractElementInst *ReduxRoot,
|
| /external/llvm-project/clang/lib/CodeGen/ |
| D | CGOpenMPRuntime.h | 1397 OpenMPDirectiveKind ReductionKind; member
|
| D | CGStmtOpenMP.cpp | 1393 const OMPExecutableDirective &D, const OpenMPDirectiveKind ReductionKind) { in EmitOMPReductionClauseFinal() argument
1422 ReductionKind == OMPD_simd; in EmitOMPReductionClauseFinal()
1423 bool SimpleReduction = ReductionKind == OMPD_simd; in EmitOMPReductionClauseFinal()
1428 {WithNowait, SimpleReduction, ReductionKind}); in EmitOMPReductionClauseFinal()
|
| D | CGOpenMPRuntimeGPU.cpp | 3919 bool ParallelReduction = isOpenMPParallelDirective(Options.ReductionKind); in emitReduction()
3921 bool TeamsReduction = isOpenMPTeamsDirective(Options.ReductionKind); in emitReduction()
|
| D | CodeGenFunction.h | 3355 const OpenMPDirectiveKind ReductionKind);
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Vectorize/ |
| D | SLPVectorizer.cpp | 6107 enum ReductionKind { enum in __anoncaa0dc781f11::HorizontalReduction
6128 ReductionKind Kind = RK_None;
6191 OperationData(unsigned Opcode, Value *LHS, Value *RHS, ReductionKind Kind, in OperationData()
6330 ReductionKind getKind() const { return Kind; } in getKind()
|
| /external/llvm-project/llvm/lib/Transforms/Vectorize/ |
| D | SLPVectorizer.cpp | 6443 enum ReductionKind { enum in __anon42e2fe642411::HorizontalReduction
6458 ReductionKind Kind = RK_None;
6516 OperationData(unsigned Opcode, ReductionKind Kind) in OperationData()
6643 ReductionKind getKind() const { return Kind; } in getKind()
|