Home
last modified time | relevance | path

Searched defs:reduce_mean (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScatterGatherKernel.cu47 static ReduceMean reduce_mean; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_math_ops.py671 def reduce_mean(input_tensor: ragged_tensor.Ragged, function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DScatterGatherKernel.cpp64 static ReduceMean reduce_mean; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmath_ops.py2587 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py1207 def reduce_mean(self, x): member in PForConfig