Searched refs:compute_shape_std (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.h | 83 TORCH_API std::vector<torch::lazy::Shape> compute_shape_std(const at::Tensor & self, bool unbiased);
84 TORCH_API std::vector<torch::lazy::Shape> compute_shape_std(const at::Tensor & self, at::OptionalIn…
85 TORCH_API std::vector<torch::lazy::Shape> compute_shape_std(const at::Tensor & self, at::OptionalIn…
|
| H A D | shape_inference.cpp | 392 std::vector<Shape> compute_shape_std(const at::Tensor& self, bool unbiased) { in compute_shape_std() function
393 return compute_shape_std(self, ::std::nullopt, ::std::nullopt, false); in compute_shape_std()
395 std::vector<Shape> compute_shape_std( in compute_shape_std() function
400 return compute_shape_std(self, dim, ::std::nullopt, keepdim); in compute_shape_std()
402 std::vector<Shape> compute_shape_std( in compute_shape_std() function
|