Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py1689 def compute_reduction_output_shape( function
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py2353 output_shape = utils.compute_reduction_output_shape(inp.shape, dims)
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py2111 return utils.compute_reduction_output_shape(self.shape, dims)
3487 output_shape = utils.compute_reduction_output_shape(