| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Reduce.cuh | 341 struct ReduceOp { struct
342 using traits = function_traits<decltype(&ops_t::reduce)>;
343 using arg_t = typename std::decay<typename traits::template arg<0>::type>::type;
345 using InputCalculator = OffsetCalculator<1, index_t>;
346 using OutputCalculator = OffsetCalculator<2, index_t>;
348 static constexpr bool can_accumulate_in_output =
352 static constexpr int input_vec_size = ReduceConfig::input_vec_size;
354 ops_t ops;
355 arg_t ident;
356 ReduceConfig config;
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Types.hpp | 34 struct TORCH_API ReduceOp : torch::CustomClassHolder { struct
36 enum RedOpType : uint8_t {
51 ReduceOp(RedOpType op) : op_(op) { in ReduceOp() function
57 ReduceOp( in ReduceOp() argument
75 operator RedOpType() const { in operator RedOpType()
79 bool operator==(const std::uint8_t other) { in operator ==()
84 bool operator==(const ReduceOp::RedOpType other) { in operator ==()
89 bool operator==(const ReduceOp& other) { in operator ==()
93 RedOpType op_ = SUM;
101 c10::intrusive_ptr<_SupplementBase> supplement_;
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | reduction.h | 141 ReduceOp( in ReduceOp() function
154 ReduceOp( in ReduceOp() function
|
| H A D | fwd_decls.h | 62 class ReduceOp; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | reduce_util.py | 24 class ReduceOp(enum.Enum): class
|
| H A D | cross_device_ops_test.py | 53 ReduceOp = reduce_util.ReduceOp variable
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
| H A D | _fsdp_collectives.py | 289 reduce_scatter_reduce_op: Optional[Union[dist.ReduceOp, dist.ReduceOp.RedOpType]],
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | distributed_c10d.py | 208 def supports_complex(reduceOp: ReduceOp) -> bool:
538 redop: Optional[ReduceOp] = None,
|
| /aosp_15_r20/external/mesa3d/src/amd/compiler/ |
| H A D | aco_ir.h | 1720 enum ReduceOp : uint16_t { enum
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/AArch64/ |
| H A D | AArch64ISelLowering.cpp | 23377 SDValue AArch64TargetLowering::LowerPredReductionToSVE(SDValue ReduceOp, in LowerPredReductionToSVE()
|
| /aosp_15_r20/prebuilts/bazel/common/android_tools/android_tools/ |
| D | desugar_jdk_libs.jar | desugar/sun/nio/fs/DesugarBasicFileAttributes.class
DesugarBasicFileAttributes.java
package desugar.sun. ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/turbine/ |
| D | core-oj.jar | java/awt/font/NumericShaper$Range.class
<Unknown>
package java.awt. ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/turbine-combined/ |
| D | core-oj.jar | com/sun/net/ssl/internal/ssl/X509ExtendedTrustManager.class
<Unknown>
package ... |
| /aosp_15_r20/out/target/common/obj/JAVA_LIBRARIES/core-oj.com.android.art.debug_intermediates/ |
| D | classes-header.jar | com/sun/net/ssl/internal/ssl/X509ExtendedTrustManager.class
<Unknown>
package ... |
| D | classes.jar | META-INF/
META-INF/MANIFEST.MF
calendars.properties
com/
com/sun ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-all/android_common/turbine-combined/ |
| D | core-all.jar | android/annotation/Discouraged.class
<Unknown>
package android.annotation
public abstract annotation android. ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-all/android_common/turbine/ |
| D | core-all.jar | java/awt/font/NumericShaper$Range.class
<Unknown>
package java.awt. ... |
| /aosp_15_r20/prebuilts/module_sdk/art/1/java/ |
| D | core-oj.jar | META-INF/
META-INF/MANIFEST.MF
calendars.properties
com/
com/sun ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/javac/ |
| D | core-oj.jar | META-INF/
META-INF/MANIFEST.MF
com/
com/sun/
com/ ... |
| /aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/withres/ |
| D | core-oj.jar | META-INF/
META-INF/MANIFEST.MF
calendars.properties
com/
com/sun ... |
| /aosp_15_r20/prebuilts/jdk/jdk8/linux-x86/lib/ |
| D | ct.sym | META-INF/
META-INF/MANIFEST.MF
META-INF/sym/rt. ... |
| /aosp_15_r20/prebuilts/jdk/jdk8/darwin-x86/lib/ |
| D | ct.sym | META-INF/
META-INF/MANIFEST.MF
META-INF/sym/rt. ... |
| /aosp_15_r20/prebuilts/jdk/jdk8/linux-x86/jre/lib/ |
| D | rt.jar | META-INF/
META-INF/MANIFEST.MF
com/oracle/net/Sdp$ ... |
| /aosp_15_r20/prebuilts/jdk/jdk8/darwin-x86/jre/lib/ |
| D | rt.jar | META-INF/
META-INF/MANIFEST.MF
apple/applescript/AppleScriptEngine.class
AppleScriptEngine ... |