/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
H A D | Types.hpp | 122 ReduceOp reduceOp = ReduceOp::SUM; member 130 ReduceOp reduceOp = ReduceOp::SUM; member 153 ReduceOp reduceOp = ReduceOp::SUM; member
|
H A D | ProcessGroupGloo.cpp | 1023 ReduceOp reduceOp, in AsyncAllreduceWork() 1039 const ReduceOp reduceOp; member in c10d::__anon2d5ad4bf0911::AsyncAllreduceWork 1074 ReduceOp reduceOp, in AsyncAllreduceCoalescedWork() 1384 ReduceOp reduceOp, in AsyncAllreduceCUDAWork() 1626 ReduceOp reduceOp, in AsyncReduceWork() 1646 const ReduceOp reduceOp; member in c10d::__anon2d5ad4bf0e11::AsyncReduceWork 1685 ReduceOp reduceOp, in AsyncReduceCUDAWork()
|
H A D | ProcessGroupNCCL.cpp | 89 bool complexViewAsRealAllowed(const ReduceOp reduceOp) { in complexViewAsRealAllowed() 108 const ReduceOp& reduceOp, in unpackPreMulSum() 132 const ReduceOp& reduceOp, in getNcclReduceOp()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
H A D | lhlo_legalize_to_parallel_loops.cc | 70 void convertToReductionOperator(Location loc, scf::ReduceOp reduceOp, in convertToReductionOperator() 199 lmhlo::ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite() 233 lmhlo::ReduceOp reduceOp, ConversionPatternRewriter* rewriter) const { in createReduceOpInNestedParallelLoops() 379 scf::ReduceOp reduceOp = createReduceOpInNestedParallelLoops( in matchAndRewrite() local
|
H A D | lhlo_legalize_to_gpu.cc | 55 ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
H A D | shape_component_analysis.cc | 345 auto reduceOp = op.getDefiningOp<mhlo::ReduceOp>(); in backwardReduceShape() local 352 auto reduceOp = op.getDefiningOp<mhlo::ReduceOp>(); in forwardReduceShape() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | TensorModeKernel.cuh | 81 ReduceOp reduceOp, in reduceBlockWithNThreadLocalReductions()
|
H A D | WeightNorm.cu | 55 ReduceOp reduceOp) in reduce_block_into_lanes()
|
/aosp_15_r20/external/pytorch/torch/distributed/ |
H A D | _functional_collectives.py | 870 def _reduce_scatter_tensor_coalesced_meta(inputs, reduceOp, tag, rankset, group_size): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
H A D | loopnest.cpp | 2828 ReduceOpPtr reduceOp{nullptr}; in cacheAccesses() local
|
/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/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/out/target/common/obj/JAVA_LIBRARIES/core-oj.com.android.art.debug_intermediates/ |
D | classes.jar | META-INF/
META-INF/MANIFEST.MF
calendars.properties
com/
com/sun ... |