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
|