Home
last modified time | relevance | path

Searched refs:ComputeArrayStrides (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor_util.cpp22 std::vector<int64_t> ComputeArrayStrides(c10::ArrayRef<int64_t> sizes) { in ComputeArrayStrides() function
H A Dtensor_util.h14 TORCH_API std::vector<int64_t> ComputeArrayStrides(
H A Dtensor_impl.cpp157 updated_strides = ComputeArrayStrides(shape.Get().sizes()); in setup_size_properties()