Home
last modified time | relevance | path

Searched defs:reduceOp (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DTypes.hpp122 ReduceOp reduceOp = ReduceOp::SUM; member
130 ReduceOp reduceOp = ReduceOp::SUM; member
153 ReduceOp reduceOp = ReduceOp::SUM; member
H A DProcessGroupGloo.cpp1023 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 DProcessGroupNCCL.cpp89 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 Dlhlo_legalize_to_parallel_loops.cc70 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 Dlhlo_legalize_to_gpu.cc55 ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/
H A Dshape_component_analysis.cc345 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 DTensorModeKernel.cuh81 ReduceOp reduceOp, in reduceBlockWithNThreadLocalReductions()
H A DWeightNorm.cu55 ReduceOp reduceOp) in reduce_block_into_lanes()
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives.py870 def _reduce_scatter_tensor_coalesced_meta(inputs, reduceOp, tag, rankset, group_size): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.cpp2828 ReduceOpPtr reduceOp{nullptr}; in cacheAccesses() local
/aosp_15_r20/prebuilts/bazel/common/android_tools/android_tools/
Ddesugar_jdk_libs.jardesugar/sun/nio/fs/DesugarBasicFileAttributes.class DesugarBasicFileAttributes.java package desugar.sun. ...
/aosp_15_r20/prebuilts/module_sdk/art/1/java/
Dcore-oj.jarMETA-INF/ META-INF/MANIFEST.MF calendars.properties com/ com/sun ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/javac/
Dcore-oj.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/sun/ com/ ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-oj/android_common_apex31/withres/
Dcore-oj.jarMETA-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/
Dclasses.jarMETA-INF/ META-INF/MANIFEST.MF calendars.properties com/ com/sun ...