Searched defs:var_mean (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
| H A D | Functions.h | 89 TORCH_API inline std::tuple<Tensor, Tensor> var_mean(const Tensor& self, int dim) { in var_mean() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReduceOps.cpp | 1972 std::tuple<Tensor, Tensor> var_mean( in var_mean() function
1994 std::tuple<Tensor, Tensor> var_mean(const Tensor& self, bool unbiased) { in var_mean() function
2011 std::tuple<Tensor, Tensor> var_mean( in var_mean() function
2106 std::tuple<Tensor,Tensor> var_mean(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim… in var_mean() function
2133 std::tuple<Tensor,Tensor> var_mean(const Tensor& self, DimnameList dim, in var_mean() function
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 2535 def var_mean( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 5294 def var_mean(x, axis=None, *, correction=None, keepdim=False): function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 5182 def var_mean(g: jit_utils.GraphContext, input, *args): function
|