Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dshape_analysis.cpp1267 static const auto multidim_reduce_with_keepdim = in PropagateTensorShapeOnNode() local
1295 return multidim_reduce_with_keepdim( in PropagateTensorShapeOnNode()
1327 auto output_types = multidim_reduce_with_keepdim( in PropagateTensorShapeOnNode()