Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py669 strides = utils.compute_elementwise_output_strides(input)
695 strides = utils.compute_elementwise_output_strides(input)
2090 strides = utils.compute_elementwise_output_strides(a)
2697 strides = utils.compute_elementwise_output_strides(a)
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py514 def compute_elementwise_output_strides(*tensors) -> Tuple[int, ...]: function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py5055 stride = utils.compute_elementwise_output_strides(*_maybe_broadcast(*inputs))