Home
last modified time | relevance | path

Searched refs:compute_shape_std (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.h83 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 Dshape_inference.cpp392 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