Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.h98 TORCH_API std::vector<Shape> compute_shape_scalar(const at::Scalar& value, const at::ScalarType& ty…
H A Dshape_inference.cpp1156 std::vector<Shape> compute_shape_scalar( in compute_shape_scalar() function