Home
last modified time | relevance | path

Searched defs:reduce_sum (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_2/
Dreduce_sum.example.cpp7 namespace generated_tests::reduce_sum { namespace
74 namespace generated_tests::reduce_sum { namespace
175 namespace generated_tests::reduce_sum { namespace
242 namespace generated_tests::reduce_sum { namespace
343 namespace generated_tests::reduce_sum { namespace
410 namespace generated_tests::reduce_sum { namespace
511 namespace generated_tests::reduce_sum { namespace
578 namespace generated_tests::reduce_sum { namespace
679 namespace generated_tests::reduce_sum { namespace
746 namespace generated_tests::reduce_sum { namespace
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnccl_ops.py127 def reduce_sum(tensors): function
H A Dmath_ops.py2251 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): function
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/portable-simd/crates/core_simd/src/simd/num/
H A Duint.rs61 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dint.rs142 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dfloat.rs169 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/portable-simd/crates/core_simd/src/simd/num/
H A Duint.rs61 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dint.rs142 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dfloat.rs169 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/portable-simd/crates/core_simd/src/simd/num/
H A Duint.rs61 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dint.rs142 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dfloat.rs169 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
/aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/portable-simd/crates/core_simd/src/simd/num/
H A Duint.rs61 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dint.rs142 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dfloat.rs169 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/portable-simd/crates/core_simd/src/simd/num/
H A Duint.rs61 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dint.rs142 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
H A Dfloat.rs169 fn reduce_sum(self) -> Self::Scalar; in reduce_sum() method
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
H A Dsoftmax.h60 __device__ __forceinline__ void reduce_sum(Tensor<Engine0, Layout0> const& tensor, Tensor<Engine1, … in reduce_sum() function
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dml_sse3.c276 static inline __m128 reduce_sum(__m128 reg) { in reduce_sum() function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_math_ops.py610 def reduce_sum(input_tensor: ragged_tensor.Ragged, function
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py1430 def reduce_sum(v): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py1220 def reduce_sum(self, x): member in PForConfig