Home
last modified time | relevance | path

Searched defs:reduce_add (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Dcomm.py70 def reduce_add(inputs, destination=None): function
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dvec_ops.h170 inline float reduce_add(const T* x, size_t size) { in reduce_add() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScatterGatherKernel.cu38 static ReduceAdd reduce_add; variable
H A DIndexing.cu311 static ReduceAdd reduce_add; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DScatterGatherKernel.cpp54 static ReduceAdd reduce_add; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc147 XlaBuilder reduce_add("reduce_add"); in TEST_F() local