Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_runtime_fusion.cpp142 static std::vector<StrideInput> summarizeInputStrides(const TensorType& tt) { in summarizeInputStrides() function
194 input_striding.push_back(summarizeInputStrides(tt)); in TryGeneralizeInputDimensionsToSymbolicShapes()